tests/test_features.py

import sys

import pytest

from jinja import contextfilter
from jinja import Environment
from jinja import Template
from jinja._compat import text_type


@pytest.mark.skipif(sys.version_info < (3, 5), reason="Requires 3.5 or later")
def test_generator_stop():
    class X(object):
        def __getattr__(self, name):
            raise StopIteration()

    t = Template("a{{ bad.bar() }}b")
    with pytest.raises(RuntimeError):
        t.render(bad=X())


@pytest.mark.skipif(sys.version_info[0] > 2, reason="Feature only supported on 2.x")
def test_ascii_str():
    @contextfilter
    def assert_func(context, value):
        assert type(value) is context["expected_type"]

    env = Environment()
    env.filters["assert"] = assert_func

    env.policies["compiler.ascii_str"] = False
    t = env.from_string('{{ "foo"|assert }}')
    t.render(expected_type=text_type)

    env.policies["compiler.ascii_str"] = True
    t = env.from_string('{{ "foo"|assert }}')
    t.render(expected_type=str)

    for val in True, False:
        env.policies["compiler.ascii_str"] = val
        t = env.from_string(u'{{ "\N{SNOWMAN}"|assert }}')
        t.render(expected_type=text_type)