diff options
Diffstat (limited to 'test/test_template.py')
-rw-r--r-- | test/test_template.py | 17 |
1 files changed, 15 insertions, 2 deletions
diff --git a/test/test_template.py b/test/test_template.py index c5873dc..a6a491f 100644 --- a/test/test_template.py +++ b/test/test_template.py @@ -757,9 +757,22 @@ class UndefinedVarsTest(TemplateTest): ['t is: T', 'a,b,c'] ) +class StopRenderingTest(TemplateTest): + def test_return_in_template(self): + t = Template(""" + Line one + <% return STOP_RENDERING %> + Line Three + """, strict_undefined=True) + + eq_( + result_lines(t.render()), + ['Line one'] + ) + class ReservedNameTest(TemplateTest): def test_names_on_context(self): - for name in ('context', 'loop', 'UNDEFINED'): + for name in ('context', 'loop', 'UNDEFINED', 'STOP_RENDERING'): assert_raises_message( exceptions.NameConflictError, r"Reserved words passed to render\(\): %s" % name, @@ -767,7 +780,7 @@ class ReservedNameTest(TemplateTest): ) def test_names_in_template(self): - for name in ('context', 'loop', 'UNDEFINED'): + for name in ('context', 'loop', 'UNDEFINED', 'STOP_RENDERING'): assert_raises_message( exceptions.NameConflictError, r"Reserved words declared in template: %s" % name, |