| # -*- coding: utf-8 -*- |
| |
| import os |
| import unittest |
| from mako import util |
| from test import eq_ |
| |
| class UtilTest(unittest.TestCase): |
| def test_fast_buffer_write(self): |
| buf = util.FastEncodingBuffer() |
| buf.write("string a ") |
| buf.write("string b") |
| eq_(buf.getvalue(), "string a string b") |
| |
| def test_fast_buffer_truncate(self): |
| buf = util.FastEncodingBuffer() |
| buf.write("string a ") |
| buf.write("string b") |
| buf.truncate() |
| buf.write("string c ") |
| buf.write("string d") |
| eq_(buf.getvalue(), "string c string d") |
| |
| def test_fast_buffer_encoded(self): |
| s = u"drôl m’a rée « S’il" |
| buf = util.FastEncodingBuffer(encoding='utf-8') |
| buf.write(s[0:10]) |
| buf.write(s[10:]) |
| q = buf.getvalue() |
| eq_(buf.getvalue(), s.encode('utf-8')) |
| |
| def test_read_file(self): |
| fn = os.path.join(os.path.dirname(__file__), 'test_util.py') |
| data = util.read_file(fn, 'rb') |
| self.failUnless('test_util' in data) |
| |
| def test_load_module(self): |
| fn = os.path.join(os.path.dirname(__file__), 'test_util.py') |
| module = util.load_module('mako.template', fn) |
| import mako.template |
| self.assertEqual(module, mako.template) |
