parent
827b4f93e4
commit
66c40bee6f
29 changed files with 96 additions and 3319 deletions
@ -0,0 +1,31 @@ |
||||
Jinja is written and maintained by the Jinja Team and various |
||||
contributors: |
||||
|
||||
Lead Developer: |
||||
|
||||
- Armin Ronacher <armin.ronacher@active-4.com> |
||||
|
||||
Developers: |
||||
|
||||
- Christoph Hack |
||||
- Georg Brandl |
||||
|
||||
Contributors: |
||||
|
||||
- Bryan McLemore |
||||
- Mickaël Guérin <kael@crocobox.org> |
||||
- Cameron Knight |
||||
- Lawrence Journal-World. |
||||
- David Cramer |
||||
|
||||
Patches and suggestions: |
||||
|
||||
- Ronny Pfannschmidt |
||||
- Axel Böhm |
||||
- Alexey Melchakov |
||||
- Bryan McLemore |
||||
- Clovis Fabricio (nosklo) |
||||
- Cameron Knight |
||||
- Peter van Dijk (Habbie) |
||||
- Stefan Ebner |
||||
- Rene Leonhardt |
@ -0,0 +1,31 @@ |
||||
Copyright (c) 2009 by the Jinja Team, see AUTHORS for more details. |
||||
|
||||
Some rights reserved. |
||||
|
||||
Redistribution and use in source and binary forms, with or without |
||||
modification, are permitted provided that the following conditions are |
||||
met: |
||||
|
||||
* Redistributions of source code must retain the above copyright |
||||
notice, this list of conditions and the following disclaimer. |
||||
|
||||
* Redistributions in binary form must reproduce the above |
||||
copyright notice, this list of conditions and the following |
||||
disclaimer in the documentation and/or other materials provided |
||||
with the distribution. |
||||
|
||||
* The names of the contributors may not be used to endorse or |
||||
promote products derived from this software without specific |
||||
prior written permission. |
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||
"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||
OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||
LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
@ -1,161 +0,0 @@ |
||||
# -*- coding: utf-8 -*- |
||||
""" |
||||
jinja2.tests |
||||
~~~~~~~~~~~~ |
||||
|
||||
Jinja test functions. Used with the "is" operator. |
||||
|
||||
:copyright: (c) 2010 by the Jinja Team. |
||||
:license: BSD, see LICENSE for more details. |
||||
""" |
||||
import re |
||||
from jinja2.runtime import Undefined |
||||
|
||||
try: |
||||
from collections import Mapping as MappingType |
||||
except ImportError: |
||||
import UserDict |
||||
MappingType = (UserDict.UserDict, UserDict.DictMixin, dict) |
||||
|
||||
# nose, nothing here to test |
||||
__test__ = False |
||||
|
||||
|
||||
number_re = re.compile(r'^-?\d+(\.\d+)?$') |
||||
regex_type = type(number_re) |
||||
|
||||
|
||||
try: |
||||
test_callable = callable |
||||
except NameError: |
||||
def test_callable(x): |
||||
return hasattr(x, '__call__') |
||||
|
||||
|
||||
def test_odd(value): |
||||
"""Return true if the variable is odd.""" |
||||
return value % 2 == 1 |
||||
|
||||
|
||||
def test_even(value): |
||||
"""Return true if the variable is even.""" |
||||
return value % 2 == 0 |
||||
|
||||
|
||||
def test_divisibleby(value, num): |
||||
"""Check if a variable is divisible by a number.""" |
||||
return value % num == 0 |
||||
|
||||
|
||||
def test_defined(value): |
||||
"""Return true if the variable is defined: |
||||
|
||||
.. sourcecode:: jinja |
||||
|
||||
{% if variable is defined %} |
||||
value of variable: {{ variable }} |
||||
{% else %} |
||||
variable is not defined |
||||
{% endif %} |
||||
|
||||
See the :func:`default` filter for a simple way to set undefined |
||||
variables. |
||||
""" |
||||
return not isinstance(value, Undefined) |
||||
|
||||
|
||||
def test_undefined(value): |
||||
"""Like :func:`defined` but the other way round.""" |
||||
return isinstance(value, Undefined) |
||||
|
||||
|
||||
def test_none(value): |
||||
"""Return true if the variable is none.""" |
||||
return value is None |
||||
|
||||
|
||||
def test_lower(value): |
||||
"""Return true if the variable is lowercased.""" |
||||
return unicode(value).islower() |
||||
|
||||
|
||||
def test_upper(value): |
||||
"""Return true if the variable is uppercased.""" |
||||
return unicode(value).isupper() |
||||
|
||||
|
||||
def test_string(value): |
||||
"""Return true if the object is a string.""" |
||||
return isinstance(value, basestring) |
||||
|
||||
|
||||
def test_mapping(value): |
||||
"""Return true if the object is a mapping (dict etc.). |
||||
|
||||
.. versionadded:: 2.6 |
||||
""" |
||||
return isinstance(value, MappingType) |
||||
|
||||
|
||||
def test_number(value): |
||||
"""Return true if the variable is a number.""" |
||||
return isinstance(value, (int, long, float, complex)) |
||||
|
||||
|
||||
def test_sequence(value): |
||||
"""Return true if the variable is a sequence. Sequences are variables |
||||
that are iterable. |
||||
""" |
||||
try: |
||||
len(value) |
||||
value.__getitem__ |
||||
except: |
||||
return False |
||||
return True |
||||
|
||||
|
||||
def test_sameas(value, other): |
||||
"""Check if an object points to the same memory address than another |
||||
object: |
||||
|
||||
.. sourcecode:: jinja |
||||
|
||||
{% if foo.attribute is sameas false %} |
||||
the foo attribute really is the `False` singleton |
||||
{% endif %} |
||||
""" |
||||
return value is other |
||||
|
||||
|
||||
def test_iterable(value): |
||||
"""Check if it's possible to iterate over an object.""" |
||||
try: |
||||
iter(value) |
||||
except TypeError: |
||||
return False |
||||
return True |
||||
|
||||
|
||||
def test_escaped(value): |
||||
"""Check if the value is escaped.""" |
||||
return hasattr(value, '__html__') |
||||
|
||||
|
||||
TESTS = { |
||||
'odd': test_odd, |
||||
'even': test_even, |
||||
'divisibleby': test_divisibleby, |
||||
'defined': test_defined, |
||||
'undefined': test_undefined, |
||||
'none': test_none, |
||||
'lower': test_lower, |
||||
'upper': test_upper, |
||||
'string': test_string, |
||||
'mapping': test_mapping, |
||||
'number': test_number, |
||||
'sequence': test_sequence, |
||||
'iterable': test_iterable, |
||||
'callable': test_callable, |
||||
'sameas': test_sameas, |
||||
'escaped': test_escaped |
||||
} |
@ -1,95 +0,0 @@ |
||||
# -*- coding: utf-8 -*- |
||||
""" |
||||
jinja2.testsuite |
||||
~~~~~~~~~~~~~~~~ |
||||
|
||||
All the unittests of Jinja2. These tests can be executed by |
||||
either running run-tests.py using multiple Python versions at |
||||
the same time. |
||||
|
||||
:copyright: (c) 2010 by the Jinja Team. |
||||
:license: BSD, see LICENSE for more details. |
||||
""" |
||||
import os |
||||
import re |
||||
import sys |
||||
import unittest |
||||
from traceback import format_exception |
||||
from jinja2 import loaders |
||||
|
||||
|
||||
here = os.path.dirname(os.path.abspath(__file__)) |
||||
|
||||
dict_loader = loaders.DictLoader({ |
||||
'justdict.html': 'FOO' |
||||
}) |
||||
package_loader = loaders.PackageLoader('jinja2.testsuite.res', 'templates') |
||||
filesystem_loader = loaders.FileSystemLoader(here + '/res/templates') |
||||
function_loader = loaders.FunctionLoader({'justfunction.html': 'FOO'}.get) |
||||
choice_loader = loaders.ChoiceLoader([dict_loader, package_loader]) |
||||
prefix_loader = loaders.PrefixLoader({ |
||||
'a': filesystem_loader, |
||||
'b': dict_loader |
||||
}) |
||||
|
||||
|
||||
class JinjaTestCase(unittest.TestCase): |
||||
|
||||
### use only these methods for testing. If you need standard |
||||
### unittest method, wrap them! |
||||
|
||||
def setup(self): |
||||
pass |
||||
|
||||
def teardown(self): |
||||
pass |
||||
|
||||
def setUp(self): |
||||
self.setup() |
||||
|
||||
def tearDown(self): |
||||
self.teardown() |
||||
|
||||
def assert_equal(self, a, b): |
||||
return self.assertEqual(a, b) |
||||
|
||||
def assert_raises(self, *args, **kwargs): |
||||
return self.assertRaises(*args, **kwargs) |
||||
|
||||
def assert_traceback_matches(self, callback, expected_tb): |
||||
try: |
||||
callback() |
||||
except Exception, e: |
||||
tb = format_exception(*sys.exc_info()) |
||||
if re.search(expected_tb.strip(), ''.join(tb)) is None: |
||||
raise self.fail('Traceback did not match:\n\n%s\nexpected:\n%s' |
||||
% (''.join(tb), expected_tb)) |
||||
else: |
||||
self.fail('Expected exception') |
||||
|
||||
|
||||
def suite(): |
||||
from jinja2.testsuite import ext, filters, tests, core_tags, \ |
||||
loader, inheritance, imports, lexnparse, security, api, \ |
||||
regression, debug, utils, doctests |
||||
suite = unittest.TestSuite() |
||||
suite.addTest(ext.suite()) |
||||
suite.addTest(filters.suite()) |
||||
suite.addTest(tests.suite()) |
||||
suite.addTest(core_tags.suite()) |
||||
suite.addTest(loader.suite()) |
||||
suite.addTest(inheritance.suite()) |
||||
suite.addTest(imports.suite()) |
||||
suite.addTest(lexnparse.suite()) |
||||
suite.addTest(security.suite()) |
||||
suite.addTest(api.suite()) |
||||
suite.addTest(regression.suite()) |
||||
suite.addTest(debug.suite()) |
||||
suite.addTest(utils.suite()) |
||||
|
||||
# doctests will not run on python 3 currently. Too many issues |
||||
# with that, do not test that on that platform. |
||||
if sys.version_info < (3, 0): |
||||
suite.addTest(doctests.suite()) |
||||
|
||||
return suite |
@ -1,245 +0,0 @@ |
||||
# -*- coding: utf-8 -*- |
||||
""" |
||||
jinja2.testsuite.api |
||||
~~~~~~~~~~~~~~~~~~~~ |
||||
|
||||
Tests the public API and related stuff. |
||||
|
||||
:copyright: (c) 2010 by the Jinja Team. |
||||
:license: BSD, see LICENSE for more details. |
||||
""" |
||||
import unittest |
||||
|
||||
from jinja2.testsuite import JinjaTestCase |
||||
|
||||
from jinja2 import Environment, Undefined, DebugUndefined, \ |
||||
StrictUndefined, UndefinedError, meta, \ |
||||
is_undefined, Template, DictLoader |
||||
from jinja2.utils import Cycler |
||||
|
||||
env = Environment() |
||||
|
||||
|
||||
class ExtendedAPITestCase(JinjaTestCase): |
||||
|
||||
def test_item_and_attribute(self): |
||||
from jinja2.sandbox import SandboxedEnvironment |
||||
|
||||
for env in Environment(), SandboxedEnvironment(): |
||||
# the |list is necessary for python3 |
||||
tmpl = env.from_string('{{ foo.items()|list }}') |
||||
assert tmpl.render(foo={'items': 42}) == "[('items', 42)]" |
||||
tmpl = env.from_string('{{ foo|attr("items")()|list }}') |
||||
assert tmpl.render(foo={'items': 42}) == "[('items', 42)]" |
||||
tmpl = env.from_string('{{ foo["items"] }}') |
||||
assert tmpl.render(foo={'items': 42}) == '42' |
||||
|
||||
def test_finalizer(self): |
||||
def finalize_none_empty(value): |
||||
if value is None: |
||||
value = u'' |
||||
return value |
||||
env = Environment(finalize=finalize_none_empty) |
||||
tmpl = env.from_string('{% for item in seq %}|{{ item }}{% endfor %}') |
||||
assert tmpl.render(seq=(None, 1, "foo")) == '||1|foo' |
||||
tmpl = env.from_string('<{{ none }}>') |
||||
assert tmpl.render() == '<>' |
||||
|
||||
def test_cycler(self): |
||||
items = 1, 2, 3 |
||||
c = Cycler(*items) |
||||
for item in items + items: |
||||
assert c.current == item |
||||
assert c.next() == item |
||||
c.next() |
||||
assert c.current == 2 |
||||
c.reset() |
||||
assert c.current == 1 |
||||
|
||||
def test_expressions(self): |
||||
expr = env.compile_expression("foo") |
||||
assert expr() is None |
||||
assert expr(foo=42) == 42 |
||||
expr2 = env.compile_expression("foo", undefined_to_none=False) |
||||
assert is_undefined(expr2()) |
||||
|
||||
expr = env.compile_expression("42 + foo") |
||||
assert expr(foo=42) == 84 |
||||
|
||||
def test_template_passthrough(self): |
||||
t = Template('Content') |
||||
assert env.get_template(t) is t |
||||
assert env.select_template([t]) is t |
||||
assert env.get_or_select_template([t]) is t |
||||
assert env.get_or_select_template(t) is t |
||||
|
||||
def test_autoescape_autoselect(self): |
||||
def select_autoescape(name): |
||||
if name is None or '.' not in name: |
||||
return False |
||||
return name.endswith('.html') |
||||
env = Environment(autoescape=select_autoescape, |
||||
loader=DictLoader({ |
||||
'test.txt': '{{ foo }}', |
||||
'test.html': '{{ foo }}' |
||||
})) |
||||
t = env.get_template('test.txt') |
||||
assert t.render(foo='<foo>') == '<foo>' |
||||
t = env.get_template('test.html') |
||||
assert t.render(foo='<foo>') == '<foo>' |
||||
t = env.from_string('{{ foo }}') |
||||
assert t.render(foo='<foo>') == '<foo>' |
||||
|
||||
|
||||
class MetaTestCase(JinjaTestCase): |
||||
|
||||
def test_find_undeclared_variables(self): |
||||
ast = env.parse('{% set foo = 42 %}{{ bar + foo }}') |
||||
x = meta.find_undeclared_variables(ast) |
||||
assert x == set(['bar']) |
||||
|
||||
ast = env.parse('{% set foo = 42 %}{{ bar + foo }}' |
||||
'{% macro meh(x) %}{{ x }}{% endmacro %}' |
||||
'{% for item in seq %}{{ muh(item) + meh(seq) }}{% endfor %}') |
||||
x = meta.find_undeclared_variables(ast) |
||||
assert x == set(['bar', 'seq', 'muh']) |
||||
|
||||
def test_find_refererenced_templates(self): |
||||
ast = env.parse('{% extends "layout.html" %}{% include helper %}') |
||||
i = meta.find_referenced_templates(ast) |
||||
assert i.next() == 'layout.html' |
||||
assert i.next() is None |
||||
assert list(i) == [] |
||||
|
||||
ast = env.parse('{% extends "layout.html" %}' |
||||
'{% from "test.html" import a, b as c %}' |
||||
'{% import "meh.html" as meh %}' |
||||
'{% include "muh.html" %}') |
||||
i = meta.find_referenced_templates(ast) |
||||
assert list(i) == ['layout.html', 'test.html', 'meh.html', 'muh.html'] |
||||
|
||||
def test_find_included_templates(self): |
||||
ast = env.parse('{% include ["foo.html", "bar.html"] %}') |
||||
i = meta.find_referenced_templates(ast) |
||||
assert list(i) == ['foo.html', 'bar.html'] |
||||
|
||||
ast = env.parse('{% include ("foo.html", "bar.html") %}') |
||||
i = meta.find_referenced_templates(ast) |
||||
assert list(i) == ['foo.html', 'bar.html'] |
||||
|
||||
ast = env.parse('{% include ["foo.html", "bar.html", foo] %}') |
||||
i = meta.find_referenced_templates(ast) |
||||
assert list(i) == ['foo.html', 'bar.html', None] |
||||
|
||||
ast = env.parse('{% include ("foo.html", "bar.html", foo) %}') |
||||
i = meta.find_referenced_templates(ast) |
||||
assert list(i) == ['foo.html', 'bar.html', None] |
||||
|
||||
|
||||
class StreamingTestCase(JinjaTestCase): |
||||
|
||||
def test_basic_streaming(self): |
||||
tmpl = env.from_string("<ul>{% for item in seq %}<li>{{ loop.index " |
||||
"}} - {{ item }}</li>{%- endfor %}</ul>") |
||||
stream = tmpl.stream(seq=range(4)) |
||||
self.assert_equal(stream.next(), '<ul>') |
||||
self.assert_equal(stream.next(), '<li>1 - 0</li>') |
||||
self.assert_equal(stream.next(), '<li>2 - 1</li>') |
||||
self.assert_equal(stream.next(), '<li>3 - 2</li>') |
||||
self.assert_equal(stream.next(), '<li>4 - 3</li>') |
||||
self.assert_equal(stream.next(), '</ul>') |
||||
|
||||
def test_buffered_streaming(self): |
||||
tmpl = env.from_string("<ul>{% for item in seq %}<li>{{ loop.index " |
||||
"}} - {{ item }}</li>{%- endfor %}</ul>") |
||||
stream = tmpl.stream(seq=range(4)) |
||||
stream.enable_buffering(size=3) |
||||
self.assert_equal(stream.next(), u'<ul><li>1 - 0</li><li>2 - 1</li>') |
||||
self.assert_equal(stream.next(), u'<li>3 - 2</li><li>4 - 3</li></ul>') |
||||
|
||||
def test_streaming_behavior(self): |
||||
tmpl = env.from_string("") |
||||
stream = tmpl.stream() |
||||
assert not stream.buffered |
||||
stream.enable_buffering(20) |
||||
assert stream.buffered |
||||
stream.disable_buffering() |
||||
assert not stream.buffered |
||||
|
||||
|
||||
class UndefinedTestCase(JinjaTestCase): |
||||
|
||||
def test_stopiteration_is_undefined(self): |
||||
def test(): |
||||
raise StopIteration() |
||||
t = Template('A{{ test() }}B') |
||||
assert t.render(test=test) == 'AB' |
||||
t = Template('A{{ test().missingattribute }}B') |
||||
self.assert_raises(UndefinedError, t.render, test=test) |
||||
|
||||
def test_undefined_and_special_attributes(self): |
||||
try: |
||||
Undefined('Foo').__dict__ |
||||
except AttributeError: |
||||
pass |
||||
else: |
||||
assert False, "Expected actual attribute error" |
||||
|
||||
def test_default_undefined(self): |
||||
env = Environment(undefined=Undefined) |
||||
self.assert_equal(env.from_string('{{ missing }}').render(), u'') |
||||
self.assert_raises(UndefinedError, |
||||
env.from_string('{{ missing.attribute }}').render) |
||||
self.assert_equal(env.from_string('{{ missing|list }}').render(), '[]') |
||||
self.assert_equal(env.from_string('{{ missing is not defined }}').render(), 'True') |
||||
self.assert_equal(env.from_string('{{ foo.missing }}').render(foo=42), '') |
||||
self.assert_equal(env.from_string('{{ not missing }}').render(), 'True') |
||||
|
||||
def test_debug_undefined(self): |
||||
env = Environment(undefined=DebugUndefined) |
||||
self.assert_equal(env.from_string('{{ missing }}').render(), '{{ missing }}') |
||||
self.assert_raises(UndefinedError, |
||||
env.from_string('{{ missing.attribute }}').render) |
||||
self.assert_equal(env.from_string('{{ missing|list }}').render(), '[]') |
||||
self.assert_equal(env.from_string('{{ missing is not defined }}').render(), 'True') |
||||
self.assert_equal(env.from_string('{{ foo.missing }}').render(foo=42), |
||||
u"{{ no such element: int object['missing'] }}") |
||||
self.assert_equal(env.from_string('{{ not missing }}').render(), 'True') |
||||
|
||||
def test_strict_undefined(self): |
||||
env = Environment(undefined=StrictUndefined) |
||||
self.assert_raises(UndefinedError, env.from_string('{{ missing }}').render) |
||||
self.assert_raises(UndefinedError, env.from_string('{{ missing.attribute }}').render) |
||||
self.assert_raises(UndefinedError, env.from_string('{{ missing|list }}').render) |
||||
self.assert_equal(env.from_string('{{ missing is not defined }}').render(), 'True') |
||||
self.assert_raises(UndefinedError, env.from_string('{{ foo.missing }}').render, foo=42) |
||||
self.assert_raises(UndefinedError, env.from_string('{{ not missing }}').render) |
||||
|
||||
def test_indexing_gives_undefined(self): |
||||
t = Template("{{ var[42].foo }}") |
||||
self.assert_raises(UndefinedError, t.render, var=0) |
||||
|
||||
def test_none_gives_proper_error(self): |
||||
try: |
||||
Environment().getattr(None, 'split')() |
||||
except UndefinedError, e: |
||||
assert e.message == "'None' has no attribute 'split'" |
||||
else: |
||||
assert False, 'expected exception' |
||||
|
||||
def test_object_repr(self): |
||||
try: |
||||
Undefined(obj=42, name='upper')() |
||||
except UndefinedError, e: |
||||
assert e.message == "'int object' has no attribute 'upper'" |
||||
else: |
||||
assert False, 'expected exception' |
||||
|
||||
|
||||
def suite(): |
||||
suite = unittest.TestSuite() |
||||
suite.addTest(unittest.makeSuite(ExtendedAPITestCase)) |
||||
suite.addTest(unittest.makeSuite(MetaTestCase)) |
||||
suite.addTest(unittest.makeSuite(StreamingTestCase)) |
||||
suite.addTest(unittest.makeSuite(UndefinedTestCase)) |
||||
return suite |
@ -1,285 +0,0 @@ |
||||
# -*- coding: utf-8 -*- |
||||
""" |
||||
jinja2.testsuite.core_tags |
||||
~~~~~~~~~~~~~~~~~~~~~~~~~~ |
||||
|
||||
Test the core tags like for and if. |
||||
|
||||
:copyright: (c) 2010 by the Jinja Team. |
||||
:license: BSD, see LICENSE for more details. |
||||
""" |
||||
import unittest |
||||
|
||||
from jinja2.testsuite import JinjaTestCase |
||||
|
||||
from jinja2 import Environment, TemplateSyntaxError, UndefinedError, \ |
||||
DictLoader |
||||
|
||||
env = Environment() |
||||
|
||||
|
||||
class ForLoopTestCase(JinjaTestCase): |
||||
|
||||
def test_simple(self): |
||||
tmpl = env.from_string('{% for item in seq %}{{ item }}{% endfor %}') |
||||
assert tmpl.render(seq=range(10)) == '0123456789' |
||||
|
||||
def test_else(self): |
||||
tmpl = env.from_string('{% for item in seq %}XXX{% else %}...{% endfor %}') |
||||
assert tmpl.render() == '...' |
||||
|
||||
def test_empty_blocks(self): |
||||
tmpl = env.from_string('<{% for item in seq %}{% else %}{% endfor %}>') |
||||
assert tmpl.render() == '<>' |
||||
|
||||
def test_context_vars(self): |
||||
tmpl = env.from_string('''{% for item in seq -%} |
||||
{{ loop.index }}|{{ loop.index0 }}|{{ loop.revindex }}|{{ |
||||
loop.revindex0 }}|{{ loop.first }}|{{ loop.last }}|{{ |
||||
loop.length }}###{% endfor %}''') |
||||
one, two, _ = tmpl.render(seq=[0, 1]).split('###') |
||||
(one_index, one_index0, one_revindex, one_revindex0, one_first, |
||||
one_last, one_length) = one.split('|') |
||||
(two_index, two_index0, two_revindex, two_revindex0, two_first, |
||||
two_last, two_length) = two.split('|') |
||||
|
||||
assert int(one_index) == 1 and int(two_index) == 2 |
||||
assert int(one_index0) == 0 and int(two_index0) == 1 |
||||
assert int(one_revindex) == 2 and int(two_revindex) == 1 |
||||
assert int(one_revindex0) == 1 and int(two_revindex0) == 0 |
||||
assert one_first == 'True' and two_first == 'False' |
||||
assert one_last == 'False' and two_last == 'True' |
||||
assert one_length == two_length == '2' |
||||
|
||||
def test_cycling(self): |
||||
tmpl = env.from_string('''{% for item in seq %}{{ |
||||
loop.cycle('<1>', '<2>') }}{% endfor %}{% |
||||
for item in seq %}{{ loop.cycle(*through) }}{% endfor %}''') |
||||
output = tmpl.render(seq=range(4), through=('<1>', '<2>')) |
||||
assert output == '<1><2>' * 4 |
||||
|
||||
def test_scope(self): |
||||
tmpl = env.from_string('{% for item in seq %}{% endfor %}{{ item }}') |
||||
output = tmpl.render(seq=range(10)) |
||||
assert not output |
||||
|
||||
def test_varlen(self): |
||||
def inner(): |
||||
for item in range(5): |
||||
yield item |
||||
tmpl = env.from_string('{% for item in iter %}{{ item }}{% endfor %}') |
||||
output = tmpl.render(iter=inner()) |
||||
assert output == '01234' |
||||
|
||||
def test_noniter(self): |
||||
tmpl = env.from_string('{% for item in none %}...{% endfor %}') |
||||
self.assert_raises(TypeError, tmpl.render) |
||||
|
||||
def test_recursive(self): |
||||
tmpl = env.from_string('''{% for item in seq recursive -%} |
||||
[{{ item.a }}{% if item.b %}<{{ loop(item.b) }}>{% endif %}] |
||||
{%- endfor %}''') |
||||
assert tmpl.render(seq=[ |
||||
dict(a=1, b=[dict(a=1), dict(a=2)]), |
||||
dict(a=2, b=[dict(a=1), dict(a=2)]), |
||||
dict(a=3, b=[dict(a='a')]) |
||||
]) == '[1<[1][2]>][2<[1][2]>][3<[a]>]' |
||||
|
||||
def test_looploop(self): |
||||
tmpl = env.from_string('''{% for row in table %} |
||||
{%- set rowloop = loop -%} |
||||
{% for cell in row -%} |
||||
[{{ rowloop.index }}|{{ loop.index }}] |
||||
{%- endfor %} |
||||
{%- endfor %}''') |
||||
assert tmpl.render(table=['ab', 'cd']) == '[1|1][1|2][2|1][2|2]' |
||||
|
||||
def test_reversed_bug(self): |
||||
tmpl = env.from_string('{% for i in items %}{{ i }}' |
||||
'{% if not loop.last %}' |
||||
',{% endif %}{% endfor %}') |
||||
assert tmpl.render(items=reversed([3, 2, 1])) == '1,2,3' |
||||
|
||||
def test_loop_errors(self): |
||||
tmpl = env.from_string('''{% for item in [1] if loop.index |
||||
== 0 %}...{% endfor %}''') |
||||
self.assert_raises(UndefinedError, tmpl.render) |
||||
tmpl = env.from_string('''{% for item in [] %}...{% else |
||||
%}{{ loop }}{% endfor %}''') |
||||
assert tmpl.render() == '' |
||||
|
||||
def test_loop_filter(self): |
||||
tmpl = env.from_string('{% for item in range(10) if item ' |
||||
'is even %}[{{ item }}]{% endfor %}') |
||||
assert tmpl.render() == '[0][2][4][6][8]' |
||||
tmpl = env.from_string(''' |
||||
{%- for item in range(10) if item is even %}[{{ |
||||
loop.index }}:{{ item }}]{% endfor %}''') |
||||
assert tmpl.render() == '[1:0][2:2][3:4][4:6][5:8]' |
||||
|
||||
def test_loop_unassignable(self): |
||||
self.assert_raises(TemplateSyntaxError, env.from_string, |
||||
'{% for loop in seq %}...{% endfor %}') |
||||
|
||||
def test_scoped_special_var(self): |
||||
t = env.from_string('{% for s in seq %}[{{ loop.first }}{% for c in s %}' |
||||
'|{{ loop.first }}{% endfor %}]{% endfor %}') |
||||
assert t.render(seq=('ab', 'cd')) == '[True|True|False][False|True|False]' |
||||
|
||||
def test_scoped_loop_var(self): |
||||
t = env.from_string('{% for x in seq %}{{ loop.first }}' |
||||
'{% for y in seq %}{% endfor %}{% endfor %}') |
||||
assert t.render(seq='ab') == 'TrueFalse' |
||||
t = env.from_string('{% for x in seq %}{% for y in seq %}' |
||||
'{{ loop.first }}{% endfor %}{% endfor %}') |
||||
assert t.render(seq='ab') == 'TrueFalseTrueFalse' |
||||
|
||||
def test_recursive_empty_loop_iter(self): |
||||
t = env.from_string(''' |
||||
{%- for item in foo recursive -%}{%- endfor -%} |
||||
''') |
||||
assert t.render(dict(foo=[])) == '' |
||||
|
||||
def test_call_in_loop(self): |
||||
t = env.from_string(''' |
||||
{%- macro do_something() -%} |
||||
[{{ caller() }}] |
||||
{%- endmacro %} |
||||
|
||||
{%- for i in [1, 2, 3] %} |
||||
{%- call do_something() -%} |
||||
{{ i }} |
||||
{%- endcall %} |
||||
{%- endfor -%} |
||||
''') |
||||
assert t.render() == '[1][2][3]' |
||||
|
||||
def test_scoping_bug(self): |
||||
t = env.from_string(''' |
||||
{%- for item in foo %}...{{ item }}...{% endfor %} |
||||
{%- macro item(a) %}...{{ a }}...{% endmacro %} |
||||
{{- item(2) -}} |
||||
''') |
||||
assert t.render(foo=(1,)) == '...1......2...' |
||||
|
||||
def test_unpacking(self): |
||||
tmpl = env.from_string('{% for a, b, c in [[1, 2, 3]] %}' |
||||
'{{ a }}|{{ b }}|{{ c }}{% endfor %}') |
||||
assert tmpl.render() == '1|2|3' |
||||
|
||||
|
||||
class IfConditionTestCase(JinjaTestCase): |
||||
|
||||
def test_simple(self): |
||||
tmpl = env.from_string('''{% if true %}...{% endif %}''') |
||||
assert tmpl.render() == '...' |
||||
|
||||
def test_elif(self): |
||||
tmpl = env.from_string('''{% if false %}XXX{% elif true |
||||
%}...{% else %}XXX{% endif %}''') |
||||
assert tmpl.render() == '...' |
||||
|
||||
def test_else(self): |
||||
tmpl = env.from_string('{% if false %}XXX{% else %}...{% endif %}') |
||||
assert tmpl.render() == '...' |
||||
|
||||
def test_empty(self): |
||||
tmpl = env.from_string('[{% if true %}{% else %}{% endif %}]') |
||||
assert tmpl.render() == '[]' |
||||
|
||||
def test_complete(self): |
||||
tmpl = env.from_string('{% if a %}A{% elif b %}B{% elif c == d %}' |
||||
'C{% else %}D{% endif %}') |
||||
assert tmpl.render(a=0, b=False, c=42, d=42.0) == 'C' |
||||
|
||||
def test_no_scope(self): |
||||
tmpl = env.from_string('{% if a %}{% set foo = 1 %}{% endif %}{{ foo }}') |
||||
assert tmpl.render(a=True) == '1' |
||||
tmpl = env.from_string('{% if true %}{% set foo = 1 %}{% endif %}{{ foo }}') |
||||
assert tmpl.render() == '1' |
||||
|
||||
|
||||
class MacrosTestCase(JinjaTestCase): |
||||
env = Environment(trim_blocks=True) |
||||
|
||||
def test_simple(self): |
||||
tmpl = self.env.from_string('''\ |
||||
{% macro say_hello(name) %}Hello {{ name }}!{% endmacro %} |
||||
{{ say_hello('Peter') }}''') |
||||
assert tmpl.render() == 'Hello Peter!' |
||||
|
||||
def test_scoping(self): |
||||
tmpl = self.env.from_string('''\ |
||||
{% macro level1(data1) %} |
||||
{% macro level2(data2) %}{{ data1 }}|{{ data2 }}{% endmacro %} |
||||
{{ level2('bar') }}{% endmacro %} |
||||
{{ level1('foo') }}''') |
||||
assert tmpl.render() == 'foo|bar' |
||||
|
||||
def test_arguments(self): |
||||
tmpl = self.env.from_string('''\ |
||||
{% macro m(a, b, c='c', d='d') %}{{ a }}|{{ b }}|{{ c }}|{{ d }}{% endmacro %} |
||||
{{ m() }}|{{ m('a') }}|{{ m('a', 'b') }}|{{ m(1, 2, 3) }}''') |
||||
assert tmpl.render() == '||c|d|a||c|d|a|b|c|d|1|2|3|d' |
||||
|
||||
def test_varargs(self): |
||||
tmpl = self.env.from_string('''\ |
||||
{% macro test() %}{{ varargs|join('|') }}{% endmacro %}\ |
||||
{{ test(1, 2, 3) }}''') |
||||
assert tmpl.render() == '1|2|3' |
||||
|
||||
def test_simple_call(self): |
||||
tmpl = self.env.from_string('''\ |
||||
{% macro test() %}[[{{ caller() }}]]{% endmacro %}\ |
||||
{% call test() %}data{% endcall %}''') |
||||
assert tmpl.render() == '[[data]]' |
||||
|
||||
def test_complex_call(self): |
||||
tmpl = self.env.from_string('''\ |
||||
{% macro test() %}[[{{ caller('data') }}]]{% endmacro %}\ |
||||
{% call(data) test() %}{{ data }}{% endcall %}''') |
||||
assert tmpl.render() == '[[data]]' |
||||
|
||||
def test_caller_undefined(self): |
||||
tmpl = self.env.from_string('''\ |
||||
{% set caller = 42 %}\ |
||||
{% macro test() %}{{ caller is not defined }}{% endmacro %}\ |
||||
{{ test() }}''') |
||||
assert tmpl.render() == 'True' |
||||
|
||||
def test_include(self): |
||||
self.env = Environment(loader=DictLoader({'include': |
||||
'{% macro test(foo) %}[{{ foo }}]{% endmacro %}'})) |
||||
tmpl = self.env.from_string('{% from "include" import test %}{{ test("foo") }}') |
||||
assert tmpl.render() == '[foo]' |
||||
|
||||
def test_macro_api(self): |
||||
tmpl = self.env.from_string('{% macro foo(a, b) %}{% endmacro %}' |
||||
'{% macro bar() %}{{ varargs }}{{ kwargs }}{% endmacro %}' |
||||
'{% macro baz() %}{{ caller() }}{% endmacro %}') |
||||
assert tmpl.module.foo.arguments == ('a', 'b') |
||||
assert tmpl.module.foo.defaults == () |
||||
assert tmpl.module.foo.name == 'foo' |
||||
assert not tmpl.module.foo.caller |
||||
assert not tmpl.module.foo.catch_kwargs |
||||
assert not tmpl.module.foo.catch_varargs |
||||
assert tmpl.module.bar.arguments == () |
||||
assert tmpl.module.bar.defaults == () |
||||
assert not tmpl.module.bar.caller |
||||
assert tmpl.module.bar.catch_kwargs |
||||
assert tmpl.module.bar.catch_varargs |
||||
assert tmpl.module.baz.caller |
||||
|
||||
def test_callself(self): |
||||
tmpl = self.env.from_string('{% macro foo(x) %}{{ x }}{% if x > 1 %}|' |
||||
'{{ foo(x - 1) }}{% endif %}{% endmacro %}' |
||||
'{{ foo(5) }}') |
||||
assert tmpl.render() == '5|4|3|2|1' |
||||
|
||||
|
||||
def suite(): |
||||
suite = unittest.TestSuite() |
||||
suite.addTest(unittest.makeSuite(ForLoopTestCase)) |
||||
suite.addTest(unittest.makeSuite(IfConditionTestCase)) |
||||
suite.addTest(unittest.makeSuite(MacrosTestCase)) |
||||
return suite |
@ -1,60 +0,0 @@ |
||||
# -*- coding: utf-8 -*- |
||||
""" |
||||
jinja2.testsuite.debug |
||||
~~~~~~~~~~~~~~~~~~~~~~ |
||||
|
||||
Tests the debug system. |
||||
|
||||
:copyright: (c) 2010 by the Jinja Team. |
||||
:license: BSD, see LICENSE for more details. |
||||
""" |
||||
import sys |
||||
import unittest |
||||
|
||||
from jinja2.testsuite import JinjaTestCase, filesystem_loader |
||||
|
||||
from jinja2 import Environment, TemplateSyntaxError |
||||
|
||||
env = Environment(loader=filesystem_loader) |
||||
|
||||
|
||||
class DebugTestCase(JinjaTestCase): |
||||
|
||||
if sys.version_info[:2] != (2, 4): |
||||
def test_runtime_error(self): |
||||
def test(): |
||||
tmpl.render(fail=lambda: 1 / 0) |
||||
tmpl = env.get_template('broken.html') |
||||
self.assert_traceback_matches(test, r''' |
||||
File ".*?broken.html", line 2, in (top-level template code|<module>) |
||||
\{\{ fail\(\) \}\} |
||||
File ".*?debug.pyc?", line \d+, in <lambda> |
||||
tmpl\.render\(fail=lambda: 1 / 0\) |
||||
ZeroDivisionError: (int(eger)? )?division (or modulo )?by zero |
||||
''') |
||||
|
||||
def test_syntax_error(self): |
||||
# XXX: the .*? is necessary for python3 which does not hide |
||||
# some of the stack frames we don't want to show. Not sure |
||||
# what's up with that, but that is not that critical. Should |
||||
# be fixed though. |
||||
self.assert_traceback_matches(lambda: env.get_template('syntaxerror.html'), r'''(?sm) |
||||
File ".*?syntaxerror.html", line 4, in (template|<module>) |
||||
\{% endif %\}.*? |
||||
(jinja2\.exceptions\.)?TemplateSyntaxError: Encountered unknown tag 'endif'. Jinja was looking for the following tags: 'endfor' or 'else'. The innermost block that needs to be closed is 'for'. |
||||
''') |
||||
|
||||
def test_regular_syntax_error(self): |
||||
def test(): |
||||
raise TemplateSyntaxError('wtf', 42) |
||||
self.assert_traceback_matches(test, r''' |
||||
File ".*debug.pyc?", line \d+, in test |
||||
raise TemplateSyntaxError\('wtf', 42\) |
||||
(jinja2\.exceptions\.)?TemplateSyntaxError: wtf |
||||
line 42''') |
||||
|
||||
|
||||
def suite(): |
||||
suite = unittest.TestSuite() |
||||
suite.addTest(unittest.makeSuite(DebugTestCase)) |
||||
return suite |
@ -1,29 +0,0 @@ |
||||
# -*- coding: utf-8 -*- |
||||
""" |
||||
jinja2.testsuite.doctests |
||||
~~~~~~~~~~~~~~~~~~~~~~~~~ |
||||
|
||||
The doctests. Collects all tests we want to test from |
||||
the Jinja modules. |
||||
|
||||
:copyright: (c) 2010 by the Jinja Team. |
||||
:license: BSD, see LICENSE for more details. |
||||
""" |
||||
import unittest |
||||
import doctest |
||||
|
||||
|
||||
def suite(): |
||||
from jinja2 import utils, sandbox, runtime, meta, loaders, \ |
||||
ext, environment, bccache, nodes |
||||
suite = unittest.TestSuite() |
||||
suite.addTest(doctest.DocTestSuite(utils)) |
||||
suite.addTest(doctest.DocTestSuite(sandbox)) |
||||
suite.addTest(doctest.DocTestSuite(runtime)) |
||||
suite.addTest(doctest.DocTestSuite(meta)) |
||||
suite.addTest(doctest.DocTestSuite(loaders)) |
||||
suite.addTest(doctest.DocTestSuite(ext)) |
||||
suite.addTest(doctest.DocTestSuite(environment)) |
||||
suite.addTest(doctest.DocTestSuite(bccache)) |
||||
suite.addTest(doctest.DocTestSuite(nodes)) |
||||
return suite |
@ -1,455 +0,0 @@ |
||||
# -*- coding: utf-8 -*- |
||||
""" |
||||
jinja2.testsuite.ext |
||||
~~~~~~~~~~~~~~~~~~~~ |
||||
|
||||
Tests for the extensions. |
||||
|
||||
:copyright: (c) 2010 by the Jinja Team. |
||||
:license: BSD, see LICENSE for more details. |
||||
""" |
||||
import re |
||||
import unittest |
||||
|
||||
from jinja2.testsuite import JinjaTestCase |
||||
|
||||
from jinja2 import Environment, DictLoader, contextfunction, nodes |
||||
from jinja2.exceptions import TemplateAssertionError |
||||
from jinja2.ext import Extension |
||||
from jinja2.lexer import Token, count_newlines |
||||
from jinja2.utils import next |
||||
|
||||
# 2.x / 3.x |
||||
try: |
||||
from io import BytesIO |
||||
except ImportError: |
||||
from StringIO import StringIO as BytesIO |
||||
|
||||
|
||||
importable_object = 23 |
||||
|
||||
_gettext_re = re.compile(r'_\((.*?)\)(?s)') |
||||
|
||||
|
||||
i18n_templates = { |
||||
'master.html': '<title>{{ page_title|default(_("missing")) }}</title>' |
||||
'{% block body %}{% endblock %}', |
||||
'child.html': '{% extends "master.html" %}{% block body %}' |
||||
'{% trans %}watch out{% endtrans %}{% endblock %}', |
||||
'plural.html': '{% trans user_count %}One user online{% pluralize %}' |
||||
'{{ user_count }} users online{% endtrans %}', |
||||
'stringformat.html': '{{ _("User: %(num)s")|format(num=user_count) }}' |
||||
} |
||||
|
||||
newstyle_i18n_templates = { |
||||
'master.html': '<title>{{ page_title|default(_("missing")) }}</title>' |
||||
'{% block body %}{% endblock %}', |
||||
'child.html': '{% extends "master.html" %}{% block body %}' |
||||
'{% trans %}watch out{% endtrans %}{% endblock %}', |
||||
'plural.html': '{% trans user_count %}One user online{% pluralize %}' |
||||
'{{ user_count }} users online{% endtrans %}', |
||||
'stringformat.html': '{{ _("User: %(num)s", num=user_count) }}', |
||||
'ngettext.html': '{{ ngettext("%(num)s apple", "%(num)s apples", apples) }}', |
||||
'ngettext_long.html': '{% trans num=apples %}{{ num }} apple{% pluralize %}' |
||||
'{{ num }} apples{% endtrans %}', |
||||
'transvars1.html': '{% trans %}User: {{ num }}{% endtrans %}', |
||||
'transvars2.html': '{% trans num=count %}User: {{ num }}{% endtrans %}', |
||||
'transvars3.html': '{% trans count=num %}User: {{ count }}{% endtrans %}', |
||||
'novars.html': '{% trans %}%(hello)s{% endtrans %}', |
||||
'vars.html': '{% trans %}{{ foo }}%(foo)s{% endtrans %}', |
||||
'explicitvars.html': '{% trans foo="42" %}%(foo)s{% endtrans %}' |
||||
} |
||||
|
||||
|
||||
languages = { |
||||
'de': { |
||||
'missing': u'fehlend', |
||||
'watch out': u'pass auf', |
||||
'One user online': u'Ein Benutzer online', |
||||
'%(user_count)s users online': u'%(user_count)s Benutzer online', |
||||
'User: %(num)s': u'Benutzer: %(num)s', |
||||
'User: %(count)s': u'Benutzer: %(count)s', |
||||
'%(num)s apple': u'%(num)s Apfel', |
||||
'%(num)s apples': u'%(num)s Äpfel' |
||||
} |
||||
} |
||||
|
||||
|
||||
@contextfunction |
||||
def gettext(context, string): |
||||
language = context.get('LANGUAGE', 'en') |
||||
return languages.get(language, {}).get(string, string) |
||||
|
||||
|
||||
@contextfunction |
||||
def ngettext(context, s, p, n): |
||||
language = context.get('LANGUAGE', 'en') |
||||
if n != 1: |
||||
return languages.get(language, {}).get(p, p) |
||||
return languages.get(language, {}).get(s, s) |
||||
|
||||
|
||||
i18n_env = Environment( |
||||
loader=DictLoader(i18n_templates), |
||||
extensions=['jinja2.ext.i18n'] |
||||
) |
||||
i18n_env.globals.update({ |
||||
'_': gettext, |
||||
'gettext': gettext, |
||||
'ngettext': ngettext |
||||
}) |
||||
|
||||
newstyle_i18n_env = Environment( |
||||
loader=DictLoader(newstyle_i18n_templates), |
||||
extensions=['jinja2.ext.i18n'] |
||||
) |
||||
newstyle_i18n_env.install_gettext_callables(gettext, ngettext, newstyle=True) |
||||
|
||||
class TestExtension(Extension): |
||||
tags = set(['test']) |
||||
ext_attr = 42 |
||||
|
||||
def parse(self, parser): |
||||
return nodes.Output([self.call_method('_dump', [ |
||||
nodes.EnvironmentAttribute('sandboxed'), |
||||
self.attr('ext_attr'), |
||||
nodes.ImportedName(__name__ + '.importable_object'), |
||||
nodes.ContextReference() |
||||
])]).set_lineno(next(parser.stream).lineno) |
||||
|
||||
def _dump(self, sandboxed, ext_attr, imported_object, context): |
||||
return '%s|%s|%s|%s' % ( |
||||
sandboxed, |
||||
ext_attr, |
||||
imported_object, |
||||
context.blocks |
||||
) |
||||
|
||||
|
||||
class PreprocessorExtension(Extension): |
||||
|
||||
def preprocess(self, source, name, filename=None): |
||||
return source.replace('[[TEST]]', '({{ foo }})') |
||||
|
||||
|
||||
class StreamFilterExtension(Extension): |
||||
|
||||
def filter_stream(self, stream): |
||||
for token in stream: |
||||
if token.type == 'data': |
||||
for t in self.interpolate(token): |
||||
yield t |
||||
else: |
||||
yield token |
||||
|
||||
def interpolate(self, token): |
||||
pos = 0 |
||||
end = len(token.value) |
||||
lineno = token.lineno |
||||
while 1: |
||||
match = _gettext_re.search(token.value, pos) |
||||
if match is None: |
||||
break |
||||
value = token.value[pos:match.start()] |
||||
if value: |
||||
yield Token(lineno, 'data', value) |
||||
lineno += count_newlines(token.value) |
||||
yield Token(lineno, 'variable_begin', None) |
||||
yield Token(lineno, 'name', 'gettext') |
||||
yield Token(lineno, 'lparen', None) |
||||
yield Token(lineno, 'string', match.group(1)) |
||||
yield Token(lineno, 'rparen', None) |
||||
yield Token(lineno, 'variable_end', None) |
||||
pos = match.end() |
||||
if pos < end: |
||||
yield Token(lineno, 'data', token.value[pos:]) |
||||
|
||||
|
||||
class ExtensionsTestCase(JinjaTestCase): |
||||
|
||||
def test_extend_late(self): |
||||
env = Environment() |
||||
env.add_extension('jinja2.ext.autoescape') |
||||
t = env.from_string('{% autoescape true %}{{ "<test>" }}{% endautoescape %}') |
||||
assert t.render() == '<test>' |
||||
|
||||
def test_loop_controls(self): |
||||
env = Environment(extensions=['jinja2.ext.loopcontrols']) |
||||
|
||||
tmpl = env.from_string(''' |
||||
{%- for item in [1, 2, 3, 4] %} |
||||
{%- if item % 2 == 0 %}{% continue %}{% endif -%} |
||||
{{ item }} |
||||
{%- endfor %}''') |
||||
assert tmpl.render() == '13' |
||||
|
||||
tmpl = env.from_string(''' |
||||
{%- for item in [1, 2, 3, 4] %} |
||||
{%- if item > 2 %}{% break %}{% endif -%} |
||||
{{ item }} |
||||
{%- endfor %}''') |
||||
assert tmpl.render() == '12' |
||||
|
||||
def test_do(self): |
||||
env = Environment(extensions=['jinja2.ext.do']) |
||||
tmpl = env.from_string(''' |
||||
{%- set items = [] %} |
||||
{%- for char in "foo" %} |
||||
{%- do items.append(loop.index0 ~ char) %} |
||||
{%- endfor %}{{ items|join(', ') }}''') |
||||
assert tmpl.render() == '0f, 1o, 2o' |
||||
|
||||
def test_with(self): |
||||
env = Environment(extensions=['jinja2.ext.with_']) |
||||
tmpl = env.from_string('''\ |
||||
{% with a=42, b=23 -%} |
||||
{{ a }} = {{ b }} |
||||
{% endwith -%} |
||||
{{ a }} = {{ b }}\ |
||||
''') |
||||
assert [x.strip() for x in tmpl.render(a=1, b=2).splitlines()] \ |
||||
== ['42 = 23', '1 = 2'] |
||||
|
||||
def test_extension_nodes(self): |
||||
env = Environment(extensions=[TestExtension]) |
||||
tmpl = env.from_string('{% test %}') |
||||
assert tmpl.render() == 'False|42|23|{}' |
||||
|
||||
def test_identifier(self): |
||||
assert TestExtension.identifier == __name__ + '.TestExtension' |
||||
|
||||
def test_rebinding(self): |
||||
original = Environment(extensions=[TestExtension]) |
||||
overlay = original.overlay() |
||||
for env in original, overlay: |
||||
for ext in env.extensions.itervalues(): |
||||
assert ext.environment is env |
||||
|
||||
def test_preprocessor_extension(self): |
||||
env = Environment(extensions=[PreprocessorExtension]) |
||||
tmpl = env.from_string('{[[TEST]]}') |
||||
assert tmpl.render(foo=42) == '{(42)}' |
||||
|
||||
def test_streamfilter_extension(self): |
||||
env = Environment(extensions=[StreamFilterExtension]) |
||||
env.globals['gettext'] = lambda x: x.upper() |
||||
tmpl = env.from_string('Foo _(bar) Baz') |
||||
out = tmpl.render() |
||||
assert out == 'Foo BAR Baz' |
||||
|
||||
def test_extension_ordering(self): |
||||
class T1(Extension): |
||||
priority = 1 |
||||
class T2(Extension): |
||||
priority = 2 |
||||
env = Environment(extensions=[T1, T2]) |
||||
ext = list(env.iter_extensions()) |
||||
assert ext[0].__class__ is T1 |
||||
assert ext[1].__class__ is T2 |
||||
|
||||
|
||||
class InternationalizationTestCase(JinjaTestCase): |
||||
|
||||
def test_trans(self): |
||||
tmpl = i18n_env.get_template('child.html') |
||||
assert tmpl.render(LANGUAGE='de') == '<title>fehlend</title>pass auf' |
||||
|
||||
def test_trans_plural(self): |
||||
tmpl = i18n_env.get_template('plural.html') |
||||
assert tmpl.render(LANGUAGE='de', user_count=1) == 'Ein Benutzer online' |
||||
assert tmpl.render(LANGUAGE='de', user_count=2) == '2 Benutzer online' |
||||
|
||||
def test_complex_plural(self): |
||||
tmpl = i18n_env.from_string('{% trans foo=42, count=2 %}{{ count }} item{% ' |
||||
'pluralize count %}{{ count }} items{% endtrans %}') |
||||
assert tmpl.render() == '2 items' |
||||
self.assert_raises(TemplateAssertionError, i18n_env.from_string, |
||||
'{% trans foo %}...{% pluralize bar %}...{% endtrans %}') |
||||
|
||||
def test_trans_stringformatting(self): |
||||
tmpl = i18n_env.get_template('stringformat.html') |
||||
assert tmpl.render(LANGUAGE='de', user_count=5) == 'Benutzer: 5' |
||||
|
||||
def test_extract(self): |
||||
from jinja2.ext import babel_extract |
||||
source = BytesIO(''' |
||||
{{ gettext('Hello World') }} |
||||
{% trans %}Hello World{% endtrans %} |
||||
{% trans %}{{ users }} user{% pluralize %}{{ users }} users{% endtrans %} |
||||
'''.encode('ascii')) # make python 3 happy |
||||
assert list(babel_extract(source, ('gettext', 'ngettext', '_'), [], {})) == [ |
||||
(2, 'gettext', u'Hello World', []), |
||||
(3, 'gettext', u'Hello World', []), |
||||
(4, 'ngettext', (u'%(users)s user', u'%(users)s users', None), []) |
||||
] |
||||
|
||||
def test_comment_extract(self): |
||||
from jinja2.ext import babel_extract |
||||
source = BytesIO(''' |
||||
{# trans first #} |
||||
{{ gettext('Hello World') }} |
||||
{% trans %}Hello World{% endtrans %}{# trans second #} |
||||
{#: third #} |
||||
{% trans %}{{ users }} user{% pluralize %}{{ users }} users{% endtrans %} |
||||
'''.encode('utf-8')) # make python 3 happy |
||||
assert list(babel_extract(source, ('gettext', 'ngettext', '_'), ['trans', ':'], {})) == [ |
||||
(3, 'gettext', u'Hello World', ['first']), |
||||
(4, 'gettext', u'Hello World', ['second']), |
||||
(6, 'ngettext', (u'%(users)s user', u'%(users)s users', None), ['third']) |
||||
] |
||||
|
||||
|
||||
class NewstyleInternationalizationTestCase(JinjaTestCase): |
||||
|
||||
def test_trans(self): |
||||
tmpl = newstyle_i18n_env.get_template('child.html') |
||||
assert tmpl.render(LANGUAGE='de') == '<title>fehlend</title>pass auf' |
||||
|
||||
def test_trans_plural(self): |
||||
tmpl = newstyle_i18n_env.get_template('plural.html') |
||||
assert tmpl.render(LANGUAGE='de', user_count=1) == 'Ein Benutzer online' |
||||
assert tmpl.render(LANGUAGE='de', user_count=2) == '2 Benutzer online' |
||||
|
||||
def test_complex_plural(self): |
||||
tmpl = newstyle_i18n_env.from_string('{% trans foo=42, count=2 %}{{ count }} item{% ' |
||||
'pluralize count %}{{ count }} items{% endtrans %}') |
||||
assert tmpl.render() == '2 items' |
||||
self.assert_raises(TemplateAssertionError, i18n_env.from_string, |
||||
'{% trans foo %}...{% pluralize bar %}...{% endtrans %}') |
||||
|
||||
def test_trans_stringformatting(self): |
||||
tmpl = newstyle_i18n_env.get_template('stringformat.html') |
||||
assert tmpl.render(LANGUAGE='de', user_count=5) == 'Benutzer: 5' |
||||
|
||||
def test_newstyle_plural(self): |
||||
tmpl = newstyle_i18n_env.get_template('ngettext.html') |
||||
assert tmpl.render(LANGUAGE='de', apples=1) == '1 Apfel' |
||||
assert tmpl.render(LANGUAGE='de', apples=5) == u'5 Äpfel' |
||||
|
||||
def test_autoescape_support(self): |
||||
env = Environment(extensions=['jinja2.ext.autoescape', |
||||
'jinja2.ext.i18n']) |
||||
env.install_gettext_callables(lambda x: u'<strong>Wert: %(name)s</strong>', |
||||
lambda s, p, n: s, newstyle=True) |
||||
t = env.from_string('{% autoescape ae %}{{ gettext("foo", name=' |
||||
'"<test>") }}{% endautoescape %}') |
||||
assert t.render(ae=True) == '<strong>Wert: <test></strong>' |
||||
assert t.render(ae=False) == '<strong>Wert: <test></strong>' |
||||
|
||||
def test_num_used_twice(self): |
||||
tmpl = newstyle_i18n_env.get_template('ngettext_long.html') |
||||
assert tmpl.render(apples=5, LANGUAGE='de') == u'5 Äpfel' |
||||
|
||||
def test_num_called_num(self): |
||||
source = newstyle_i18n_env.compile(''' |
||||
{% trans num=3 %}{{ num }} apple{% pluralize |
||||
%}{{ num }} apples{% endtrans %} |
||||
''', raw=True) |
||||
# quite hacky, but the only way to properly test that. The idea is |
||||
# that the generated code does not pass num twice (although that |
||||
# would work) for better performance. This only works on the |
||||
# newstyle gettext of course |
||||
assert re.search(r"l_ngettext, u?'\%\(num\)s apple', u?'\%\(num\)s " |
||||
r"apples', 3", source) is not None |
||||
|
||||
def test_trans_vars(self): |
||||
t1 = newstyle_i18n_env.get_template('transvars1.html') |
||||
t2 = newstyle_i18n_env.get_template('transvars2.html') |
||||
t3 = newstyle_i18n_env.get_template('transvars3.html') |
||||
assert t1.render(num=1, LANGUAGE='de') == 'Benutzer: 1' |
||||
assert t2.render(count=23, LANGUAGE='de') == 'Benutzer: 23' |
||||
assert t3.render(num=42, LANGUAGE='de') == 'Benutzer: 42' |
||||
|
||||
def test_novars_vars_escaping(self): |
||||
t = newstyle_i18n_env.get_template('novars.html') |
||||
assert t.render() == '%(hello)s' |
||||
t = newstyle_i18n_env.get_template('vars.html') |
||||
assert t.render(foo='42') == '42%(foo)s' |
||||
t = newstyle_i18n_env.get_template('explicitvars.html') |
||||
assert t.render() == '%(foo)s' |
||||
|
||||
|
||||
class AutoEscapeTestCase(JinjaTestCase): |
||||
|
||||
def test_scoped_setting(self): |
||||
env = Environment(extensions=['jinja2.ext.autoescape'], |
||||
autoescape=True) |
||||
tmpl = env.from_string(''' |
||||
{{ "<HelloWorld>" }} |
||||
{% autoescape false %} |
||||
{{ "<HelloWorld>" }} |
||||
{% endautoescape %} |
||||
{{ "<HelloWorld>" }} |
||||
''') |
||||
assert tmpl.render().split() == \ |
||||
[u'<HelloWorld>', u'<HelloWorld>', u'<HelloWorld>'] |
||||
|
||||
env = Environment(extensions=['jinja2.ext.autoescape'], |
||||
autoescape=False) |
||||
tmpl = env.from_string(''' |
||||
{{ "<HelloWorld>" }} |
||||
{% autoescape true %} |
||||
{{ "<HelloWorld>" }} |
||||
{% endautoescape %} |
||||
{{ "<HelloWorld>" }} |
||||
''') |
||||
assert tmpl.render().split() == \ |
||||
[u'<HelloWorld>', u'<HelloWorld>', u'<HelloWorld>'] |
||||
|
||||
def test_nonvolatile(self): |
||||
env = Environment(extensions=['jinja2.ext.autoescape'], |
||||
autoescape=True) |
||||
tmpl = env.from_string('{{ {"foo": "<test>"}|xmlattr|escape }}') |
||||
assert tmpl.render() == ' foo="<test>"' |
||||
tmpl = env.from_string('{% autoescape false %}{{ {"foo": "<test>"}' |
||||
'|xmlattr|escape }}{% endautoescape %}') |
||||
assert tmpl.render() == ' foo="&lt;test&gt;"' |
||||
|
||||
def test_volatile(self): |
||||
env = Environment(extensions=['jinja2.ext.autoescape'], |
||||
autoescape=True) |
||||
tmpl = env.from_string('{% autoescape foo %}{{ {"foo": "<test>"}' |
||||
'|xmlattr|escape }}{% endautoescape %}') |
||||
assert tmpl.render(foo=False) == ' foo="&lt;test&gt;"' |
||||
assert tmpl.render(foo=True) == ' foo="<test>"' |
||||
|
||||
def test_scoping(self): |
||||
env = Environment(extensions=['jinja2.ext.autoescape']) |
||||
tmpl = env.from_string('{% autoescape true %}{% set x = "<x>" %}{{ x }}' |
||||
'{% endautoescape %}{{ x }}{{ "<y>" }}') |
||||
assert tmpl.render(x=1) == '<x>1<y>' |
||||
|
||||
def test_volatile_scoping(self): |
||||
env = Environment(extensions=['jinja2.ext.autoescape']) |
||||
tmplsource = ''' |
||||
{% autoescape val %} |
||||
{% macro foo(x) %} |
||||
[{{ x }}] |
||||
{% endmacro %} |
||||
{{ foo().__class__.__name__ }} |
||||
{% endautoescape %} |
||||
{{ '<testing>' }} |
||||
''' |
||||
tmpl = env.from_string(tmplsource) |
||||
assert tmpl.render(val=True).split()[0] == 'Markup' |
||||
assert tmpl.render(val=False).split()[0] == unicode.__name__ |
||||
|
||||
# looking at the source we should see <testing> there in raw |
||||
# (and then escaped as well) |
||||
env = Environment(extensions=['jinja2.ext.autoescape']) |
||||
pysource = env.compile(tmplsource, raw=True) |
||||
assert '<testing>\\n' in pysource |
||||
|
||||
env = Environment(extensions=['jinja2.ext.autoescape'], |
||||
autoescape=True) |
||||
pysource = env.compile(tmplsource, raw=True) |
||||
assert '<testing>\\n' in pysource |
||||
|
||||
|
||||
def suite(): |
||||
suite = unittest.TestSuite() |
||||
suite.addTest(unittest.makeSuite(ExtensionsTestCase)) |
||||
suite.addTest(unittest.makeSuite(InternationalizationTestCase)) |
||||
suite.addTest(unittest.makeSuite(NewstyleInternationalizationTestCase)) |
||||
suite.addTest(unittest.makeSuite(AutoEscapeTestCase)) |
||||
return suite |
@ -1,396 +0,0 @@ |
||||
# -*- coding: utf-8 -*- |
||||
""" |
||||
jinja2.testsuite.filters |
||||
~~~~~~~~~~~~~~~~~~~~~~~~ |
||||
|
||||
Tests for the jinja filters. |
||||
|
||||
:copyright: (c) 2010 by the Jinja Team. |
||||
:license: BSD, see LICENSE for more details. |
||||
""" |
||||
import unittest |
||||
from jinja2.testsuite import JinjaTestCase |
||||
|
||||
from jinja2 import Markup, Environment |
||||
|
||||
env = Environment() |
||||
|
||||
|
||||
class FilterTestCase(JinjaTestCase): |
||||
|
||||
def test_capitalize(self): |
||||
tmpl = env.from_string('{{ "foo bar"|capitalize }}') |
||||
assert tmpl.render() == 'Foo bar' |
||||
|
||||
def test_center(self): |
||||
tmpl = env.from_string('{{ "foo"|center(9) }}') |
||||
assert tmpl.render() == ' foo ' |
||||
|
||||
def test_default(self): |
||||
tmpl = env.from_string( |
||||
"{{ missing|default('no') }}|{{ false|default('no') }}|" |
||||
"{{ false|default('no', true) }}|{{ given|default('no') }}" |
||||
) |
||||
assert tmpl.render(given='yes') == 'no|False|no|yes' |
||||
|
||||
def test_dictsort(self): |
||||
tmpl = env.from_string( |
||||
'{{ foo|dictsort }}|' |
||||
'{{ foo|dictsort(true) }}|' |
||||
'{{ foo|dictsort(false, "value") }}' |
||||
) |
||||
out = tmpl.render(foo={"aa": 0, "b": 1, "c": 2, "AB": 3}) |
||||
assert out == ("[('aa', 0), ('AB', 3), ('b', 1), ('c', 2)]|" |
||||
"[('AB', 3), ('aa', 0), ('b', 1), ('c', 2)]|" |
||||
"[('aa', 0), ('b', 1), ('c', 2), ('AB', 3)]") |
||||
|
||||
def test_batch(self): |
||||
tmpl = env.from_string("{{ foo|batch(3)|list }}|" |
||||
"{{ foo|batch(3, 'X')|list }}") |
||||
out = tmpl.render(foo=range(10)) |
||||
assert out == ("[[0, 1, 2], [3, 4, 5], [6, 7, 8], [9]]|" |
||||
"[[0, 1, 2], [3, 4, 5], [6, 7, 8], [9, 'X', 'X']]") |
||||
|
||||
def test_slice(self): |
||||
tmpl = env.from_string('{{ foo|slice(3)|list }}|' |
||||
'{{ foo|slice(3, "X")|list }}') |
||||
out = tmpl.render(foo=range(10)) |
||||
assert out == ("[[0, 1, 2, 3], [4, 5, 6], [7, 8, 9]]|" |
||||
"[[0, 1, 2, 3], [4, 5, 6, 'X'], [7, 8, 9, 'X']]") |
||||
|
||||
def test_escape(self): |
||||
tmpl = env.from_string('''{{ '<">&'|escape }}''') |
||||
out = tmpl.render() |
||||
assert out == '<">&' |
||||
|
||||
def test_striptags(self): |
||||
tmpl = env.from_string('''{{ foo|striptags }}''') |
||||
out = tmpl.render(foo=' <p>just a small \n <a href="#">' |
||||
'example</a> link</p>\n<p>to a webpage</p> ' |
||||
'<!-- <p>and some commented stuff</p> -->') |
||||
assert out == 'just a small example link to a webpage' |
||||
|
||||
def test_filesizeformat(self): |
||||
tmpl = env.from_string( |
||||
'{{ 100|filesizeformat }}|' |
||||
'{{ 1000|filesizeformat }}|' |
||||
'{{ 1000000|filesizeformat }}|' |
||||
'{{ 1000000000|filesizeformat }}|' |
||||
'{{ 1000000000000|filesizeformat }}|' |
||||
'{{ 100|filesizeformat(true) }}|' |
||||
'{{ 1000|filesizeformat(true) }}|' |
||||
'{{ 1000000|filesizeformat(true) }}|' |
||||
'{{ 1000000000|filesizeformat(true) }}|' |
||||
'{{ 1000000000000|filesizeformat(true) }}' |
||||
) |
||||
out = tmpl.render() |
||||
self.assert_equal(out, ( |
||||
'100 Bytes|1.0 kB|1.0 MB|1.0 GB|1.0 TB|100 Bytes|' |
||||
'1000 Bytes|976.6 KiB|953.7 MiB|931.3 GiB' |
||||
)) |
||||
|
||||
def test_filesizeformat_issue59(self): |
||||
tmpl = env.from_string( |
||||
'{{ 300|filesizeformat }}|' |
||||
'{{ 3000|filesizeformat }}|' |
||||
'{{ 3000000|filesizeformat }}|' |
||||
'{{ 3000000000|filesizeformat }}|' |
||||
'{{ 3000000000000|filesizeformat }}|' |
||||
'{{ 300|filesizeformat(true) }}|' |
||||
'{{ 3000|filesizeformat(true) }}|' |
||||
'{{ 3000000|filesizeformat(true) }}' |
||||
) |
||||
out = tmpl.render() |
||||
self.assert_equal(out, ( |
||||
'300 Bytes|3.0 kB|3.0 MB|3.0 GB|3.0 TB|300 Bytes|' |
||||
'2.9 KiB|2.9 MiB' |
||||
)) |
||||
|
||||
|
||||
def test_first(self): |
||||
tmpl = env.from_string('{{ foo|first }}') |
||||
out = tmpl.render(foo=range(10)) |
||||
assert out == '0' |
||||
|
||||
def test_float(self): |
||||
tmpl = env.from_string('{{ "42"|float }}|' |
||||
'{{ "ajsghasjgd"|float }}|' |
||||
'{{ "32.32"|float }}') |
||||
out = tmpl.render() |
||||
assert out == '42.0|0.0|32.32' |
||||
|
||||
def test_format(self): |
||||
tmpl = env.from_string('''{{ "%s|%s"|format("a", "b") }}''') |
||||
out = tmpl.render() |
||||
assert out == 'a|b' |
||||
|
||||
def test_indent(self): |
||||
tmpl = env.from_string('{{ foo|indent(2) }}|{{ foo|indent(2, true) }}') |
||||
text = '\n'.join([' '.join(['foo', 'bar'] * 2)] * 2) |
||||
out = tmpl.render(foo=text) |
||||
assert out == ('foo bar foo bar\n foo bar foo bar| ' |
||||
'foo bar foo bar\n foo bar foo bar') |
||||
|
||||
def test_int(self): |
||||
tmpl = env.from_string('{{ "42"|int }}|{{ "ajsghasjgd"|int }}|' |
||||
'{{ "32.32"|int }}') |
||||
out = tmpl.render() |
||||
assert out == '42|0|32' |
||||
|
||||
def test_join(self): |
||||
tmpl = env.from_string('{{ [1, 2, 3]|join("|") }}') |
||||
out = tmpl.render() |
||||
assert out == '1|2|3' |
||||
|
||||
env2 = Environment(autoescape=True) |
||||
tmpl = env2.from_string('{{ ["<foo>", "<span>foo</span>"|safe]|join }}') |
||||
assert tmpl.render() == '<foo><span>foo</span>' |
||||
|
||||
def test_join_attribute(self): |
||||
class User(object): |
||||
def __init__(self, username): |
||||
self.username = username |
||||
tmpl = env.from_string('''{{ users|join(', ', 'username') }}''') |
||||
assert tmpl.render(users=map(User, ['foo', 'bar'])) == 'foo, bar' |
||||
|
||||
def test_last(self): |
||||
tmpl = env.from_string('''{{ foo|last }}''') |
||||
out = tmpl.render(foo=range(10)) |
||||
assert out == '9' |
||||
|
||||
def test_length(self): |
||||
tmpl = env.from_string('''{{ "hello world"|length }}''') |
||||
out = tmpl.render() |
||||
assert out == '11' |
||||
|
||||
def test_lower(self): |
||||
tmpl = env.from_string('''{{ "FOO"|lower }}''') |
||||
out = tmpl.render() |
||||
assert out == 'foo' |
||||
|
||||
def test_pprint(self): |
||||
from pprint import pformat |
||||
tmpl = env.from_string('''{{ data|pprint }}''') |
||||
data = range(1000) |
||||
assert tmpl.render(data=data) == pformat(data) |
||||
|
||||
def test_random(self): |
||||
tmpl = env.from_string('''{{ seq|random }}''') |
||||
seq = range(100) |
||||
for _ in range(10): |
||||
assert int(tmpl.render(seq=seq)) in seq |
||||
|
||||
def test_reverse(self): |
||||
tmpl = env.from_string('{{ "foobar"|reverse|join }}|' |
||||
'{{ [1, 2, 3]|reverse|list }}') |
||||
assert tmpl.render() == 'raboof|[3, 2, 1]' |
||||
|
||||
def test_string(self): |
||||
x = [1, 2, 3, 4, 5] |
||||
tmpl = env.from_string('''{{ obj|string }}''') |
||||
assert tmpl.render(obj=x) == unicode(x) |
||||
|
||||
def test_title(self): |
||||
tmpl = env.from_string('''{{ "foo bar"|title }}''') |
||||
assert tmpl.render() == "Foo Bar" |
||||
tmpl = env.from_string('''{{ "foo's bar"|title }}''') |
||||
assert tmpl.render() == "Foo's Bar" |
||||
tmpl = env.from_string('''{{ "foo bar"|title }}''') |
||||
assert tmpl.render() == "Foo Bar" |
||||
tmpl = env.from_string('''{{ "f bar f"|title }}''') |
||||
assert tmpl.render() == "F Bar F" |
||||
tmpl = env.from_string('''{{ "foo-bar"|title }}''') |
||||
assert tmpl.render() == "Foo-Bar" |
||||
tmpl = env.from_string('''{{ "foo\tbar"|title }}''') |
||||
assert tmpl.render() == "Foo\tBar" |
||||
|
||||
def test_truncate(self): |
||||
tmpl = env.from_string( |
||||
'{{ data|truncate(15, true, ">>>") }}|' |
||||
'{{ data|truncate(15, false, ">>>") }}|' |
||||
'{{ smalldata|truncate(15) }}' |
||||
) |
||||
out = tmpl.render(data='foobar baz bar' * 1000, |
||||
smalldata='foobar baz bar') |
||||
assert out == 'foobar baz barf>>>|foobar baz >>>|foobar baz bar' |
||||
|
||||
def test_upper(self): |
||||
tmpl = env.from_string('{{ "foo"|upper }}') |
||||
assert tmpl.render() == 'FOO' |
||||
|
||||
def test_urlize(self): |
||||
tmpl = env.from_string('{{ "foo http://www.example.com/ bar"|urlize }}') |
||||
assert tmpl.render() == 'foo <a href="http://www.example.com/">'\ |
||||
'http://www.example.com/</a> bar' |
||||
|
||||
def test_wordcount(self): |
||||
tmpl = env.from_string('{{ "foo bar baz"|wordcount }}') |
||||
assert tmpl.render() == '3' |
||||
|
||||
def test_block(self): |
||||
tmpl = env.from_string('{% filter lower|escape %}<HEHE>{% endfilter %}') |
||||
assert tmpl.render() == '<hehe>' |
||||
|
||||
def test_chaining(self): |
||||
tmpl = env.from_string('''{{ ['<foo>', '<bar>']|first|upper|escape }}''') |
||||
assert tmpl.render() == '<FOO>' |
||||
|
||||
def test_sum(self): |
||||
tmpl = env.from_string('''{{ [1, 2, 3, 4, 5, 6]|sum }}''') |
||||
assert tmpl.render() == '21' |
||||
|
||||
def test_sum_attributes(self): |
||||
tmpl = env.from_string('''{{ values|sum('value') }}''') |
||||
assert tmpl.render(values=[ |
||||
{'value': 23}, |
||||
{'value': 1}, |
||||
{'value': 18}, |
||||
]) == '42' |
||||
|
||||
def test_sum_attributes_nested(self): |
||||
tmpl = env.from_string('''{{ values|sum('real.value') }}''') |
||||
assert tmpl.render(values=[ |
||||
{'real': {'value': 23}}, |
||||
{'real': {'value': 1}}, |
||||
{'real': {'value': 18}}, |
||||
]) == '42' |
||||
|
||||
def test_abs(self): |
||||
tmpl = env.from_string('''{{ -1|abs }}|{{ 1|abs }}''') |
||||
assert tmpl.render() == '1|1', tmpl.render() |
||||
|
||||
def test_round_positive(self): |
||||
tmpl = env.from_string('{{ 2.7|round }}|{{ 2.1|round }}|' |
||||
"{{ 2.1234|round(3, 'floor') }}|" |
||||
"{{ 2.1|round(0, 'ceil') }}") |
||||
assert tmpl.render() == '3.0|2.0|2.123|3.0', tmpl.render() |
||||
|
||||
def test_round_negative(self): |
||||
tmpl = env.from_string('{{ 21.3|round(-1)}}|' |
||||
"{{ 21.3|round(-1, 'ceil')}}|" |
||||
"{{ 21.3|round(-1, 'floor')}}") |
||||
assert tmpl.render() == '20.0|30.0|20.0',tmpl.render() |
||||
|
||||
def test_xmlattr(self): |
||||
tmpl = env.from_string("{{ {'foo': 42, 'bar': 23, 'fish': none, " |
||||
"'spam': missing, 'blub:blub': '<?>'}|xmlattr }}") |
||||
out = tmpl.render().split() |
||||
assert len(out) == 3 |
||||
assert 'foo="42"' in out |
||||
assert 'bar="23"' in out |
||||
assert 'blub:blub="<?>"' in out |
||||
|
||||
def test_sort1(self): |
||||
tmpl = env.from_string('{{ [2, 3, 1]|sort }}|{{ [2, 3, 1]|sort(true) }}') |
||||
assert tmpl.render() == '[1, 2, 3]|[3, 2, 1]' |
||||
|
||||
def test_sort2(self): |
||||
tmpl = env.from_string('{{ "".join(["c", "A", "b", "D"]|sort) }}') |
||||
assert tmpl.render() == 'AbcD' |
||||
|
||||
def test_sort3(self): |
||||
tmpl = env.from_string('''{{ ['foo', 'Bar', 'blah']|sort }}''') |
||||
assert tmpl.render() == "['Bar', 'blah', 'foo']" |
||||
|
||||
def test_sort4(self): |
||||
class Magic(object): |
||||
def __init__(self, value): |
||||
self.value = value |
||||
def __unicode__(self): |
||||
return unicode(self.value) |
||||
tmpl = env.from_string('''{{ items|sort(attribute='value')|join }}''') |
||||
assert tmpl.render(items=map(Magic, [3, 2, 4, 1])) == '1234' |
||||
|
||||
def test_groupby(self): |
||||
tmpl = env.from_string(''' |
||||
{%- for grouper, list in [{'foo': 1, 'bar': 2}, |
||||
{'foo': 2, 'bar': 3}, |
||||
{'foo': 1, 'bar': 1}, |
||||
{'foo': 3, 'bar': 4}]|groupby('foo') -%} |
||||
{{ grouper }}{% for x in list %}: {{ x.foo }}, {{ x.bar }}{% endfor %}| |
||||
{%- endfor %}''') |
||||
assert tmpl.render().split('|') == [ |
||||
"1: 1, 2: 1, 1", |
||||
"2: 2, 3", |
||||
"3: 3, 4", |
||||
"" |
||||
] |
||||
|
||||
def test_groupby_tuple_index(self): |
||||
tmpl = env.from_string(''' |
||||
{%- for grouper, list in [('a', 1), ('a', 2), ('b', 1)]|groupby(0) -%} |
||||
{{ grouper }}{% for x in list %}:{{ x.1 }}{% endfor %}| |
||||
{%- endfor %}''') |
||||
assert tmpl.render() == 'a:1:2|b:1|' |
||||
|
||||
def test_groupby_multidot(self): |
||||
class Date(object): |
||||
def __init__(self, day, month, year): |
||||
self.day = day |
||||
self.month = month |
||||
self.year = year |
||||
class Article(object): |
||||
def __init__(self, title, *date): |
||||
self.date = Date(*date) |
||||
self.title = title |
||||
articles = [ |
||||
Article('aha', 1, 1, 1970), |
||||
Article('interesting', 2, 1, 1970), |
||||
Article('really?', 3, 1, 1970), |
||||
Article('totally not', 1, 1, 1971) |
||||
] |
||||
tmpl = env.from_string(''' |
||||
{%- for year, list in articles|groupby('date.year') -%} |
||||
{{ year }}{% for x in list %}[{{ x.title }}]{% endfor %}| |
||||
{%- endfor %}''') |
||||
assert tmpl.render(articles=articles).split('|') == [ |
||||
'1970[aha][interesting][really?]', |
||||
'1971[totally not]', |
||||
'' |
||||
] |
||||
|
||||
def test_filtertag(self): |
||||
tmpl = env.from_string("{% filter upper|replace('FOO', 'foo') %}" |
||||
"foobar{% endfilter %}") |
||||
assert tmpl.render() == 'fooBAR' |
||||
|
||||
def test_replace(self): |
||||
env = Environment() |
||||
tmpl = env.from_string('{{ string|replace("o", 42) }}') |
||||
assert tmpl.render(string='<foo>') == '<f4242>' |
||||
env = Environment(autoescape=True) |
||||
tmpl = env.from_string('{{ string|replace("o", 42) }}') |
||||
assert tmpl.render(string='<foo>') == '<f4242>' |
||||
tmpl = env.from_string('{{ string|replace("<", 42) }}') |
||||
assert tmpl.render(string='<foo>') == '42foo>' |
||||
tmpl = env.from_string('{{ string|replace("o", ">x<") }}') |
||||
assert tmpl.render(string=Markup('foo')) == 'f>x<>x<' |
||||
|
||||
def test_forceescape(self): |
||||
tmpl = env.from_string('{{ x|forceescape }}') |
||||
assert tmpl.render(x=Markup('<div />')) == u'<div />' |
||||
|
||||
def test_safe(self): |
||||
env = Environment(autoescape=True) |
||||
tmpl = env.from_string('{{ "<div>foo</div>"|safe }}') |
||||
assert tmpl.render() == '<div>foo</div>' |
||||
tmpl = env.from_string('{{ "<div>foo</div>" }}') |
||||
assert tmpl.render() == '<div>foo</div>' |
||||
|
||||
def test_urlencode(self): |
||||
env = Environment(autoescape=True) |
||||
tmpl = env.from_string('{{ "Hello, world!"|urlencode }}') |
||||
assert tmpl.render() == 'Hello%2C%20world%21' |
||||
tmpl = env.from_string('{{ o|urlencode }}') |
||||
assert tmpl.render(o=u"Hello, world\u203d") == "Hello%2C%20world%E2%80%BD" |
||||
assert tmpl.render(o=(("f", 1),)) == "f=1" |
||||
assert tmpl.render(o=(('f', 1), ("z", 2))) == "f=1&z=2" |
||||
assert tmpl.render(o=((u"\u203d", 1),)) == "%E2%80%BD=1" |
||||
assert tmpl.render(o={u"\u203d": 1}) == "%E2%80%BD=1" |
||||
assert tmpl.render(o={0: 1}) == "0=1" |
||||
|
||||
|
||||
def suite(): |
||||
suite = unittest.TestSuite() |
||||
suite.addTest(unittest.makeSuite(FilterTestCase)) |
||||
return suite |
@ -1,141 +0,0 @@ |
||||
# -*- coding: utf-8 -*- |
||||
""" |
||||
jinja2.testsuite.imports |
||||
~~~~~~~~~~~~~~~~~~~~~~~~ |
||||
|
||||
Tests the import features (with includes). |
||||
|
||||
:copyright: (c) 2010 by the Jinja Team. |
||||
:license: BSD, see LICENSE for more details. |
||||
""" |
||||
import unittest |
||||
|
||||
from jinja2.testsuite import JinjaTestCase |
||||
|
||||
from jinja2 import Environment, DictLoader |
||||
from jinja2.exceptions import TemplateNotFound, TemplatesNotFound |
||||
|
||||
|
||||
test_env = Environment(loader=DictLoader(dict( |
||||
module='{% macro test() %}[{{ foo }}|{{ bar }}]{% endmacro %}', |
||||
header='[{{ foo }}|{{ 23 }}]', |
||||
o_printer='({{ o }})' |
||||
))) |
||||
test_env.globals['bar'] = 23 |
||||
|
||||
|
||||
class ImportsTestCase(JinjaTestCase): |
||||
|
||||
def test_context_imports(self): |
||||
t = test_env.from_string('{% import "module" as m %}{{ m.test() }}') |
||||
assert t.render(foo=42) == '[|23]' |
||||
t = test_env.from_string('{% import "module" as m without context %}{{ m.test() }}') |
||||
assert t.render(foo=42) == '[|23]' |
||||
t = test_env.from_string('{% import "module" as m with context %}{{ m.test() }}') |
||||
assert t.render(foo=42) == '[42|23]' |
||||
t = test_env.from_string('{% from "module" import test %}{{ test() }}') |
||||
assert t.render(foo=42) == '[|23]' |
||||
t = test_env.from_string('{% from "module" import test without context %}{{ test() }}') |
||||
assert t.render(foo=42) == '[|23]' |
||||
t = test_env.from_string('{% from "module" import test with context %}{{ test() }}') |
||||
assert t.render(foo=42) == '[42|23]' |
||||
|
||||
def test_trailing_comma(self): |
||||
test_env.from_string('{% from "foo" import bar, baz with context %}') |
||||
test_env.from_string('{% from "foo" import bar, baz, with context %}') |
||||
test_env.from_string('{% from "foo" import bar, with context %}') |
||||
test_env.from_string('{% from "foo" import bar, with, context %}') |
||||
test_env.from_string('{% from "foo" import bar, with with context %}') |
||||
|
||||
def test_exports(self): |
||||
m = test_env.from_string(''' |
||||
{% macro toplevel() %}...{% endmacro %} |
||||
{% macro __private() %}...{% endmacro %} |
||||
{% set variable = 42 %} |
||||
{% for item in [1] %} |
||||
{% macro notthere() %}{% endmacro %} |
||||
{% endfor %} |
||||
''').module |
||||
assert m.toplevel() == '...' |
||||
assert not hasattr(m, '__missing') |
||||
assert m.variable == 42 |
||||
assert not hasattr(m, 'notthere') |
||||
|
||||
|
||||
class IncludesTestCase(JinjaTestCase): |
||||
|
||||
def test_context_include(self): |
||||
t = test_env.from_string('{% include "header" %}') |
||||
assert t.render(foo=42) == '[42|23]' |
||||
t = test_env.from_string('{% include "header" with context %}') |
||||
assert t.render(foo=42) == '[42|23]' |
||||
t = test_env.from_string('{% include "header" without context %}') |
||||
assert t.render(foo=42) == '[|23]' |
||||
|
||||
def test_choice_includes(self): |
||||
t = test_env.from_string('{% include ["missing", "header"] %}') |
||||
assert t.render(foo=42) == '[42|23]' |
||||
|
||||
t = test_env.from_string('{% include ["missing", "missing2"] ignore missing %}') |
||||
assert t.render(foo=42) == '' |
||||
|
||||
t = test_env.from_string('{% include ["missing", "missing2"] %}') |
||||
self.assert_raises(TemplateNotFound, t.render) |
||||
try: |
||||
t.render() |
||||
except TemplatesNotFound, e: |
||||
assert e.templates == ['missing', 'missing2'] |
||||
assert e.name == 'missing2' |
||||
else: |
||||
assert False, 'thou shalt raise' |
||||
|
||||
def test_includes(t, **ctx): |
||||
ctx['foo'] = 42 |
||||
assert t.render(ctx) == '[42|23]' |
||||
|
||||
t = test_env.from_string('{% include ["missing", "header"] %}') |
||||
test_includes(t) |
||||
t = test_env.from_string('{% include x %}') |
||||
test_includes(t, x=['missing', 'header']) |
||||
t = test_env.from_string('{% include [x, "header"] %}') |
||||
test_includes(t, x='missing') |
||||
t = test_env.from_string('{% include x %}') |
||||
test_includes(t, x='header') |
||||
t = test_env.from_string('{% include x %}') |
||||
test_includes(t, x='header') |
||||
t = test_env.from_string('{% include [x] %}') |
||||
test_includes(t, x='header') |
||||
|
||||
def test_include_ignoring_missing(self): |
||||
t = test_env.from_string('{% include "missing" %}') |
||||
self.assert_raises(TemplateNotFound, t.render) |
||||
for extra in '', 'with context', 'without context': |
||||
t = test_env.from_string('{% include "missing" ignore missing ' + |
||||
extra + ' %}') |
||||
assert t.render() == '' |
||||
|
||||
def test_context_include_with_overrides(self): |
||||
env = Environment(loader=DictLoader(dict( |
||||
main="{% for item in [1, 2, 3] %}{% include 'item' %}{% endfor %}", |
||||
item="{{ item }}" |
||||
))) |
||||
assert env.get_template("main").render() == "123" |
||||
|
||||
def test_unoptimized_scopes(self): |
||||
t = test_env.from_string(""" |
||||
{% macro outer(o) %} |
||||
{% macro inner() %} |
||||
{% include "o_printer" %} |
||||
{% endmacro %} |
||||
{{ inner() }} |
||||
{% endmacro %} |
||||
{{ outer("FOO") }} |
||||
""") |
||||
assert t.render().strip() == '(FOO)' |
||||
|
||||
|
||||
def suite(): |
||||
suite = unittest.TestSuite() |
||||
suite.addTest(unittest.makeSuite(ImportsTestCase)) |
||||
suite.addTest(unittest.makeSuite(IncludesTestCase)) |
||||
return suite |
@ -1,227 +0,0 @@ |
||||
# -*- coding: utf-8 -*- |
||||
""" |
||||
jinja2.testsuite.inheritance |
||||
~~~~~~~~~~~~~~~~~~~~~~~~~~~~ |
||||
|
||||
Tests the template inheritance feature. |
||||
|
||||
:copyright: (c) 2010 by the Jinja Team. |
||||
:license: BSD, see LICENSE for more details. |
||||
""" |
||||
import unittest |
||||
|
||||
from jinja2.testsuite import JinjaTestCase |
||||
|
||||
from jinja2 import Environment, DictLoader |
||||
|
||||
|
||||
LAYOUTTEMPLATE = '''\ |
||||
|{% block block1 %}block 1 from layout{% endblock %} |
||||
|{% block block2 %}block 2 from layout{% endblock %} |
||||
|{% block block3 %} |
||||
{% block block4 %}nested block 4 from layout{% endblock %} |
||||
{% endblock %}|''' |
||||
|
||||
LEVEL1TEMPLATE = '''\ |
||||
{% extends "layout" %} |
||||
{% block block1 %}block 1 from level1{% endblock %}''' |
||||
|
||||
LEVEL2TEMPLATE = '''\ |
||||
{% extends "level1" %} |
||||
{% block block2 %}{% block block5 %}nested block 5 from level2{% |
||||
endblock %}{% endblock %}''' |
||||
|
||||
LEVEL3TEMPLATE = '''\ |
||||
{% extends "level2" %} |
||||
{% block block5 %}block 5 from level3{% endblock %} |
||||
{% block block4 %}block 4 from level3{% endblock %} |
||||
''' |
||||
|
||||
LEVEL4TEMPLATE = '''\ |
||||
{% extends "level3" %} |
||||
{% block block3 %}block 3 from level4{% endblock %} |
||||
''' |
||||
|
||||
WORKINGTEMPLATE = '''\ |
||||
{% extends "layout" %} |
||||
{% block block1 %} |
||||
{% if false %} |
||||
{% block block2 %} |
||||
this should workd |
||||
{% endblock %} |
||||
{% endif %} |
||||
{% endblock %} |
||||
''' |
||||
|
||||
env = Environment(loader=DictLoader({ |
||||
'layout': LAYOUTTEMPLATE, |
||||
'level1': LEVEL1TEMPLATE, |
||||
'level2': LEVEL2TEMPLATE, |
||||
'level3': LEVEL3TEMPLATE, |
||||
'level4': LEVEL4TEMPLATE, |
||||
'working': WORKINGTEMPLATE |
||||
}), trim_blocks=True) |
||||
|
||||
|
||||
class InheritanceTestCase(JinjaTestCase): |
||||
|
||||
def test_layout(self): |
||||
tmpl = env.get_template('layout') |
||||
assert tmpl.render() == ('|block 1 from layout|block 2 from ' |
||||
'layout|nested block 4 from layout|') |
||||
|
||||
def test_level1(self): |
||||
tmpl = env.get_template('level1') |
||||
assert tmpl.render() == ('|block 1 from level1|block 2 from ' |
||||
'layout|nested block 4 from layout|') |
||||
|
||||
def test_level2(self): |
||||
tmpl = env.get_template('level2') |
||||
assert tmpl.render() == ('|block 1 from level1|nested block 5 from ' |
||||
'level2|nested block 4 from layout|') |
||||
|
||||
def test_level3(self): |
||||
tmpl = env.get_template('level3') |
||||
assert tmpl.render() == ('|block 1 from level1|block 5 from level3|' |
||||
'block 4 from level3|') |
||||
|
||||
def test_level4(sel): |
||||
tmpl = env.get_template('level4') |
||||
assert tmpl.render() == ('|block 1 from level1|block 5 from ' |
||||
'level3|block 3 from level4|') |
||||
|
||||
def test_super(self): |
||||
env = Environment(loader=DictLoader({ |
||||
'a': '{% block intro %}INTRO{% endblock %}|' |
||||
'BEFORE|{% block data %}INNER{% endblock %}|AFTER', |
||||
'b': '{% extends "a" %}{% block data %}({{ ' |
||||
'super() }}){% endblock %}', |
||||
'c': '{% extends "b" %}{% block intro %}--{{ ' |
||||
'super() }}--{% endblock %}\n{% block data ' |
||||
'%}[{{ super() }}]{% endblock %}' |
||||
})) |
||||
tmpl = env.get_template('c') |
||||
assert tmpl.render() == '--INTRO--|BEFORE|[(INNER)]|AFTER' |
||||
|
||||
def test_working(self): |
||||
tmpl = env.get_template('working') |
||||
|
||||
def test_reuse_blocks(self): |
||||
tmpl = env.from_string('{{ self.foo() }}|{% block foo %}42' |
||||
'{% endblock %}|{{ self.foo() }}') |
||||
assert tmpl.render() == '42|42|42' |
||||
|
||||
def test_preserve_blocks(self): |
||||
env = Environment(loader=DictLoader({ |
||||
'a': '{% if false %}{% block x %}A{% endblock %}{% endif %}{{ self.x() }}', |
||||
'b': '{% extends "a" %}{% block x %}B{{ super() }}{% endblock %}' |
||||
})) |
||||
tmpl = env.get_template('b') |
||||
assert tmpl.render() == 'BA' |
||||
|
||||
def test_dynamic_inheritance(self): |
||||
env = Environment(loader=DictLoader({ |
||||
'master1': 'MASTER1{% block x %}{% endblock %}', |
||||
'master2': 'MASTER2{% block x %}{% endblock %}', |
||||
'child': '{% extends master %}{% block x %}CHILD{% endblock %}' |
||||
})) |
||||
tmpl = env.get_template('child') |
||||
for m in range(1, 3): |
||||
assert tmpl.render(master='master%d' % m) == 'MASTER%dCHILD' % m |
||||
|
||||
def test_multi_inheritance(self): |
||||
env = Environment(loader=DictLoader({ |
||||
'master1': 'MASTER1{% block x %}{% endblock %}', |
||||
'master2': 'MASTER2{% block x %}{% endblock %}', |
||||
'child': '''{% if master %}{% extends master %}{% else %}{% extends |
||||
'master1' %}{% endif %}{% block x %}CHILD{% endblock %}''' |
||||
})) |
||||
tmpl = env.get_template('child') |
||||
assert tmpl.render(master='master2') == 'MASTER2CHILD' |
||||
assert tmpl.render(master='master1') == 'MASTER1CHILD' |
||||
assert tmpl.render() == 'MASTER1CHILD' |
||||
|
||||
def test_scoped_block(self): |
||||
env = Environment(loader=DictLoader({ |
||||
'master.html': '{% for item in seq %}[{% block item scoped %}' |
||||
'{% endblock %}]{% endfor %}' |
||||
})) |
||||
t = env.from_string('{% extends "master.html" %}{% block item %}' |
||||
'{{ item }}{% endblock %}') |
||||
assert t.render(seq=range(5)) == '[0][1][2][3][4]' |
||||
|
||||
def test_super_in_scoped_block(self): |
||||
env = Environment(loader=DictLoader({ |
||||
'master.html': '{% for item in seq %}[{% block item scoped %}' |
||||
'{{ item }}{% endblock %}]{% endfor %}' |
||||
})) |
||||
t = env.from_string('{% extends "master.html" %}{% block item %}' |
||||
'{{ super() }}|{{ item * 2 }}{% endblock %}') |
||||
assert t.render(seq=range(5)) == '[0|0][1|2][2|4][3|6][4|8]' |
||||
|
||||
def test_scoped_block_after_inheritance(self): |
||||
env = Environment(loader=DictLoader({ |
||||
'layout.html': ''' |
||||
{% block useless %}{% endblock %} |
||||
''', |
||||
'index.html': ''' |
||||
{%- extends 'layout.html' %} |
||||
{% from 'helpers.html' import foo with context %} |
||||
{% block useless %} |
||||
{% for x in [1, 2, 3] %} |
||||
{% block testing scoped %} |
||||
{{ foo(x) }} |
||||
{% endblock %} |
||||
{% endfor %} |
||||
{% endblock %} |
||||
''', |
||||
'helpers.html': ''' |
||||
{% macro foo(x) %}{{ the_foo + x }}{% endmacro %} |
||||
''' |
||||
})) |
||||
rv = env.get_template('index.html').render(the_foo=42).split() |
||||
assert rv == ['43', '44', '45'] |
||||
|
||||
|
||||
class BugFixTestCase(JinjaTestCase): |
||||
|
||||
def test_fixed_macro_scoping_bug(self): |
||||
assert Environment(loader=DictLoader({ |
||||
'test.html': '''\ |
||||
{% extends 'details.html' %} |
||||
|
||||
{% macro my_macro() %} |
||||
my_macro |
||||
{% endmacro %} |
||||
|
||||
{% block inner_box %} |
||||
{{ my_macro() }} |
||||
{% endblock %} |
||||
''', |
||||
'details.html': '''\ |
||||
{% extends 'standard.html' %} |
||||
|
||||
{% macro my_macro() %} |
||||
my_macro |
||||
{% endmacro %} |
||||
|
||||
{% block content %} |
||||
{% block outer_box %} |
||||
outer_box |
||||
{% block inner_box %} |
||||
inner_box |
||||
{% endblock %} |
||||
{% endblock %} |
||||
{% endblock %} |
||||
''', |
||||
'standard.html': ''' |
||||
{% block content %} {% endblock %} |
||||
''' |
||||
})).get_template("test.html").render().split() == [u'outer_box', u'my_macro'] |
||||
|
||||
|
||||
def suite(): |
||||
suite = unittest.TestSuite() |
||||
suite.addTest(unittest.makeSuite(InheritanceTestCase)) |
||||
suite.addTest(unittest.makeSuite(BugFixTestCase)) |
||||
return suite |
@ -1,387 +0,0 @@ |
||||
# -*- coding: utf-8 -*- |
||||
""" |
||||
jinja2.testsuite.lexnparse |
||||
~~~~~~~~~~~~~~~~~~~~~~~~~~ |
||||
|
||||
All the unittests regarding lexing, parsing and syntax. |
||||
|
||||
:copyright: (c) 2010 by the Jinja Team. |
||||
:license: BSD, see LICENSE for more details. |
||||
""" |
||||
import sys |
||||
import unittest |
||||
|
||||
from jinja2.testsuite import JinjaTestCase |
||||
|
||||
from jinja2 import Environment, Template, TemplateSyntaxError, \ |
||||
UndefinedError, nodes |
||||
|
||||
env = Environment() |
||||
|
||||
|
||||
# how does a string look like in jinja syntax? |
||||
if sys.version_info < (3, 0): |
||||
def jinja_string_repr(string): |
||||
return repr(string)[1:] |
||||
else: |
||||
jinja_string_repr = repr |
||||
|
||||
|
||||
class LexerTestCase(JinjaTestCase): |
||||
|
||||
def test_raw1(self): |
||||
tmpl = env.from_string('{% raw %}foo{% endraw %}|' |
||||
'{%raw%}{{ bar }}|{% baz %}{% endraw %}') |
||||
assert tmpl.render() == 'foo|{{ bar }}|{% baz %}' |
||||
|
||||
def test_raw2(self): |
||||
tmpl = env.from_string('1 {%- raw -%} 2 {%- endraw -%} 3') |
||||
assert tmpl.render() == '123' |
||||
|
||||
def test_balancing(self): |
||||
env = Environment('{%', '%}', '${', '}') |
||||
tmpl = env.from_string('''{% for item in seq |
||||
%}${{'foo': item}|upper}{% endfor %}''') |
||||
assert tmpl.render(seq=range(3)) == "{'FOO': 0}{'FOO': 1}{'FOO': 2}" |
||||
|
||||
def test_comments(self): |
||||
env = Environment('<!--', '-->', '{', '}') |
||||
tmpl = env.from_string('''\ |
||||
<ul> |
||||
<!--- for item in seq --> |
||||
<li>{item}</li> |
||||
<!--- endfor --> |
||||
</ul>''') |
||||
assert tmpl.render(seq=range(3)) == ("<ul>\n <li>0</li>\n " |
||||
"<li>1</li>\n <li>2</li>\n</ul>") |
||||
|
||||
def test_string_escapes(self): |
||||
for char in u'\0', u'\u2668', u'\xe4', u'\t', u'\r', u'\n': |
||||
tmpl = env.from_string('{{ %s }}' % jinja_string_repr(char)) |
||||
assert tmpl.render() == char |
||||
assert env.from_string('{{ "\N{HOT SPRINGS}" }}').render() == u'\u2668' |
||||
|
||||
def test_bytefallback(self): |
||||
from pprint import pformat |
||||
tmpl = env.from_string(u'''{{ 'foo'|pprint }}|{{ 'bär'|pprint }}''') |
||||
assert tmpl.render() == pformat('foo') + '|' + pformat(u'bär') |
||||
|
||||
def test_operators(self): |
||||
from jinja2.lexer import operators |
||||
for test, expect in operators.iteritems(): |
||||
if test in '([{}])': |
||||
continue |
||||
stream = env.lexer.tokenize('{{ %s }}' % test) |
||||
stream.next() |
||||
assert stream.current.type == expect |
||||
|
||||
def test_normalizing(self): |
||||
for seq in '\r', '\r\n', '\n': |
||||
env = Environment(newline_sequence=seq) |
||||
tmpl = env.from_string('1\n2\r\n3\n4\n') |
||||
result = tmpl.render() |
||||
assert result.replace(seq, 'X') == '1X2X3X4' |
||||
|
||||
|
||||
class ParserTestCase(JinjaTestCase): |
||||
|
||||
def test_php_syntax(self): |
||||
env = Environment('<?', '?>', '<?=', '?>', '<!--', '-->') |
||||
tmpl = env.from_string('''\ |
||||
<!-- I'm a comment, I'm not interesting -->\ |
||||
<? for item in seq -?> |
||||
<?= item ?> |
||||
<?- endfor ?>''') |
||||
assert tmpl.render(seq=range(5)) == '01234' |
||||
|
||||
def test_erb_syntax(self): |
||||
env = Environment('<%', '%>', '<%=', '%>', '<%#', '%>') |
||||
tmpl = env.from_string('''\ |
||||
<%# I'm a comment, I'm not interesting %>\ |
||||
<% for item in seq -%> |
||||
<%= item %> |
||||
<%- endfor %>''') |
||||
assert tmpl.render(seq=range(5)) == '01234' |
||||
|
||||
def test_comment_syntax(self): |
||||
env = Environment('<!--', '-->', '${', '}', '<!--#', '-->') |
||||
tmpl = env.from_string('''\ |
||||
<!--# I'm a comment, I'm not interesting -->\ |
||||
<!-- for item in seq ---> |
||||
${item} |
||||
<!--- endfor -->''') |
||||
assert tmpl.render(seq=range(5)) == '01234' |
||||
|
||||
def test_balancing(self): |
||||
tmpl = env.from_string('''{{{'foo':'bar'}.foo}}''') |
||||
assert tmpl.render() == 'bar' |
||||
|
||||
def test_start_comment(self): |
||||
tmpl = env.from_string('''{# foo comment |
||||
and bar comment #} |
||||
{% macro blub() %}foo{% endmacro %} |
||||
{{ blub() }}''') |
||||
assert tmpl.render().strip() == 'foo' |
||||
|
||||
def test_line_syntax(self): |
||||
env = Environment('<%', '%>', '${', '}', '<%#', '%>', '%') |
||||
tmpl = env.from_string('''\ |
||||
<%# regular comment %> |
||||
% for item in seq: |
||||
${item} |
||||
% endfor''') |
||||
assert [int(x.strip()) for x in tmpl.render(seq=range(5)).split()] == \ |
||||
range(5) |
||||
|
||||
env = Environment('<%', '%>', '${', '}', '<%#', '%>', '%', '##') |
||||
tmpl = env.from_string('''\ |
||||
<%# regular comment %> |
||||
% for item in seq: |
||||
${item} ## the rest of the stuff |
||||
% endfor''') |
||||
assert [int(x.strip()) for x in tmpl.render(seq=range(5)).split()] == \ |
||||
range(5) |
||||
|
||||
def test_line_syntax_priority(self): |
||||
# XXX: why is the whitespace there in front of the newline? |
||||
env = Environment('{%', '%}', '${', '}', '/*', '*/', '##', '#') |
||||
tmpl = env.from_string('''\ |
||||
/* ignore me. |
||||
I'm a multiline comment */ |
||||
## for item in seq: |
||||
* ${item} # this is just extra stuff |
||||
## endfor''') |
||||
assert tmpl.render(seq=[1, 2]).strip() == '* 1\n* 2' |
||||
env = Environment('{%', '%}', '${', '}', '/*', '*/', '#', '##') |
||||
tmpl = env.from_string('''\ |
||||
/* ignore me. |
||||
I'm a multiline comment */ |
||||
# for item in seq: |
||||
* ${item} ## this is just extra stuff |
||||
## extra stuff i just want to ignore |
||||
# endfor''') |
||||
assert tmpl.render(seq=[1, 2]).strip() == '* 1\n\n* 2' |
||||
|
||||
def test_error_messages(self): |
||||
def assert_error(code, expected): |
||||
try: |
||||
Template(code) |
||||
except TemplateSyntaxError, e: |
||||
assert str(e) == expected, 'unexpected error message' |
||||
else: |
||||
assert False, 'that was supposed to be an error' |
||||
|
||||
assert_error('{% for item in seq %}...{% endif %}', |
||||
"Encountered unknown tag 'endif'. Jinja was looking " |
||||
"for the following tags: 'endfor' or 'else'. The " |
||||
"innermost block that needs to be closed is 'for'.") |
||||
assert_error('{% if foo %}{% for item in seq %}...{% endfor %}{% endfor %}', |
||||
"Encountered unknown tag 'endfor'. Jinja was looking for " |
||||
"the following tags: 'elif' or 'else' or 'endif'. The " |
||||
"innermost block that needs to be closed is 'if'.") |
||||
assert_error('{% if foo %}', |
||||
"Unexpected end of template. Jinja was looking for the " |
||||
"following tags: 'elif' or 'else' or 'endif'. The " |
||||
"innermost block that needs to be closed is 'if'.") |
||||
assert_error('{% for item in seq %}', |
||||
"Unexpected end of template. Jinja was looking for the " |
||||
"following tags: 'endfor' or 'else'. The innermost block " |
||||
"that needs to be closed is 'for'.") |
||||
assert_error('{% block foo-bar-baz %}', |
||||
"Block names in Jinja have to be valid Python identifiers " |
||||
"and may not contain hyphens, use an underscore instead.") |
||||
assert_error('{% unknown_tag %}', |
||||
"Encountered unknown tag 'unknown_tag'.") |
||||
|
||||
|
||||
class SyntaxTestCase(JinjaTestCase): |
||||
|
||||
def test_call(self): |
||||
env = Environment() |
||||
env.globals['foo'] = lambda a, b, c, e, g: a + b + c + e + g |
||||
tmpl = env.from_string("{{ foo('a', c='d', e='f', *['b'], **{'g': 'h'}) }}") |
||||
assert tmpl.render() == 'abdfh' |
||||
|
||||
def test_slicing(self): |
||||
tmpl = env.from_string('{{ [1, 2, 3][:] }}|{{ [1, 2, 3][::-1] }}') |
||||
assert tmpl.render() == '[1, 2, 3]|[3, 2, 1]' |
||||
|
||||
def test_attr(self): |
||||
tmpl = env.from_string("{{ foo.bar }}|{{ foo['bar'] }}") |
||||
assert tmpl.render(foo={'bar': 42}) == '42|42' |
||||
|
||||
def test_subscript(self): |
||||
tmpl = env.from_string("{{ foo[0] }}|{{ foo[-1] }}") |
||||
assert tmpl.render(foo=[0, 1, 2]) == '0|2' |
||||
|
||||
def test_tuple(self): |
||||
tmpl = env.from_string('{{ () }}|{{ (1,) }}|{{ (1, 2) }}') |
||||
assert tmpl.render() == '()|(1,)|(1, 2)' |
||||
|
||||
def test_math(self): |
||||
tmpl = env.from_string('{{ (1 + 1 * 2) - 3 / 2 }}|{{ 2**3 }}') |
||||
assert tmpl.render() == '1.5|8' |
||||
|
||||
def test_div(self): |
||||
tmpl = env.from_string('{{ 3 // 2 }}|{{ 3 / 2 }}|{{ 3 % 2 }}') |
||||
assert tmpl.render() == '1|1.5|1' |
||||
|
||||
def test_unary(self): |
||||
tmpl = env.from_string('{{ +3 }}|{{ -3 }}') |
||||
assert tmpl.render() == '3|-3' |
||||
|
||||
def test_concat(self): |
||||
tmpl = env.from_string("{{ [1, 2] ~ 'foo' }}") |
||||
assert tmpl.render() == '[1, 2]foo' |
||||
|
||||
def test_compare(self): |
||||
tmpl = env.from_string('{{ 1 > 0 }}|{{ 1 >= 1 }}|{{ 2 < 3 }}|' |
||||
'{{ 2 == 2 }}|{{ 1 <= 1 }}') |
||||
assert tmpl.render() == 'True|True|True|True|True' |
||||
|
||||
def test_inop(self): |
||||
tmpl = env.from_string('{{ 1 in [1, 2, 3] }}|{{ 1 not in [1, 2, 3] }}') |
||||
assert tmpl.render() == 'True|False' |
||||
|
||||
def test_literals(self): |
||||
tmpl = env.from_string('{{ [] }}|{{ {} }}|{{ () }}') |
||||
assert tmpl.render().lower() == '[]|{}|()' |
||||
|
||||
def test_bool(self): |
||||
tmpl = env.from_string('{{ true and false }}|{{ false ' |
||||
'or true }}|{{ not false }}') |
||||
assert tmpl.render() == 'False|True|True' |
||||
|
||||
def test_grouping(self): |
||||
tmpl = env.from_string('{{ (true and false) or (false and true) and not false }}') |
||||
assert tmpl.render() == 'False' |
||||
|
||||
def test_django_attr(self): |
||||
tmpl = env.from_string('{{ [1, 2, 3].0 }}|{{ [[1]].0.0 }}') |
||||
assert tmpl.render() == '1|1' |
||||
|
||||
def test_conditional_expression(self): |
||||
tmpl = env.from_string('''{{ 0 if true else 1 }}''') |
||||
assert tmpl.render() == '0' |
||||
|
||||
def test_short_conditional_expression(self): |
||||
tmpl = env.from_string('<{{ 1 if false }}>') |
||||
assert tmpl.render() == '<>' |
||||
|
||||
tmpl = env.from_string('<{{ (1 if false).bar }}>') |
||||
self.assert_raises(UndefinedError, tmpl.render) |
||||
|
||||
def test_filter_priority(self): |
||||
tmpl = env.from_string('{{ "foo"|upper + "bar"|upper }}') |
||||
assert tmpl.render() == 'FOOBAR' |
||||
|
||||
def test_function_calls(self): |
||||
tests = [ |
||||
(True, '*foo, bar'), |
||||
(True, '*foo, *bar'), |
||||
(True, '*foo, bar=42'), |
||||
(True, '**foo, *bar'), |
||||
(True, '**foo, bar'), |
||||
(False, 'foo, bar'), |
||||
(False, 'foo, bar=42'), |
||||
(False, 'foo, bar=23, *args'), |
||||
(False, 'a, b=c, *d, **e'), |
||||
(False, '*foo, **bar') |
||||
] |
||||
for should_fail, sig in tests: |
||||
if should_fail: |
||||
self.assert_raises(TemplateSyntaxError, |
||||
env.from_string, '{{ foo(%s) }}' % sig) |
||||
else: |
||||
env.from_string('foo(%s)' % sig) |
||||
|
||||
def test_tuple_expr(self): |
||||
for tmpl in [ |
||||
'{{ () }}', |
||||
'{{ (1, 2) }}', |
||||
'{{ (1, 2,) }}', |
||||
'{{ 1, }}', |
||||
'{{ 1, 2 }}', |
||||
'{% for foo, bar in seq %}...{% endfor %}', |
||||
'{% for x in foo, bar %}...{% endfor %}', |
||||
'{% for x in foo, %}...{% endfor %}' |
||||
]: |
||||
assert env.from_string(tmpl) |
||||
|
||||
def test_trailing_comma(self): |
||||
tmpl = env.from_string('{{ (1, 2,) }}|{{ [1, 2,] }}|{{ {1: 2,} }}') |
||||
assert tmpl.render().lower() == '(1, 2)|[1, 2]|{1: 2}' |
||||
|
||||
def test_block_end_name(self): |
||||
env.from_string('{% block foo %}...{% endblock foo %}') |
||||
self.assert_raises(TemplateSyntaxError, env.from_string, |
||||
'{% block x %}{% endblock y %}') |
||||
|
||||
def test_constant_casing(self): |
||||
for const in True, False, None: |
||||
tmpl = env.from_string('{{ %s }}|{{ %s }}|{{ %s }}' % ( |
||||
str(const), str(const).lower(), str(const).upper() |
||||
)) |
||||
assert tmpl.render() == '%s|%s|' % (const, const) |
||||
|
||||
def test_test_chaining(self): |
||||
self.assert_raises(TemplateSyntaxError, env.from_string, |
||||
'{{ foo is string is sequence }}') |
||||
assert env.from_string('{{ 42 is string or 42 is number }}' |
||||
).render() == 'True' |
||||
|
||||
def test_string_concatenation(self): |
||||
tmpl = env.from_string('{{ "foo" "bar" "baz" }}') |
||||
assert tmpl.render() == 'foobarbaz' |
||||
|
||||
def test_notin(self): |
||||
bar = xrange(100) |
||||
tmpl = env.from_string('''{{ not 42 in bar }}''') |
||||
assert tmpl.render(bar=bar) == unicode(not 42 in bar) |
||||
|
||||
def test_implicit_subscribed_tuple(self): |
||||
class Foo(object): |
||||
def __getitem__(self, x): |
||||
return x |
||||
t = env.from_string('{{ foo[1, 2] }}') |
||||
assert t.render(foo=Foo()) == u'(1, 2)' |
||||
|
||||
def test_raw2(self): |
||||
tmpl = env.from_string('{% raw %}{{ FOO }} and {% BAR %}{% endraw %}') |
||||
assert tmpl.render() == '{{ FOO }} and {% BAR %}' |
||||
|
||||
def test_const(self): |
||||
tmpl = env.from_string('{{ true }}|{{ false }}|{{ none }}|' |
||||
'{{ none is defined }}|{{ missing is defined }}') |
||||
assert tmpl.render() == 'True|False|None|True|False' |
||||
|
||||
def test_neg_filter_priority(self): |
||||
node = env.parse('{{ -1|foo }}') |
||||
assert isinstance(node.body[0].nodes[0], nodes.Filter) |
||||
assert isinstance(node.body[0].nodes[0].node, nodes.Neg) |
||||
|
||||
def test_const_assign(self): |
||||
constass1 = '''{% set true = 42 %}''' |
||||
constass2 = '''{% for none in seq %}{% endfor %}''' |
||||
for tmpl in constass1, constass2: |
||||
self.assert_raises(TemplateSyntaxError, env.from_string, tmpl) |
||||
|
||||
def test_localset(self): |
||||
tmpl = env.from_string('''{% set foo = 0 %}\ |
||||
{% for item in [1, 2] %}{% set foo = 1 %}{% endfor %}\ |
||||
{{ foo }}''') |
||||
assert tmpl.render() == '0' |
||||
|
||||
def test_parse_unary(self): |
||||
tmpl = env.from_string('{{ -foo["bar"] }}') |
||||
assert tmpl.render(foo={'bar': 42}) == '-42' |
||||
tmpl = env.from_string('{{ -foo["bar"]|abs }}') |
||||
assert tmpl.render(foo={'bar': 42}) == '42' |
||||
|
||||
|
||||
def suite(): |
||||
suite = unittest.TestSuite() |
||||
suite.addTest(unittest.makeSuite(LexerTestCase)) |
||||
suite.addTest(unittest.makeSuite(ParserTestCase)) |
||||
suite.addTest(unittest.makeSuite(SyntaxTestCase)) |
||||
return suite |
@ -1,218 +0,0 @@ |
||||
# -*- coding: utf-8 -*- |
||||
""" |
||||
jinja2.testsuite.loader |
||||
~~~~~~~~~~~~~~~~~~~~~~~ |
||||
|
||||
Test the loaders. |
||||
|
||||
:copyright: (c) 2010 by the Jinja Team. |
||||
:license: BSD, see LICENSE for more details. |
||||
""" |
||||
import os |
||||
import sys |
||||
import tempfile |
||||
import shutil |
||||
import unittest |
||||
|
||||
from jinja2.testsuite import JinjaTestCase, dict_loader, \ |
||||
package_loader, filesystem_loader, function_loader, \ |
||||
choice_loader, prefix_loader |
||||
|
||||
from jinja2 import Environment, loaders |
||||
from jinja2.loaders import split_template_path |
||||
from jinja2.exceptions import TemplateNotFound |
||||
|
||||
|
||||
class LoaderTestCase(JinjaTestCase): |
||||
|
||||
def test_dict_loader(self): |
||||
env = Environment(loader=dict_loader) |
||||
tmpl = env.get_template('justdict.html') |
||||
assert tmpl.render().strip() == 'FOO' |
||||
self.assert_raises(TemplateNotFound, env.get_template, 'missing.html') |
||||
|
||||
def test_package_loader(self): |
||||
env = Environment(loader=package_loader) |
||||
tmpl = env.get_template('test.html') |
||||
assert tmpl.render().strip() == 'BAR' |
||||
self.assert_raises(TemplateNotFound, env.get_template, 'missing.html') |
||||
|
||||
def test_filesystem_loader(self): |
||||
env = Environment(loader=filesystem_loader) |
||||
tmpl = env.get_template('test.html') |
||||
assert tmpl.render().strip() == 'BAR' |
||||
tmpl = env.get_template('foo/test.html') |
||||
assert tmpl.render().strip() == 'FOO' |
||||
self.assert_raises(TemplateNotFound, env.get_template, 'missing.html') |
||||
|
||||
def test_choice_loader(self): |
||||
env = Environment(loader=choice_loader) |
||||
tmpl = env.get_template('justdict.html') |
||||
assert tmpl.render().strip() == 'FOO' |
||||
tmpl = env.get_template('test.html') |
||||
assert tmpl.render().strip() == 'BAR' |
||||
self.assert_raises(TemplateNotFound, env.get_template, 'missing.html') |
||||
|
||||
def test_function_loader(self): |
||||
env = Environment(loader=function_loader) |
||||
tmpl = env.get_template('justfunction.html') |
||||
assert tmpl.render().strip() == 'FOO' |
||||
self.assert_raises(TemplateNotFound, env.get_template, 'missing.html') |
||||
|
||||
def test_prefix_loader(self): |
||||
env = Environment(loader=prefix_loader) |
||||
tmpl = env.get_template('a/test.html') |
||||
assert tmpl.render().strip() == 'BAR' |
||||
tmpl = env.get_template('b/justdict.html') |
||||
assert tmpl.render().strip() == 'FOO' |
||||
self.assert_raises(TemplateNotFound, env.get_template, 'missing') |
||||
|
||||
def test_caching(self): |
||||
changed = False |
||||
class TestLoader(loaders.BaseLoader): |
||||
def get_source(self, environment, template): |
||||
return u'foo', None, lambda: not changed |
||||
env = Environment(loader=TestLoader(), cache_size=-1) |
||||
tmpl = env.get_template('template') |
||||
assert tmpl is env.get_template('template') |
||||
changed = True |
||||
assert tmpl is not env.get_template('template') |
||||
changed = False |
||||
|
||||
env = Environment(loader=TestLoader(), cache_size=0) |
||||
assert env.get_template('template') \ |
||||
is not env.get_template('template') |
||||
|
||||
env = Environment(loader=TestLoader(), cache_size=2) |
||||
t1 = env.get_template('one') |
||||
t2 = env.get_template('two') |
||||
assert t2 is env.get_template('two') |
||||
assert t1 is env.get_template('one') |
||||
t3 = env.get_template('three') |
||||
assert 'one' in env.cache |
||||
assert 'two' not in env.cache |
||||
assert 'three' in env.cache |
||||
|
||||
def test_split_template_path(self): |
||||
assert split_template_path('foo/bar') == ['foo', 'bar'] |
||||
assert split_template_path('./foo/bar') == ['foo', 'bar'] |
||||
self.assert_raises(TemplateNotFound, split_template_path, '../foo') |
||||
|
||||
|
||||
class ModuleLoaderTestCase(JinjaTestCase): |
||||
archive = None |
||||
|
||||
def compile_down(self, zip='deflated', py_compile=False): |
||||
super(ModuleLoaderTestCase, self).setup() |
||||
log = [] |
||||
self.reg_env = Environment(loader=prefix_loader) |
||||
if zip is not None: |
||||
self.archive = tempfile.mkstemp(suffix='.zip')[1] |
||||
else: |
||||
self.archive = tempfile.mkdtemp() |
||||
self.reg_env.compile_templates(self.archive, zip=zip, |
||||
log_function=log.append, |
||||
py_compile=py_compile) |
||||
self.mod_env = Environment(loader=loaders.ModuleLoader(self.archive)) |
||||
return ''.join(log) |
||||
|
||||
def teardown(self): |
||||
super(ModuleLoaderTestCase, self).teardown() |
||||
if hasattr(self, 'mod_env'): |
||||
if os.path.isfile(self.archive): |
||||
os.remove(self.archive) |
||||
else: |
||||
shutil.rmtree(self.archive) |
||||
self.archive = None |
||||
|
||||
def test_log(self): |
||||
log = self.compile_down() |
||||
assert 'Compiled "a/foo/test.html" as ' \ |
||||
'tmpl_a790caf9d669e39ea4d280d597ec891c4ef0404a' in log |
||||
assert 'Finished compiling templates' in log |
||||
assert 'Could not compile "a/syntaxerror.html": ' \ |
||||
'Encountered unknown tag \'endif\'' in log |
||||
|
||||
def _test_common(self): |
||||
tmpl1 = self.reg_env.get_template('a/test.html') |
||||
tmpl2 = self.mod_env.get_template('a/test.html') |
||||
assert tmpl1.render() == tmpl2.render() |
||||
|
||||
tmpl1 = self.reg_env.get_template('b/justdict.html') |
||||
tmpl2 = self.mod_env.get_template('b/justdict.html') |
||||
assert tmpl1.render() == tmpl2.render() |
||||
|
||||
def test_deflated_zip_compile(self): |
||||
self.compile_down(zip='deflated') |
||||
self._test_common() |
||||
|
||||
def test_stored_zip_compile(self): |
||||
self.compile_down(zip='stored') |
||||
self._test_common() |
||||
|
||||
def test_filesystem_compile(self): |
||||
self.compile_down(zip=None) |
||||
self._test_common() |
||||
|
||||
def test_weak_references(self): |
||||
self.compile_down() |
||||
tmpl = self.mod_env.get_template('a/test.html') |
||||
key = loaders.ModuleLoader.get_template_key('a/test.html') |
||||
name = self.mod_env.loader.module.__name__ |
||||
|
||||
assert hasattr(self.mod_env.loader.module, key) |
||||
assert name in sys.modules |
||||
|
||||
# unset all, ensure the module is gone from sys.modules |
||||
self.mod_env = tmpl = None |
||||
|
||||
try: |
||||
import gc |
||||
gc.collect() |
||||
except: |
||||
pass |
||||
|
||||
assert name not in sys.modules |
||||
|
||||
def test_byte_compilation(self): |
||||
log = self.compile_down(py_compile=True) |
||||
assert 'Byte-compiled "a/test.html"' in log |
||||
tmpl1 = self.mod_env.get_template('a/test.html') |
||||
mod = self.mod_env.loader.module. \ |
||||
tmpl_3c4ddf650c1a73df961a6d3d2ce2752f1b8fd490 |
||||
assert mod.__file__.endswith('.pyc') |
||||
|
||||
def test_choice_loader(self): |
||||
log = self.compile_down(py_compile=True) |
||||
assert 'Byte-compiled "a/test.html"' in log |
||||
|
||||
self.mod_env.loader = loaders.ChoiceLoader([ |
||||
self.mod_env.loader, |
||||
loaders.DictLoader({'DICT_SOURCE': 'DICT_TEMPLATE'}) |
||||
]) |
||||
|
||||
tmpl1 = self.mod_env.get_template('a/test.html') |
||||
self.assert_equal(tmpl1.render(), 'BAR') |
||||
tmpl2 = self.mod_env.get_template('DICT_SOURCE') |
||||
self.assert_equal(tmpl2.render(), 'DICT_TEMPLATE') |
||||
|
||||
def test_prefix_loader(self): |
||||
log = self.compile_down(py_compile=True) |
||||
assert 'Byte-compiled "a/test.html"' in log |
||||
|
||||
self.mod_env.loader = loaders.PrefixLoader({ |
||||
'MOD': self.mod_env.loader, |
||||
'DICT': loaders.DictLoader({'test.html': 'DICT_TEMPLATE'}) |
||||
}) |
||||
|
||||
tmpl1 = self.mod_env.get_template('MOD/a/test.html') |
||||
self.assert_equal(tmpl1.render(), 'BAR') |
||||
tmpl2 = self.mod_env.get_template('DICT/test.html') |
||||
self.assert_equal(tmpl2.render(), 'DICT_TEMPLATE') |
||||
|
||||
|
||||
def suite(): |
||||
suite = unittest.TestSuite() |
||||
suite.addTest(unittest.makeSuite(LoaderTestCase)) |
||||
suite.addTest(unittest.makeSuite(ModuleLoaderTestCase)) |
||||
return suite |
@ -1,255 +0,0 @@ |
||||
# -*- coding: utf-8 -*- |
||||
""" |
||||
jinja2.testsuite.regression |
||||
~~~~~~~~~~~~~~~~~~~~~~~~~~~ |
||||
|
||||
Tests corner cases and bugs. |
||||
|
||||
:copyright: (c) 2010 by the Jinja Team. |
||||
:license: BSD, see LICENSE for more details. |
||||
""" |
||||
import unittest |
||||
|
||||
from jinja2.testsuite import JinjaTestCase |
||||
|
||||
from jinja2 import Template, Environment, DictLoader, TemplateSyntaxError, \ |
||||
TemplateNotFound, PrefixLoader |
||||
|
||||
env = Environment() |
||||
|
||||
|
||||
class CornerTestCase(JinjaTestCase): |
||||
|
||||
def test_assigned_scoping(self): |
||||
t = env.from_string(''' |
||||
{%- for item in (1, 2, 3, 4) -%} |
||||
[{{ item }}] |
||||
{%- endfor %} |
||||
{{- item -}} |
||||
''') |
||||
assert t.render(item=42) == '[1][2][3][4]42' |
||||
|
||||
t = env.from_string(''' |
||||
{%- for item in (1, 2, 3, 4) -%} |
||||
[{{ item }}] |
||||
{%- endfor %} |
||||
{%- set item = 42 %} |
||||
{{- item -}} |
||||
''') |
||||
assert t.render() == '[1][2][3][4]42' |
||||
|
||||
t = env.from_string(''' |
||||
{%- set item = 42 %} |
||||
{%- for item in (1, 2, 3, 4) -%} |
||||
[{{ item }}] |
||||
{%- endfor %} |
||||
{{- item -}} |
||||
''') |
||||
assert t.render() == '[1][2][3][4]42' |
||||
|
||||
def test_closure_scoping(self): |
||||
t = env.from_string(''' |
||||
{%- set wrapper = "<FOO>" %} |
||||
{%- for item in (1, 2, 3, 4) %} |
||||
{%- macro wrapper() %}[{{ item }}]{% endmacro %} |
||||
{{- wrapper() }} |
||||
{%- endfor %} |
||||
{{- wrapper -}} |
||||
''') |
||||
assert t.render() == '[1][2][3][4]<FOO>' |
||||
|
||||
t = env.from_string(''' |
||||
{%- for item in (1, 2, 3, 4) %} |
||||
{%- macro wrapper() %}[{{ item }}]{% endmacro %} |
||||
{{- wrapper() }} |
||||
{%- endfor %} |
||||
{%- set wrapper = "<FOO>" %} |
||||
{{- wrapper -}} |
||||
''') |
||||
assert t.render() == '[1][2][3][4]<FOO>' |
||||
|
||||
t = env.from_string(''' |
||||
{%- for item in (1, 2, 3, 4) %} |
||||
{%- macro wrapper() %}[{{ item }}]{% endmacro %} |
||||
{{- wrapper() }} |
||||
{%- endfor %} |
||||
{{- wrapper -}} |
||||
''') |
||||
assert t.render(wrapper=23) == '[1][2][3][4]23' |
||||
|
||||
|
||||
class BugTestCase(JinjaTestCase): |
||||
|
||||
def test_keyword_folding(self): |
||||
env = Environment() |
||||
env.filters['testing'] = lambda value, some: value + some |
||||
assert env.from_string("{{ 'test'|testing(some='stuff') }}") \ |
||||
.render() == 'teststuff' |
||||
|
||||
def test_extends_output_bugs(self): |
||||
env = Environment(loader=DictLoader({ |
||||
'parent.html': '(({% block title %}{% endblock %}))' |
||||
})) |
||||
|
||||
t = env.from_string('{% if expr %}{% extends "parent.html" %}{% endif %}' |
||||
'[[{% block title %}title{% endblock %}]]' |
||||
'{% for item in [1, 2, 3] %}({{ item }}){% endfor %}') |
||||
assert t.render(expr=False) == '[[title]](1)(2)(3)' |
||||
assert t.render(expr=True) == '((title))' |
||||
|
||||
def test_urlize_filter_escaping(self): |
||||
tmpl = env.from_string('{{ "http://www.example.org/<foo"|urlize }}') |
||||
assert tmpl.render() == '<a href="http://www.example.org/<foo">http://www.example.org/<foo</a>' |
||||
|
||||
def test_loop_call_loop(self): |
||||
tmpl = env.from_string(''' |
||||
|
||||
{% macro test() %} |
||||
{{ caller() }} |
||||
{% endmacro %} |
||||
|
||||
{% for num1 in range(5) %} |
||||
{% call test() %} |
||||
{% for num2 in range(10) %} |
||||
{{ loop.index }} |
||||
{% endfor %} |
||||
{% endcall %} |
||||
{% endfor %} |
||||
|
||||
''') |
||||
|
||||
assert tmpl.render().split() == map(unicode, range(1, 11)) * 5 |
||||
|
||||
def test_weird_inline_comment(self): |
||||
env = Environment(line_statement_prefix='%') |
||||
self.assert_raises(TemplateSyntaxError, env.from_string, |
||||
'% for item in seq {# missing #}\n...% endfor') |
||||
|
||||
def test_old_macro_loop_scoping_bug(self): |
||||
tmpl = env.from_string('{% for i in (1, 2) %}{{ i }}{% endfor %}' |
||||
'{% macro i() %}3{% endmacro %}{{ i() }}') |
||||
assert tmpl.render() == '123' |
||||
|
||||
def test_partial_conditional_assignments(self): |
||||
tmpl = env.from_string('{% if b %}{% set a = 42 %}{% endif %}{{ a }}') |
||||
assert tmpl.render(a=23) == '23' |
||||
assert tmpl.render(b=True) == '42' |
||||
|
||||
def test_stacked_locals_scoping_bug(self): |
||||
env = Environment(line_statement_prefix='#') |
||||
t = env.from_string('''\ |
||||
# for j in [1, 2]: |
||||
# set x = 1 |
||||
# for i in [1, 2]: |
||||
# print x |
||||
# if i % 2 == 0: |
||||
# set x = x + 1 |
||||
# endif |
||||
# endfor |
||||
# endfor |
||||
# if a |
||||
# print 'A' |
||||
# elif b |
||||
# print 'B' |
||||
# elif c == d |
||||
# print 'C' |
||||
# else |
||||
# print 'D' |
||||
# endif |
||||
''') |
||||
assert t.render(a=0, b=False, c=42, d=42.0) == '1111C' |
||||
|
||||
def test_stacked_locals_scoping_bug_twoframe(self): |
||||
t = Template(''' |
||||
{% set x = 1 %} |
||||
{% for item in foo %} |
||||
{% if item == 1 %} |
||||
{% set x = 2 %} |
||||
{% endif %} |
||||
{% endfor %} |
||||
{{ x }} |
||||
''') |
||||
rv = t.render(foo=[1]).strip() |
||||
assert rv == u'1' |
||||
|
||||
def test_call_with_args(self): |
||||
t = Template("""{% macro dump_users(users) -%} |
||||
<ul> |
||||
{%- for user in users -%} |
||||
<li><p>{{ user.username|e }}</p>{{ caller(user) }}</li> |
||||
{%- endfor -%} |
||||
</ul> |
||||
{%- endmacro -%} |
||||
|
||||
{% call(user) dump_users(list_of_user) -%} |
||||
<dl> |
||||
<dl>Realname</dl> |
||||
<dd>{{ user.realname|e }}</dd> |
||||
<dl>Description</dl> |
||||
<dd>{{ user.description }}</dd> |
||||
</dl> |
||||
{% endcall %}""") |
||||
|
||||
assert [x.strip() for x in t.render(list_of_user=[{ |
||||
'username':'apo', |
||||
'realname':'something else', |
||||
'description':'test' |
||||
}]).splitlines()] == [ |
||||
u'<ul><li><p>apo</p><dl>', |
||||
u'<dl>Realname</dl>', |
||||
u'<dd>something else</dd>', |
||||
u'<dl>Description</dl>', |
||||
u'<dd>test</dd>', |
||||
u'</dl>', |
||||
u'</li></ul>' |
||||
] |
||||
|
||||
def test_empty_if_condition_fails(self): |
||||
self.assert_raises(TemplateSyntaxError, Template, '{% if %}....{% endif %}') |
||||
self.assert_raises(TemplateSyntaxError, Template, '{% if foo %}...{% elif %}...{% endif %}') |
||||
self.assert_raises(TemplateSyntaxError, Template, '{% for x in %}..{% endfor %}') |
||||
|
||||
def test_recursive_loop_bug(self): |
||||
tpl1 = Template(""" |
||||
{% for p in foo recursive%} |
||||
{{p.bar}} |
||||
{% for f in p.fields recursive%} |
||||
{{f.baz}} |
||||
{{p.bar}} |
||||
{% if f.rec %} |
||||
{{ loop(f.sub) }} |
||||
{% endif %} |
||||
{% endfor %} |
||||
{% endfor %} |
||||
""") |
||||
|
||||
tpl2 = Template(""" |
||||
{% for p in foo%} |
||||
{{p.bar}} |
||||
{% for f in p.fields recursive%} |
||||
{{f.baz}} |
||||
{{p.bar}} |
||||
{% if f.rec %} |
||||
{{ loop(f.sub) }} |
||||
{% endif %} |
||||
{% endfor %} |
||||
{% endfor %} |
||||
""") |
||||
|
||||
def test_correct_prefix_loader_name(self): |
||||
env = Environment(loader=PrefixLoader({ |
||||
'foo': DictLoader({}) |
||||
})) |
||||
try: |
||||
env.get_template('foo/bar.html') |
||||
except TemplateNotFound, e: |
||||
assert e.name == 'foo/bar.html' |
||||
else: |
||||
assert False, 'expected error here' |
||||
|
||||
|
||||
def suite(): |
||||
suite = unittest.TestSuite() |
||||
suite.addTest(unittest.makeSuite(CornerTestCase)) |
||||
suite.addTest(unittest.makeSuite(BugTestCase)) |
||||
return suite |
@ -1,3 +0,0 @@ |
||||
Before |
||||
{{ fail() }} |
||||
After |
@ -1 +0,0 @@ |
||||
FOO |
@ -1,4 +0,0 @@ |
||||
Foo |
||||
{% for item in broken %} |
||||
... |
||||
{% endif %} |
@ -1 +0,0 @@ |
||||
BAR |
@ -1,165 +0,0 @@ |
||||
# -*- coding: utf-8 -*- |
||||
""" |
||||
jinja2.testsuite.security |
||||
~~~~~~~~~~~~~~~~~~~~~~~~~ |
||||
|
||||
Checks the sandbox and other security features. |
||||
|
||||
:copyright: (c) 2010 by the Jinja Team. |
||||
:license: BSD, see LICENSE for more details. |
||||
""" |
||||
import unittest |
||||
|
||||
from jinja2.testsuite import JinjaTestCase |
||||
|
||||
from jinja2 import Environment |
||||
from jinja2.sandbox import SandboxedEnvironment, \ |
||||
ImmutableSandboxedEnvironment, unsafe |
||||
from jinja2 import Markup, escape |
||||
from jinja2.exceptions import SecurityError, TemplateSyntaxError, \ |
||||
TemplateRuntimeError |
||||
|
||||
|
||||
class PrivateStuff(object): |
||||
|
||||
def bar(self): |
||||
return 23 |
||||
|
||||
@unsafe |
||||
def foo(self): |
||||
return 42 |
||||
|
||||
def __repr__(self): |
||||
return 'PrivateStuff' |
||||
|
||||
|
||||
class PublicStuff(object): |
||||
bar = lambda self: 23 |
||||
_foo = lambda self: 42 |
||||
|
||||
def __repr__(self): |
||||
return 'PublicStuff' |
||||
|
||||
|
||||
class SandboxTestCase(JinjaTestCase): |
||||
|
||||
def test_unsafe(self): |
||||
env = SandboxedEnvironment() |
||||
self.assert_raises(SecurityError, env.from_string("{{ foo.foo() }}").render, |
||||
foo=PrivateStuff()) |
||||
self.assert_equal(env.from_string("{{ foo.bar() }}").render(foo=PrivateStuff()), '23') |
||||
|
||||
self.assert_raises(SecurityError, env.from_string("{{ foo._foo() }}").render, |
||||
foo=PublicStuff()) |
||||
self.assert_equal(env.from_string("{{ foo.bar() }}").render(foo=PublicStuff()), '23') |
||||
self.assert_equal(env.from_string("{{ foo.__class__ }}").render(foo=42), '') |
||||
self.assert_equal(env.from_string("{{ foo.func_code }}").render(foo=lambda:None), '') |
||||
# security error comes from __class__ already. |
||||
self.assert_raises(SecurityError, env.from_string( |
||||
"{{ foo.__class__.__subclasses__() }}").render, foo=42) |
||||
|
||||
def test_immutable_environment(self): |
||||
env = ImmutableSandboxedEnvironment() |
||||
self.assert_raises(SecurityError, env.from_string( |
||||
'{{ [].append(23) }}').render) |
||||
self.assert_raises(SecurityError, env.from_string( |
||||
'{{ {1:2}.clear() }}').render) |
||||
|
||||
def test_restricted(self): |
||||
env = SandboxedEnvironment() |
||||
self.assert_raises(TemplateSyntaxError, env.from_string, |
||||
"{% for item.attribute in seq %}...{% endfor %}") |
||||
self.assert_raises(TemplateSyntaxError, env.from_string, |
||||
"{% for foo, bar.baz in seq %}...{% endfor %}") |
||||
|
||||
def test_markup_operations(self): |
||||
# adding two strings should escape the unsafe one |
||||
unsafe = '<script type="application/x-some-script">alert("foo");</script>' |
||||
safe = Markup('<em>username</em>') |
||||
assert unsafe + safe == unicode(escape(unsafe)) + unicode(safe) |
||||
|
||||
# string interpolations are safe to use too |
||||
assert Markup('<em>%s</em>') % '<bad user>' == \ |
||||
'<em><bad user></em>' |
||||
assert Markup('<em>%(username)s</em>') % { |
||||
'username': '<bad user>' |
||||
} == '<em><bad user></em>' |
||||
|
||||
# an escaped object is markup too |
||||
assert type(Markup('foo') + 'bar') is Markup |
||||
|
||||
# and it implements __html__ by returning itself |
||||
x = Markup("foo") |
||||
assert x.__html__() is x |
||||
|
||||
# it also knows how to treat __html__ objects |
||||
class Foo(object): |
||||
def __html__(self): |
||||
return '<em>awesome</em>' |
||||
def __unicode__(self): |
||||
return 'awesome' |
||||
assert Markup(Foo()) == '<em>awesome</em>' |
||||
assert Markup('<strong>%s</strong>') % Foo() == \ |
||||
'<strong><em>awesome</em></strong>' |
||||
|
||||
# escaping and unescaping |
||||
assert escape('"<>&\'') == '"<>&'' |
||||
assert Markup("<em>Foo & Bar</em>").striptags() == "Foo & Bar" |
||||
assert Markup("<test>").unescape() == "<test>" |
||||
|
||||
def test_template_data(self): |
||||
env = Environment(autoescape=True) |
||||
t = env.from_string('{% macro say_hello(name) %}' |
||||
'<p>Hello {{ name }}!</p>{% endmacro %}' |
||||
'{{ say_hello("<blink>foo</blink>") }}') |
||||
escaped_out = '<p>Hello <blink>foo</blink>!</p>' |
||||
assert t.render() == escaped_out |
||||
assert unicode(t.module) == escaped_out |
||||
assert escape(t.module) == escaped_out |
||||
assert t.module.say_hello('<blink>foo</blink>') == escaped_out |
||||
assert escape(t.module.say_hello('<blink>foo</blink>')) == escaped_out |
||||
|
||||
def test_attr_filter(self): |
||||
env = SandboxedEnvironment() |
||||
tmpl = env.from_string('{{ cls|attr("__subclasses__")() }}') |
||||
self.assert_raises(SecurityError, tmpl.render, cls=int) |
||||
|
||||
def test_binary_operator_intercepting(self): |
||||
def disable_op(left, right): |
||||
raise TemplateRuntimeError('that operator so does not work') |
||||
for expr, ctx, rv in ('1 + 2', {}, '3'), ('a + 2', {'a': 2}, '4'): |
||||
env = SandboxedEnvironment() |
||||
env.binop_table['+'] = disable_op |
||||
t = env.from_string('{{ %s }}' % expr) |
||||
assert t.render(ctx) == rv |
||||
env.intercepted_binops = frozenset(['+']) |
||||
t = env.from_string('{{ %s }}' % expr) |
||||
try: |
||||
t.render(ctx) |
||||
except TemplateRuntimeError, e: |
||||
pass |
||||
else: |
||||
self.fail('expected runtime error') |
||||
|
||||
def test_unary_operator_intercepting(self): |
||||
def disable_op(arg): |
||||
raise TemplateRuntimeError('that operator so does not work') |
||||
for expr, ctx, rv in ('-1', {}, '-1'), ('-a', {'a': 2}, '-2'): |
||||
env = SandboxedEnvironment() |
||||
env.unop_table['-'] = disable_op |
||||
t = env.from_string('{{ %s }}' % expr) |
||||
assert t.render(ctx) == rv |
||||
env.intercepted_unops = frozenset(['-']) |
||||
t = env.from_string('{{ %s }}' % expr) |
||||
try: |
||||
t.render(ctx) |
||||
except TemplateRuntimeError, e: |
||||
pass |
||||
else: |
||||
self.fail('expected runtime error') |
||||
|
||||
|
||||
def suite(): |
||||
suite = unittest.TestSuite() |
||||
suite.addTest(unittest.makeSuite(SandboxTestCase)) |
||||
return suite |
@ -1,93 +0,0 @@ |
||||
# -*- coding: utf-8 -*- |
||||
""" |
||||
jinja2.testsuite.tests |
||||
~~~~~~~~~~~~~~~~~~~~~~ |
||||
|
||||
Who tests the tests? |
||||
|
||||
:copyright: (c) 2010 by the Jinja Team. |
||||
:license: BSD, see LICENSE for more details. |
||||
""" |
||||
import unittest |
||||
from jinja2.testsuite import JinjaTestCase |
||||
|
||||
from jinja2 import Markup, Environment |
||||
|
||||
env = Environment() |
||||
|
||||
|
||||
class TestsTestCase(JinjaTestCase): |
||||
|
||||
def test_defined(self): |
||||
tmpl = env.from_string('{{ missing is defined }}|{{ true is defined }}') |
||||
assert tmpl.render() == 'False|True' |
||||
|
||||
def test_even(self): |
||||
tmpl = env.from_string('''{{ 1 is even }}|{{ 2 is even }}''') |
||||
assert tmpl.render() == 'False|True' |
||||
|
||||
def test_odd(self): |
||||
tmpl = env.from_string('''{{ 1 is odd }}|{{ 2 is odd }}''') |
||||
assert tmpl.render() == 'True|False' |
||||
|
||||
def test_lower(self): |
||||
tmpl = env.from_string('''{{ "foo" is lower }}|{{ "FOO" is lower }}''') |
||||
assert tmpl.render() == 'True|False' |
||||
|
||||
def test_typechecks(self): |
||||
tmpl = env.from_string(''' |
||||
{{ 42 is undefined }} |
||||
{{ 42 is defined }} |
||||
{{ 42 is none }} |
||||
{{ none is none }} |
||||
{{ 42 is number }} |
||||
{{ 42 is string }} |
||||
{{ "foo" is string }} |
||||
{{ "foo" is sequence }} |
||||
{{ [1] is sequence }} |
||||
{{ range is callable }} |
||||
{{ 42 is callable }} |
||||
{{ range(5) is iterable }} |
||||
{{ {} is mapping }} |
||||
{{ mydict is mapping }} |
||||
{{ [] is mapping }} |
||||
''') |
||||
class MyDict(dict): |
||||
pass |
||||
assert tmpl.render(mydict=MyDict()).split() == [ |
||||
'False', 'True', 'False', 'True', 'True', 'False', |
||||
'True', 'True', 'True', 'True', 'False', 'True', |
||||
'True', 'True', 'False' |
||||
] |
||||
|
||||
def test_sequence(self): |
||||
tmpl = env.from_string( |
||||
'{{ [1, 2, 3] is sequence }}|' |
||||
'{{ "foo" is sequence }}|' |
||||
'{{ 42 is sequence }}' |
||||
) |
||||
assert tmpl.render() == 'True|True|False' |
||||
|
||||
def test_upper(self): |
||||
tmpl = env.from_string('{{ "FOO" is upper }}|{{ "foo" is upper }}') |
||||
assert tmpl.render() == 'True|False' |
||||
|
||||
def test_sameas(self): |
||||
tmpl = env.from_string('{{ foo is sameas false }}|' |
||||
'{{ 0 is sameas false }}') |
||||
assert tmpl.render(foo=False) == 'True|False' |
||||
|
||||
def test_no_paren_for_arg1(self): |
||||
tmpl = env.from_string('{{ foo is sameas none }}') |
||||
assert tmpl.render(foo=None) == 'True' |
||||
|
||||
def test_escaped(self): |
||||
env = Environment(autoescape=True) |
||||
tmpl = env.from_string('{{ x is escaped }}|{{ y is escaped }}') |
||||
assert tmpl.render(x='foo', y=Markup('foo')) == 'False|True' |
||||
|
||||
|
||||
def suite(): |
||||
suite = unittest.TestSuite() |
||||
suite.addTest(unittest.makeSuite(TestsTestCase)) |
||||
return suite |
@ -1,82 +0,0 @@ |
||||
# -*- coding: utf-8 -*- |
||||
""" |
||||
jinja2.testsuite.utils |
||||
~~~~~~~~~~~~~~~~~~~~~~ |
||||
|
||||
Tests utilities jinja uses. |
||||
|
||||
:copyright: (c) 2010 by the Jinja Team. |
||||
:license: BSD, see LICENSE for more details. |
||||
""" |
||||
import gc |
||||
import unittest |
||||
|
||||
import pickle |
||||
|
||||
from jinja2.testsuite import JinjaTestCase |
||||
|
||||
from jinja2.utils import LRUCache, escape, object_type_repr |
||||
|
||||
|
||||
class LRUCacheTestCase(JinjaTestCase): |
||||
|
||||
def test_simple(self): |
||||
d = LRUCache(3) |
||||
d["a"] = 1 |
||||
d["b"] = 2 |
||||
d["c"] = 3 |
||||
d["a"] |
||||
d["d"] = 4 |
||||
assert len(d) == 3 |
||||
assert 'a' in d and 'c' in d and 'd' in d and 'b' not in d |
||||
|
||||
def test_pickleable(self): |
||||
cache = LRUCache(2) |
||||
cache["foo"] = 42 |
||||
cache["bar"] = 23 |
||||
cache["foo"] |
||||
|
||||
for protocol in range(3): |
||||
copy = pickle.loads(pickle.dumps(cache, protocol)) |
||||
assert copy.capacity == cache.capacity |
||||
assert copy._mapping == cache._mapping |
||||
assert copy._queue == cache._queue |
||||
|
||||
|
||||
class HelpersTestCase(JinjaTestCase): |
||||
|
||||
def test_object_type_repr(self): |
||||
class X(object): |
||||
pass |
||||
self.assert_equal(object_type_repr(42), 'int object') |
||||
self.assert_equal(object_type_repr([]), 'list object') |
||||
self.assert_equal(object_type_repr(X()), |
||||
'jinja2.testsuite.utils.X object') |
||||
self.assert_equal(object_type_repr(None), 'None') |
||||
self.assert_equal(object_type_repr(Ellipsis), 'Ellipsis') |
||||
|
||||
|
||||
class MarkupLeakTestCase(JinjaTestCase): |
||||
|
||||
def test_markup_leaks(self): |
||||
counts = set() |
||||
for count in xrange(20): |
||||
for item in xrange(1000): |
||||
escape("foo") |
||||
escape("<foo>") |
||||
escape(u"foo") |
||||
escape(u"<foo>") |
||||
counts.add(len(gc.get_objects())) |
||||
assert len(counts) == 1, 'ouch, c extension seems to leak objects' |
||||
|
||||
|
||||
def suite(): |
||||
suite = unittest.TestSuite() |
||||
suite.addTest(unittest.makeSuite(LRUCacheTestCase)) |
||||
suite.addTest(unittest.makeSuite(HelpersTestCase)) |
||||
|
||||
# this test only tests the c extension |
||||
if not hasattr(escape, 'func_code'): |
||||
suite.addTest(unittest.makeSuite(MarkupLeakTestCase)) |
||||
|
||||
return suite |
Loading…
Reference in new issue