tests/test_old_bugs.py

# -*- coding: utf-8 -*-
"""
    Tests for old bugs
    ~~~~~~~~~~~~~~~~~~

    Unittest that test situations caused by various older bugs.

    :copyright: (c) 2009 by the Jinja Team.
    :license: BSD.
"""
from jinja2 import Template, Environment, DictLoader, TemplateSyntaxError, \
     TemplateNotFound, PrefixLoader

env = Environment()

from nose import SkipTest
from nose.tools import assert_raises


def test_keyword_folding():
    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():
    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():
    tmpl = env.from_string('{{ "http://www.example.org/<foo"|urlize }}')
    assert tmpl.render() == '<a href="http://www.example.org/&lt;foo">http://www.example.org/&lt;foo</a>'


def test_loop_call_loop():
    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():
    env = Environment(line_statement_prefix='%')
    assert_raises(TemplateSyntaxError, env.from_string,
                  '% for item in seq {# missing #}\n...% endfor')


def test_old_macro_loop_scoping_bug():
    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():
    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():
    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_call_with_args():
    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():
    assert_raises(TemplateSyntaxError, Template, '{% if %}....{% endif %}')
    assert_raises(TemplateSyntaxError, Template, '{% if foo %}...{% elif %}...{% endif %}')
    assert_raises(TemplateSyntaxError, Template, '{% for x in %}..{% endfor %}')


def test_recursive_loop_bug():
    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():
    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'