Newer
Older
from mako.template import Template
import unittest
import re
class ComponentTest(unittest.TestCase):
def test_component_noargs(self):
template = Template("""
${mycomp()}
<%component name="mycomp">
hello mycomp ${variable}
</%component>
""")
print template.code
assert template.render(variable='hi').strip() == """hello mycomp hi"""
def test_component_blankargs(self):
template = Template("""
<%component name="mycomp()">
hello mycomp ${variable}
</%component>
${mycomp()}""")
#print template.code
assert template.render(variable='hi').strip() == """hello mycomp hi"""
def test_component_args(self):
template = Template("""
<%component name="mycomp(a, b)">
hello mycomp ${variable}, ${a}, ${b}
</%component>
${mycomp(5, 6)}""")
#print template.code
assert template.render(variable='hi', a=5, b=6).strip() == """hello mycomp hi, 5, 6"""
def test_inter_component(self):
"""test components calling each other"""
template = Template("""
${b()}
<%component name="a">\
im a
</%component>
<%component name="b">
im b
and heres a: ${a()}
</%component>
<%component name="c">
im c
</%component>
""")
#print template.code
# check that "a" is declared in "b", but not in "c"
assert "a" not in template.module.render_c.func_code.co_varnames
assert "a" in template.module.render_b.func_code.co_varnames
# then test output
assert template.render().strip() == "im b\nand heres a: im a"
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
def test_local_names(self):
template = """
<%component name="a">
this is a, and y is ${y}
</%component>
${a()}
<%
y = 7
%>
${a()}
"""
t = Template(template)
print t.code
result = t.render()
result = re.sub(r'[\s\n]+', ' ', result).strip()
assert result == "this is a, and y is None this is a, and y is 7"
def test_local_names_2(self):
template = """
y is ${y}
<%
y = 7
%>
y is ${y}
"""
t = Template(template)
result = t.render()
result = re.sub(r'[\s\n]+', ' ', result).strip()
assert result == "y is None y is 7"
class NestedComponentTest(unittest.TestCase):
def test_nested_component(self):
template = """
${hi()}
<%component name="hi">
hey, im hi.
and heres ${foo()}, ${bar()}
<%component name="foo">
this is foo
</%component>
<%component name="bar">
this is bar
</%component>
</%component>
"""
t = Template(template)
print t.code
result = t.render()
result = re.sub(r'[\s\n]+', ' ', result).strip()
assert result == "hey, im hi. and heres this is foo , this is bar"
def test_nested_with_args(self):
template = Template("""
${a()}
<%component name="a">
<%component name="b(x, y=2)">
b x is ${x} y is ${y}
</%component>
a ${b(5)}
</%component>
""")
#print template.code
result = template.render()
#print result
result = re.sub(r'[\s\n]+', ' ', result).strip()
assert result == "a b x is 5 y is 2"
def test_nested_component_2(self):
template = Template("""
${a()}
<%component name="a">
<%component name="b">
<%component name="c">
comp c
</%component>
${c()}
</%component>
${b()}
</%component>
""")
print template.code
def test_nested_nested_component(self):
template = """
${a()}
<%component name="a">
a_b1
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
</%component>
<%component name="b2">
a_b2 ${c1()}
<%component name="c1">
a_b2_c1
</%component>
</%component>
<%component name="b3">
a_b3 ${c1()}
<%component name="c1">
a_b3_c1 heres x: ${x}
<%
y = 7
%>
y is ${y}
</%component>
<%component name="c2">
a_b3_c2
y is ${y}
c1 is ${c1()}
</%component>
${c2()}
</%component>
${b1()} ${b2()} ${b3()}
</%component>
"""
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
t = Template(template, format_exceptions=False)
print t.code
result = t.render(x=5)
result = re.sub(r'[\s\n]+', ' ', result).strip()
#print result
assert result == "a a_b1 a_b2 a_b2_c1 a_b3 a_b3_c1 heres x: 5 y is 7 a_b3_c2 y is None c1 is a_b3_c1 heres x: 5 y is 7"
def test_nested_nested_component_2(self):
template = """
<%component name="a">
this is a ${b()}
<%component name="b">
this is b
${c()}
</%component>
<%component name="c">
this is c
</%component>
</%component>
${a()}
"""
t = Template(template)
print t.code
def test_local_local_names(self):
"""test assignment of variables inside nested components, which requires extra scoping logic"""
template = """
heres y: ${y}
<%component name="a">
<%component name="b">
b, heres y: ${y}
<%
y = 19
%>
b, heres c: ${c()}
b, heres y again: ${y}
</%component>
a, heres y: ${y}
<%
y = 10
x = 12
%>
a, now heres y: ${y}
a, heres b: ${b()}
<%component name="c">
this is c
</%component>
</%component>
<%
y = 7
%>
now heres y ${y}
${a()}
heres y again: ${y}
"""
t = Template(template, format_exceptions=False)
print t.code
result = t.render(y=5)
result = re.sub(r'[\s\n]+', ' ', result).strip()
print result
assert result == "heres y: 5 now heres y 7 a, heres y: 7 a, now heres y: 10 a, heres b: b, heres y: 10 b, heres c: this is c b, heres y again: 19 heres y again: 7"
class NamespaceTest(unittest.TestCase):
def test_inline(self):
template = """
<%namespace name="x">
<%component name="a">
this is x a
</%component>
<%component name="b">
this is x b, and heres ${x.a()}
</%component>
</%namespace>
${x.a()}
${x.b()}
"""
t = Template(template)
print t.code
print t.render()
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
class ExceptionTest(unittest.TestCase):
def test_raise(self):
template = Template("""
<%
raise Exception("this is a test")
%>
""", format_exceptions=False)
try:
template.render()
assert False
except Exception, e:
assert str(e) == "this is a test"
def test_handler(self):
def handle(context, error):
context.write("error message is " + str(error))
return True
template = Template("""
<%
raise Exception("this is a test")
%>
""", error_handler=handle)
assert template.render().strip() == """error message is this is a test"""
if __name__ == '__main__':
unittest.main()