diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_cache.py | 31 | ||||
| -rw-r--r-- | test/test_filters.py | 37 |
2 files changed, 65 insertions, 3 deletions
diff --git a/test/test_cache.py b/test/test_cache.py index d898b02..1c7b42a 100644 --- a/test/test_cache.py +++ b/test/test_cache.py @@ -139,6 +139,37 @@ class CacheTest(TemplateTest): ] assert m.kwargs == {} + def test_dynamic_key_with_context(self): + t = Template(""" + <%block name="foo" cached="True" cache_key="${mykey}"> + some block + </%block> + """) + m = self._install_mock_cache(t) + t.render(mykey="thekey") + t.render(mykey="thekey") + eq_( + result_lines(t.render(mykey="thekey")), + ["some block"] + ) + eq_(m.key, "thekey") + + t = Template(""" + <%def name="foo()" cached="True" cache_key="${mykey}"> + some def + </%def> + ${foo()} + """) + m = self._install_mock_cache(t) + t.render(mykey="thekey") + t.render(mykey="thekey") + eq_( + result_lines(t.render(mykey="thekey")), + ["some def"] + ) + eq_(m.key, "thekey") + + def test_dynamic_key_with_funcargs(self): t = Template(""" <%def name="foo(num=5)" cached="True" cache_key="foo_${str(num)}"> diff --git a/test/test_filters.py b/test/test_filters.py index 13492d8..684705d 100644 --- a/test/test_filters.py +++ b/test/test_filters.py @@ -3,7 +3,7 @@ from mako.template import Template import unittest from mako import util -from test import TemplateTest, eq_, skip_if +from test import TemplateTest, eq_, skip_if, assert_raises from util import result_lines, flatten_result class FilterTest(TemplateTest): @@ -60,7 +60,11 @@ class FilterTest(TemplateTest): ${foo()} """) - assert flatten_result(t.render(x="this is x", myfilter=lambda t: "MYFILTER->%s<-MYFILTER" % t)) == "MYFILTER-> this is foo <-MYFILTER" + eq_( + flatten_result(t.render(x="this is x", + myfilter=lambda t: "MYFILTER->%s<-MYFILTER" % t)), + "MYFILTER-> this is foo <-MYFILTER" + ) def test_import(self): t = Template(""" @@ -104,6 +108,33 @@ class FilterTest(TemplateTest): """) assert t.render().strip() == "<tag>this is html</tag>" + def test_block_via_context(self): + t = Template(""" + <%block name="foo" filter="myfilter"> + some text + </%block> + """) + def myfilter(text): + return "MYTEXT" + text + eq_( + result_lines(t.render(myfilter=myfilter)), + ["MYTEXT", "some text"] + ) + + def test_def_via_context(self): + t = Template(""" + <%def name="foo()" filter="myfilter"> + some text + </%def> + ${foo()} + """) + def myfilter(text): + return "MYTEXT" + text + eq_( + result_lines(t.render(myfilter=myfilter)), + ["MYTEXT", "some text"] + ) + def test_nflag(self): t = Template(""" ${"<tag>this is html</tag>" | n} @@ -122,7 +153,7 @@ class FilterTest(TemplateTest): """) assert t.render().strip() == "<tag>this is html</tag>" - def testnonexpression(self): + def test_non_expression(self): t = Template(""" <%! def a(text): |