1
# -*- coding: utf-8 -*-
3
from mako.template import Template
4
from mako.lookup import TemplateLookup
5
from mako.ext.preprocessors import convert_comments
6
from mako import exceptions
7
import unittest, re, os
8
from util import flatten_result, result_lines
11
if not os.access('./test_htdocs', os.F_OK):
12
os.mkdir('./test_htdocs')
13
if not os.access('./test_htdocs/subdir', os.F_OK):
14
os.mkdir('./test_htdocs/subdir')
15
file('./test_htdocs/unicode.html', 'w').write("""## -*- coding: utf-8 -*-
16
Alors vous imaginez ma surprise, au lever du jour, quand une drôle de petit voix m’a réveillé. Elle disait: « S’il vous plaît… dessine-moi un mouton! »""")
17
file('./test_htdocs/unicode_syntax_error.html', 'w').write("""## -*- coding: utf-8 -*-
18
<% print 'Alors vous imaginez ma surprise, au lever du jour, quand une drôle de petit voix m’a réveillé. Elle disait: « S’il vous plaît… dessine-moi un mouton! » %>""")
19
file('./test_htdocs/unicode_runtime_error.html', 'w').write("""## -*- coding: utf-8 -*-
20
<% print 'Alors vous imaginez ma surprise, au lever du jour, quand une drôle de petit voix m’a réveillé. Elle disait: « S’il vous plaît… dessine-moi un mouton! »' + int(5/0) %>""")
22
file('./test_htdocs/bommagic.html', 'w').write(codecs.BOM_UTF8 + """## -*- coding: utf-8 -*-
23
Alors vous imaginez ma surprise, au lever du jour, quand une drôle de petit voix m’a réveillé. Elle disait: « S’il vous plaît… dessine-moi un mouton! »""")
24
file('./test_htdocs/badbom.html', 'w').write(codecs.BOM_UTF8 + """## -*- coding: ascii -*-
25
Alors vous imaginez ma surprise, au lever du jour, quand une drôle de petit voix m’a réveillé. Elle disait: « S’il vous plaît… dessine-moi un mouton! »""")
26
file('./test_htdocs/bom.html', 'w').write(codecs.BOM_UTF8 + """Alors vous imaginez ma surprise, au lever du jour, quand une drôle de petit voix m’a réveillé. Elle disait: « S’il vous plaît… dessine-moi un mouton! »""")
28
class EncodingTest(unittest.TestCase):
29
def test_unicode(self):
30
template = Template(u"""Alors vous imaginez ma surprise, au lever du jour, quand une drôle de petit voix m’a réveillé. Elle disait: « S’il vous plaît… dessine-moi un mouton! »""")
31
assert template.render_unicode() == u"""Alors vous imaginez ma surprise, au lever du jour, quand une drôle de petit voix m’a réveillé. Elle disait: « S’il vous plaît… dessine-moi un mouton! »"""
33
def test_unicode_arg(self):
34
val = u"""Alors vous imaginez ma surprise, au lever du jour, quand une drôle de petit voix m’a réveillé. Elle disait: « S’il vous plaît… dessine-moi un mouton! »"""
35
template = Template("${val}")
36
assert template.render_unicode(val=val) == u"""Alors vous imaginez ma surprise, au lever du jour, quand une drôle de petit voix m’a réveillé. Elle disait: « S’il vous plaît… dessine-moi un mouton! »"""
38
def test_unicode_file(self):
39
template = Template(filename='./test_htdocs/unicode.html', module_directory='./test_htdocs')
40
assert template.render_unicode() == u"""Alors vous imaginez ma surprise, au lever du jour, quand une drôle de petit voix m’a réveillé. Elle disait: « S’il vous plaît… dessine-moi un mouton! »"""
42
def test_unicode_bom(self):
43
template = Template(filename='./test_htdocs/bom.html', module_directory='./test_htdocs')
44
assert template.render_unicode() == u"""Alors vous imaginez ma surprise, au lever du jour, quand une drôle de petit voix m’a réveillé. Elle disait: « S’il vous plaît… dessine-moi un mouton! »"""
46
template = Template(filename='./test_htdocs/bommagic.html', module_directory='./test_htdocs')
47
assert template.render_unicode() == u"""Alors vous imaginez ma surprise, au lever du jour, quand une drôle de petit voix m’a réveillé. Elle disait: « S’il vous plaît… dessine-moi un mouton! »"""
50
template = Template(filename='./test_htdocs/badbom.html', module_directory='./test_htdocs')
52
except exceptions.CompileException:
55
def test_unicode_memory(self):
56
val = u"""Alors vous imaginez ma surprise, au lever du jour, quand une drôle de petit voix m’a réveillé. Elle disait: « S’il vous plaît… dessine-moi un mouton! »"""
57
val = "## coding: utf-8\n" + val.encode('utf-8')
58
template = Template(val)
60
assert template.render_unicode() == u"""Alors vous imaginez ma surprise, au lever du jour, quand une drôle de petit voix m’a réveillé. Elle disait: « S’il vous plaît… dessine-moi un mouton! »"""
62
def test_unicode_text(self):
63
val = u"""<%text>Alors vous imaginez ma surprise, au lever du jour, quand une drôle de petit voix m’a réveillé. Elle disait: « S’il vous plaît… dessine-moi un mouton! »</%text>"""
64
val = "## -*- coding: utf-8 -*-\n" + val.encode('utf-8')
65
template = Template(val)
67
assert template.render_unicode() == u"""Alors vous imaginez ma surprise, au lever du jour, quand une drôle de petit voix m’a réveillé. Elle disait: « S’il vous plaît… dessine-moi un mouton! »"""
69
def test_unicode_text_ccall(self):
72
${capture(caller.body)}
75
<%text>Alors vous imaginez ma surprise, au lever du jour, quand une drôle de petit voix m’a réveillé. Elle disait: « S’il vous plaît… dessine-moi un mouton! »</%text>
77
val = "## -*- coding: utf-8 -*-\n" + val.encode('utf-8')
78
template = Template(val)
80
assert flatten_result(template.render_unicode()) == u"""Alors vous imaginez ma surprise, au lever du jour, quand une drôle de petit voix m’a réveillé. Elle disait: « S’il vous plaît… dessine-moi un mouton! »"""
82
def test_unicode_literal_in_expr(self):
83
template = Template(u"""## -*- coding: utf-8 -*-
84
${u"Alors vous imaginez ma surprise, au lever du jour, quand une drôle de petit voix m’a réveillé. Elle disait: « S’il vous plaît… dessine-moi un mouton! »"}
86
assert template.render_unicode().strip() == u"""Alors vous imaginez ma surprise, au lever du jour, quand une drôle de petit voix m’a réveillé. Elle disait: « S’il vous plaît… dessine-moi un mouton! »"""
88
def test_unicode_literal_in_code(self):
89
template = Template(u"""## -*- coding: utf-8 -*-
91
context.write(u"Alors vous imaginez ma surprise, au lever du jour, quand une drôle de petit voix m’a réveillé. Elle disait: « S’il vous plaît… dessine-moi un mouton! »")
94
assert template.render_unicode().strip() == u"""Alors vous imaginez ma surprise, au lever du jour, quand une drôle de petit voix m’a réveillé. Elle disait: « S’il vous plaît… dessine-moi un mouton! »"""
96
def test_unicode_literal_in_controlline(self):
97
template = Template(u"""## -*- coding: utf-8 -*-
99
x = u"drôle de petit voix m’a réveillé."
101
% if x==u"drôle de petit voix m’a réveillé.":
105
assert template.render_unicode().strip() == u"""hi, drôle de petit voix m’a réveillé."""
107
def test_unicode_literal_in_def(self):
108
template = Template(u"""## -*- coding: utf-8 -*-
109
<%def name="bello(foo, bar)">
113
<%call expr="bello(foo=u'árvíztűrő tükörfúrógép', bar=u'ÁRVÍZTŰRŐ TÜKÖRFÚRÓGÉP')">
114
</%call>""".encode('utf-8'))
115
assert flatten_result(template.render_unicode()) == u"""Foo: árvíztűrő tükörfúrógép Bar: ÁRVÍZTŰRŐ TÜKÖRFÚRÓGÉP"""
117
template = Template(u"""## -*- coding: utf-8 -*-
118
<%def name="hello(foo=u'árvíztűrő tükörfúrógép', bar=u'ÁRVÍZTŰRŐ TÜKÖRFÚRÓGÉP')">
122
${ hello() }""".encode('utf-8'))
123
assert flatten_result(template.render_unicode()) == u"""Foo: árvíztűrő tükörfúrógép Bar: ÁRVÍZTŰRŐ TÜKÖRFÚRÓGÉP"""
125
def test_input_encoding(self):
126
"""test the 'input_encoding' flag on Template, and that unicode objects arent double-decoded"""
127
s2 = u"hello ${f(u'śląsk')}"
128
res = Template(s2, input_encoding='utf-8').render_unicode(f=lambda x:x)
129
assert res == u"hello śląsk"
131
s2 = u"## -*- coding: utf-8 -*-\nhello ${f(u'śląsk')}"
132
res = Template(s2).render_unicode(f=lambda x:x)
133
assert res == u"hello śląsk"
135
def test_raw_strings(self):
136
"""test that raw strings go straight thru with default_filters turned off"""
138
s = u"## -*- coding: utf-8 -*-\nhello ${x}"
139
t = Template(s, default_filters=[])
141
assert y == "hello śląsk"
143
# now, the way you *should* be doing it....
144
q = g.decode('utf-8')
145
y = t.render_unicode(x=q)
146
assert y == u"hello śląsk"
148
def test_encoding(self):
149
val = u"""Alors vous imaginez ma surprise, au lever du jour, quand une drôle de petit voix m’a réveillé. Elle disait: « S’il vous plaît… dessine-moi un mouton! »"""
150
template = Template(val, output_encoding='utf-8')
151
assert template.render() == val.encode('utf-8')
153
def test_encoding_errors(self):
154
val = u"""KGB (transliteration of "КГБ") is the Russian-language abbreviation for Committee for State Security, (Russian: Комит́ет Госуд́арственной Безоп́асности (help·info); Komitet Gosudarstvennoy Bezopasnosti)"""
155
template = Template(val, output_encoding='iso-8859-1', encoding_errors='replace')
156
assert template.render() == val.encode('iso-8859-1', 'replace')
158
def test_read_unicode(self):
159
lookup = TemplateLookup(directories=['./test_htdocs'], filesystem_checks=True, output_encoding='utf-8')
160
template = lookup.get_template('/read_unicode.html')
161
data = template.render(path=os.path.join('./test_htdocs', 'internationalization.html'))
164
class PageArgsTest(unittest.TestCase):
165
def test_basic(self):
166
template = Template("""
167
<%page args="x, y, z=7"/>
169
this is page, ${x}, ${y}, ${z}
172
assert flatten_result(template.render(x=5, y=10)) == "this is page, 5, 10, 7"
173
assert flatten_result(template.render(x=5, y=10, z=32)) == "this is page, 5, 10, 32"
175
template.render(y=10)
180
def test_with_context(self):
181
template = Template("""
182
<%page args="x, y, z=7"/>
184
this is page, ${x}, ${y}, ${z}, ${w}
187
assert flatten_result(template.render(x=5, y=10, w=17)) == "this is page, 5, 10, 7, 17"
189
def test_overrides_builtins(self):
190
template = Template("""
193
this is page, id is ${id}
196
assert flatten_result(template.render(id="im the id")) == "this is page, id is im the id"
198
class IncludeTest(unittest.TestCase):
199
def test_basic(self):
200
lookup = TemplateLookup()
201
lookup.put_string("a", """
203
<%include file="b" args="a=3,b=4,c=5"/>
205
lookup.put_string("b", """
206
<%page args="a,b,c"/>
207
this is b. ${a}, ${b}, ${c}
209
assert flatten_result(lookup.get_template("a").render()) == "this is a this is b. 3, 4, 5"
211
def test_localargs(self):
212
lookup = TemplateLookup()
213
lookup.put_string("a", """
215
<%include file="b" args="a=a,b=b,c=5"/>
217
lookup.put_string("b", """
218
<%page args="a,b,c"/>
219
this is b. ${a}, ${b}, ${c}
221
assert flatten_result(lookup.get_template("a").render(a=7,b=8)) == "this is a this is b. 7, 8, 5"
223
def test_viakwargs(self):
224
lookup = TemplateLookup()
225
lookup.put_string("a", """
227
<%include file="b" args="c=5, **context.kwargs"/>
229
lookup.put_string("b", """
230
<%page args="a,b,c"/>
231
this is b. ${a}, ${b}, ${c}
233
#print lookup.get_template("a").code
234
assert flatten_result(lookup.get_template("a").render(a=7,b=8)) == "this is a this is b. 7, 8, 5"
236
def test_include_withargs(self):
237
lookup = TemplateLookup()
238
lookup.put_string("a", """
240
<%include file="${i}" args="c=5, **context.kwargs"/>
242
lookup.put_string("b", """
243
<%page args="a,b,c"/>
244
this is b. ${a}, ${b}, ${c}
246
assert flatten_result(lookup.get_template("a").render(a=7,b=8,i='b')) == "this is a this is b. 7, 8, 5"
248
class ControlTest(unittest.TestCase):
249
def test_control(self):
251
## this is a template.
253
% if x.has_key('test'):
256
no x does not have test
260
assert result_lines(t.render(y=[{'test':'one'}, {'foo':'bar'}, {'foo':'bar', 'test':'two'}])) == [
262
"no x does not have test",
266
def test_multiline_control(self):
269
[y for y in [1,2,3]]:
274
assert flatten_result(t.render()) == "1 2 3"
276
class GlobalsTest(unittest.TestCase):
277
def test_globals(self):
284
assert t.render().strip() == "y is hi"
286
class RichTracebackTest(unittest.TestCase):
287
def do_test_traceback(self, utf8, memory, syntax):
290
source = u'## coding: utf-8\n<% print "m’a réveillé. Elle disait: « S’il vous plaît… dessine-moi un mouton! » %>'
292
source = u'## coding: utf-8\n<% print u"m’a réveillé. Elle disait: « S’il vous plaît… dessine-moi un mouton! »" + str(5/0) %>'
294
source = source.encode('utf-8')
297
templateargs = {'text':source}
300
filename = './test_htdocs/unicode_syntax_error.html'
302
filename = './test_htdocs/unicode_runtime_error.html'
303
source = file(filename).read()
305
source = source.decode('utf-8')
306
templateargs = {'filename':filename}
308
template = Template(**templateargs)
310
template.render_unicode()
313
tback = exceptions.RichTraceback()
315
assert tback.source == source.decode('utf-8')
317
assert tback.source == source
319
for utf8 in (True, False):
320
for memory in (True, False):
321
for syntax in (True, False):
323
self.do_test_traceback(utf8, memory, syntax)
324
name = 'test_%s_%s_%s' % (utf8 and 'utf8' or 'unicode', memory and 'memory' or 'file', syntax and 'syntax' or 'runtime')
326
do_test.__name__ = name
329
setattr(RichTracebackTest, name, do_test)
332
class ModuleDirTest(unittest.TestCase):
333
def test_basic(self):
334
file('./test_htdocs/modtest.html', 'w').write("""this is a test""")
335
file('./test_htdocs/subdir/modtest.html', 'w').write("""this is a test""")
336
t = Template(filename='./test_htdocs/modtest.html', module_directory='./test_htdocs/modules')
337
t2 = Template(filename='./test_htdocs/subdir/modtest.html', module_directory='./test_htdocs/modules')
338
assert t.module.__file__ == os.path.abspath('./test_htdocs/modules/test_htdocs/modtest.html.py')
339
assert t2.module.__file__ == os.path.abspath('./test_htdocs/modules/test_htdocs/subdir/modtest.html.py')
340
def test_callable(self):
341
file('./test_htdocs/modtest.html', 'w').write("""this is a test""")
342
file('./test_htdocs/subdir/modtest.html', 'w').write("""this is a test""")
343
def get_modname(filename, uri):
344
return os.path.dirname(filename) + "/foo/" + os.path.basename(filename) + ".py"
345
lookup = TemplateLookup('./test_htdocs', modulename_callable=get_modname)
346
t = lookup.get_template('/modtest.html')
347
t2 = lookup.get_template('/subdir/modtest.html')
348
assert t.module.__file__ == 'test_htdocs/foo/modtest.html.py'
349
assert t2.module.__file__ == 'test_htdocs/subdir/foo/modtest.html.py'
351
class PreprocessTest(unittest.TestCase):
352
def test_old_comments(self):
356
# more old style comment
361
""", preprocessor=convert_comments)
363
assert flatten_result(t.render()) == "im a template - # not a comment - ## not a comment"
365
if __name__ == '__main__':