1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
# -*- coding: utf-8 -*-
import os
import sys
import unittest
from mako import compat
from mako import exceptions
from mako import util
from test import assert_raises_message
from test import eq_
from test import skip_if
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 = ("drôl m’a rée « S’il")
buf = util.FastEncodingBuffer(encoding="utf-8")
buf.write(s[0:10])
buf.write(s[10:])
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")
assert b"test_util" in data
@skip_if(lambda: compat.pypy, "Pypy does this differently")
def test_load_module(self):
path = os.path.join(os.path.dirname(__file__), "module_to_import.py")
some_module = compat.load_module("test.module_to_import", path)
self.assertNotIn("test.module_to_import", sys.modules)
self.assertIn("some_function", dir(some_module))
import test.module_to_import
self.assertNotEqual(some_module, test.module_to_import)
def test_load_plugin_failure(self):
loader = util.PluginLoader("fakegroup")
assert_raises_message(
exceptions.RuntimeException,
"Can't load plugin fakegroup fake",
loader.load,
"fake",
)
|
