def test_macro_call_3():
template = '''
{% macro format_hello(name, n, m='test', o='test') -%}
Hello, {{ name }}!
{{ n }}
{%- endmacro %}
{{ format_hello({}, 2, 'value', 'value') }}
'''
with pytest.raises(MergeException) as e:
infer(template)
assert str(e.value) == (
'unnamed variable (used as dictionary on lines 7) conflicts with '
'variable "argument #1" (used as scalar on lines: 2)')
template = '''
{% macro format_hello(name, n, m='test', o='test') -%}
Hello, {{ name }}!
{{ n }}
{%- endmacro %}
{{ format_hello(a, 2, 'value', {}) }}
'''
with pytest.raises(MergeException) as e:
infer(template)
assert str(e.value) == (
'unnamed variable (used as dictionary on lines 7) conflicts with '
'variable "argument "o"" (used as string on lines: 2)')
def test_basics_15():
# 1. Accept no unknown filter (default behavior)
config = Config()
template = '''
{% for item in items %}
{{ item.attr1|lower }}
{{ item.attr2|bar }}
{% endfor %}
'''
with pytest.raises(InvalidExpression) as e:
infer(template, config)
assert 'line 4: unknown filter "bar"' == str(e.value)
# 2. Accept all unknown filters
config = Config()
config.IGNORE_UNKNOWN_FILTERS = True
template = '''
{% for item in items %}
{{ item.attr1|lower }}
{{ item.attr2|bar }}
{% endfor %}
'''
struct = infer(template, config)
expected_struct = Dictionary({
'items': List(Dictionary({
'attr1': String(label='attr1', linenos=[3]),
'attr2': Unknown(label='attr2', linenos=[4])
}, label='item', linenos=[3, 4]), label='items', linenos=[2]),
})
assert struct == expected_struct
def test_basics_3():
template = '''
{% if x %}
{% set x = '123' %}
{% else %}
{% set x = '456' %}
{% endif %}
{{ x }}
'''
struct = infer(template)
expected_struct = Dictionary({
'x': String(label='x', linenos=[2, 3, 5, 7]),
})
assert struct == expected_struct
template = '''
{% if z %}
{% set x = '123' %}
{% else %}
{% set x = '456' %}
{% endif %}
{{ x }}
'''
struct = infer(template)
expected_struct = Dictionary({
'z': Unknown(label='z', linenos=[2]),
})
assert struct == expected_struct
def test_macro_call_3():
template = '''
{% macro format_hello(name, n, m='test', o='test') -%}
Hello, {{ name }}!
{{ n }}
{%- endmacro %}
{{ format_hello({}, 2, 'value', 'value') }}
'''
with pytest.raises(MergeException) as e:
infer(template)
assert str(e.value) == ('unnamed variable (used as dictionary on lines 7) conflicts with '
'variable "argument #1" (used as scalar on lines: 2)')
template = '''
{% macro format_hello(name, n, m='test', o='test') -%}
Hello, {{ name }}!
{{ n }}
{%- endmacro %}
{{ format_hello(a, 2, 'value', {}) }}
'''
with pytest.raises(MergeException) as e:
infer(template)
assert str(e.value) == ('unnamed variable (used as dictionary on lines 7) conflicts with '
'variable "argument "o"" (used as string on lines: 2)')
def test_basics_3():
template = '''
{% if x %}
{% set x = '123' %}
{% else %}
{% set x = '456' %}
{% endif %}
{{ x }}
'''
struct = infer(template)
expected_struct = Dictionary({
'x': String(label='x', linenos=[2, 3, 5, 7]),
})
assert struct == expected_struct
template = '''
{% if z %}
{% set x = '123' %}
{% else %}
{% set x = '456' %}
{% endif %}
{{ x }}
'''
struct = infer(template)
expected_struct = Dictionary({
'z': Unknown(label='z', linenos=[2]),
})
assert struct == expected_struct
def test_items_noarg():
template = '''
{% macro selectinputdict(name, values, value=0, addemptyrow=false ,extrataginfo='') -%}
<select name="{{ name }}" id="{{ name }}" {{ extrataginfo }}>
{% for k,v in values.items(456) %}
<option value="{{ k or '' }}" {{ 'selected' if value==k+1 }}>{{ v }}</option>
{% endfor %}
</select>
{%- endmacro %}
<td>{{ selectinputdict('priv_new_member', g.users, 5,true) }}</td>
'''
with pytest.raises(InvalidExpression):
infer(template)
def test_basics_2():
template = '''
{% if test1 %}
{% if test2 %}
{% set d = {'x': 123, a: z.qwerty} %}
{% else %}
{% set d = {'x': 456, a: z.gsom} %}
{% endif %}
{% endif %}
{% if d %}
{{ d.x }}
{% endif %}
{{ z.gsom }}
'''
struct = infer(template)
expected_struct = Dictionary({
'a': Scalar(label='a', linenos=[4, 6]),
'test1': Unknown(label='test1', linenos=[2]),
'test2': Unknown(label='test2', linenos=[3]),
'z': Dictionary(data={
'qwerty': Unknown(label='qwerty', linenos=[4]),
'gsom': Scalar(label='gsom', linenos=[6, 12]),
}, label='z', linenos=[4, 6, 12]),
})
assert struct == expected_struct
def test_1():
template = '''
{%- if x is undefined %}
{{ test }}
{%- endif %}
{%- if y is undefined %}
{% set y = 123 %}
{%- endif %}
{%- if y is defined %}
{{ y }}
{%- endif %}
{%- if z is undefined %}
{{ z }}
{%- endif %}
'''
struct = infer(template)
expected_struct = Dictionary({
'x':
Unknown(label='x', checked_as_undefined=True, linenos=[2]),
'test':
Scalar(label='test', linenos=[3]),
'y':
Number(label='y', may_be_defined=True, linenos=[6, 7, 10, 11]),
'z':
Scalar(label='z', linenos=[14, 15]),
})
assert struct == expected_struct
def test_basics_2():
template = '''
{% if test1 %}
{% if test2 %}
{% set d = {'x': 123, a: z.qwerty} %}
{% else %}
{% set d = {'x': 456, a: z.gsom} %}
{% endif %}
{% endif %}
{% if d %}
{{ d.x }}
{% endif %}
{{ z.gsom }}
'''
struct = infer(template)
expected_struct = Dictionary({
'a':
Scalar(label='a', linenos=[4, 6]),
'test1':
Unknown(label='test1', linenos=[2]),
'test2':
Unknown(label='test2', linenos=[3]),
'z':
Dictionary(data={
'qwerty': Unknown(label='qwerty', linenos=[4]),
'gsom': Scalar(label='gsom', linenos=[6, 12]),
},
label='z',
linenos=[4, 6, 12]),
})
assert struct == expected_struct
def test_items():
template = '''
{% macro selectinputdict(name, values, value=0, addemptyrow=false ,extrataginfo='') -%}
<select name="{{ name }}" id="{{ name }}" {{ extrataginfo }}>
{% if addemptyrow %}
<option></option>
{% endif %}
{% for k,v in values.items() %}
<option value="{{ k or '' }}" {{ 'selected' if value==k+1 }}>{{ v }}</option>
{% endfor %}
</select>
{%- endmacro %}
<td>{{ selectinputdict('priv_new_member', g.users, 5,true) }}</td>
'''
struct = infer(template)
expected_struct = Dictionary({
'g':
Dictionary(label="g",
data={
'users':
Dictionary(label="users", data={}, linenos=[2, 12]),
},
linenos=[12]),
})
assert struct == expected_struct
def test_macro_call_2():
template = '''
{% macro user(login, name, is_active=True) %}
{{ login }} {{ name.first }} {{ name.last }} {{ is_active }}
{% endmacro %}
{{ user(data.login, data.name, is_active=data.is_active) }}
'''
struct = infer(template)
assert struct['data'] == Dictionary(
{
'login':
Scalar(label='login', linenos=[2, 5]),
'is_active':
Boolean(label='is_active', linenos=[2, 5]),
'name':
Dictionary(
{
'first': Scalar(label='first', linenos=[3]),
'last': Scalar(label='last', linenos=[3]),
},
label='name',
linenos=[2, 5]),
},
label='data',
linenos=[5])
def test_extend_with_block_override_1(env, config):
struct = infer(
env.loader.get_source(env, 'inner_extend_override_1.html')[0], config)
expected_struct = Dictionary({
'name': Scalar(label='name', linenos=[3]),
})
assert struct == expected_struct
def test_list_becomes_dict():
template = '''
{% for r in PrjList %}
<tr><td><a class='btn btn-primary' href='/part/prj/{{ r[0] }}'>Select</a></td>
<td> {{ r[1] }} [{{ r[0] }}] {{ ("<a href='mailto:"+r['email']+"'>") |safe if r['email'] }}</td>
<td>{{ r['samplecount'] }}</td>
</tr>
{% endfor %}
'''
struct = infer(template)
expected_struct = Dictionary({
'PrjList':
List(Dictionary(
{
0: Scalar(),
1: Scalar(),
'email': String(label="email", linenos=[4]),
'samplecount': Scalar(label="samplecount", linenos=[5]),
},
label="r",
linenos=[3, 4, 5]),
label="PrjList",
linenos=[2]),
})
assert struct == expected_struct
def test_basics_4():
template = '''
{% set xys = [
('a', 0.3),
('b', 0.3),
] %}
{% if configuration is undefined %}
{% set configuration = 'prefix-' ~ timestamp %}
{% endif %}
queue: {{ queue if queue is defined else 'wizard' }}
description: >-
{% for x, y in xys %}
{{ loop.index }}:
{{ x }} {{ y }}
{% endfor %}
'''
struct = infer(template)
expected_struct = Dictionary({
'configuration':
String(label='configuration',
may_be_defined=True,
checked_as_undefined=True,
constant=False,
linenos=[6, 7]),
'queue':
Scalar(label='queue',
checked_as_defined=True,
constant=False,
linenos=[9]),
'timestamp':
String(label='timestamp', constant=False, linenos=[7])
})
assert struct == expected_struct
def test_basics_13():
config = Config()
config.TYPE_OF_VARIABLE_INDEXED_WITH_INTEGER_TYPE = 'tuple'
template = '''
{% for x in xs %}
{{ x[2] }}
{{ x[3] }}
{% endfor %}
'''
struct = infer(template, config)
expected_struct = Dictionary({
'xs':
List(Tuple((
Unknown(label=None, linenos=[]),
Unknown(label=None, linenos=[]),
Scalar(label=None, linenos=[3]),
Scalar(label=None, linenos=[4]),
),
label='x',
linenos=[3, 4]),
label='xs',
linenos=[2])
})
assert struct == expected_struct
def test_1():
template = '''
{%- if x is undefined %}
{{ test }}
{%- endif %}
{%- if y is undefined %}
{% set y = 123 %}
{%- endif %}
{%- if y is defined %}
{{ y }}
{%- endif %}
{%- if z is undefined %}
{{ z }}
{%- endif %}
'''
struct = infer(template)
expected_struct = Dictionary({
'x': Unknown(label='x', checked_as_undefined=True, linenos=[2]),
'test': Scalar(label='test', linenos=[3]),
'y': Number(label='y', may_be_defined=True, linenos=[6, 7, 10, 11]),
'z': Scalar(label='z', linenos=[14, 15]),
})
assert struct == expected_struct
def test_2():
template = '''
{% if x is undefined %}
{% set x = 'atata' %}
{% endif %}
{{ x }}
{% if y is defined %}
{# pass #}
{% else %}
{% set y = 'atata' %}
{% endif %}
{{ y }}
'''
struct = infer(template)
expected_struct = Dictionary({
'x':
String(linenos=[2, 3, 5],
label='x',
constant=False,
may_be_defined=True),
'y':
String(linenos=[7, 10, 12],
label='y',
constant=False,
may_be_defined=True),
})
assert struct == expected_struct
template = '''
{{ x }}
{% if x is undefined %}
{% set x = 'atata' %}
{% endif %}
{{ x }}
'''
struct = infer(template)
expected_struct = Dictionary({
'x':
String(linenos=[2, 3, 4, 6],
label='x',
constant=False,
may_be_defined=False),
})
assert struct == expected_struct
def test_basics_1():
template = '''
{% set d = {'x': 123, a: z.qwerty} %}
{{ d.x }}
'''
struct = infer(template)
expected_struct = Dictionary({
'a': Scalar(label='a', linenos=[2]),
'z': Dictionary(label='z', data={
'qwerty': Unknown(label='qwerty', linenos=[2]),
}, linenos=[2]),
})
assert struct == expected_struct
template = '''
{% set d = {'x': 123, a: z.qwerty} %}
{{ d.x.field }}
'''
with pytest.raises(MergeException):
infer(template)
template = '''
{% set x = '123' %}
{{ x.test }}
'''
with pytest.raises(MergeException):
infer(template)
template = '''
{% set a = {'x': 123} %}
{% set b = {a: 'test'} %}
'''
with pytest.raises(MergeException):
infer(template)
def test_basics_1():
template = '''
{% set d = {'x': 123, a: z.qwerty} %}
{{ d.x }}
'''
struct = infer(template)
expected_struct = Dictionary({
'a': Scalar(label='a', linenos=[2]),
'z': Dictionary(label='z', data={
'qwerty': Unknown(label='qwerty', linenos=[2]),
}, linenos=[2]),
})
assert struct == expected_struct
template = '''
{% set d = {'x': 123, a: z.qwerty} %}
{{ d.x.field }}
'''
with pytest.raises(MergeException):
infer(template)
template = '''
{% set x = '123' %}
{{ x.test }}
'''
with pytest.raises(MergeException):
infer(template)
template = '''
{% set a = {'x': 123} %}
{% set b = {a: 'test'} %}
'''
with pytest.raises(MergeException):
infer(template)
def test_raw():
template = '''
{% raw %}
{{ x }}
{% endraw %}
'''
struct = infer(template)
expected_struct = Dictionary()
assert struct == expected_struct
def test_hints_2(config):
template = """
{% for message in get_flashed_messages(true) %}
<div class="alert alert-{{ message[0]|replace("message","info")|replace("error","danger") }} alert-dismissible" role="alert" >{{ message[1] }}</div>
{% endfor %}
"""
struct = infer(template, config)
expected_struct = Dictionary()
assert struct == expected_struct
def test_basics_8():
template = '''
{% for row in items|batch(3, ' ')|batch(1) %}
{{ row[1].name }}
{% endfor %}
'''
with pytest.raises(UnexpectedExpression) as excinfo:
infer(template)
e = excinfo.value
assert isinstance(e.actual_ast, nodes.Filter)
assert e.expected_struct == List(
Dictionary({
'name': Scalar(label='name', constant=False, linenos=[3])
}, constant=False, linenos=[3]),
label='row', constant=False, linenos=[2, 3]
)
assert e.actual_struct == List(List(Unknown()))
def test_extend_with_block_override_1():
env = Environment(loader=PackageLoader('tests', 'templates'))
struct = infer(
env.loader.get_source(env, 'inner_override_1.html')[0],
Config(PACKAGE_NAME='tests'))
expected_struct = Dictionary({
'name': Scalar(label='name', linenos=[3]),
})
assert struct == expected_struct
def test_raw():
template = '''
{% raw %}
{{ x }}
{% endraw %}
'''
struct = infer(template)
expected_struct = Dictionary()
assert struct == expected_struct
def test_basics_8():
template = '''
{% for row in items|batch(3, ' ')|batch(1) %}
{{ row[1].name }}
{% endfor %}
'''
with pytest.raises(UnexpectedExpression) as excinfo:
infer(template)
e = excinfo.value
assert isinstance(e.actual_ast, nodes.Filter)
assert e.expected_struct == List(
Dictionary({
'name': Scalar(label='name', constant=False, linenos=[3])
}, constant=False, linenos=[3]),
label='row', constant=False, linenos=[2, 3]
)
assert e.actual_struct == List(List(Unknown()))
def test_basics_6():
template = '''
{% for row in items|batch(3, ' ') %}
{% endfor %}
'''
struct = infer(template)
expected_struct = Dictionary({
'items': List(Unknown(), label='items', linenos=[2]),
})
assert struct == expected_struct
def test_basics_9():
template = '''
{% set xs = items|batch(3, ' ') %}
{{ xs[0][0] }}
'''
struct = infer(template)
expected_struct = Dictionary({
'items': List(Unknown(), label='items', linenos=[2]), # TODO it should be Scalar
})
assert struct == expected_struct
def test_basics_9():
template = '''
{% set xs = items|batch(3, ' ') %}
{{ xs[0][0] }}
'''
struct = infer(template)
expected_struct = Dictionary({
'items': List(Unknown(), label='items', linenos=[2]), # TODO it should be Scalar
})
assert struct == expected_struct
def test_include_override_1(env, config):
struct = infer(
env.loader.get_source(env, 'inner_include_override_1.html')[0], config)
expected_struct = Dictionary({
'default_name':
Scalar(label='default_name', linenos=[2]),
'name':
Scalar(label='name', linenos=[3]),
})
assert struct == expected_struct
def test_basics_6():
template = '''
{% for row in items|batch(3, ' ') %}
{% endfor %}
'''
struct = infer(template)
expected_struct = Dictionary({
'items': List(Unknown(), label='items', linenos=[2]),
})
assert struct == expected_struct
def test_order_number_setting_1_5():
config = Config(ORDER_NUMBER=True)
template = '''
{% if yy %}
{{ intooo }}
{{ zz }}
{% endif %}
'''
struct = infer(template, config)
assert struct['yy'].order_nr < struct['zz'].order_nr
def test_order_number_setting_1_5():
config = Config(ORDER_NUMBER=True)
template = '''
{% if yy %}
{{ intooo }}
{{ zz }}
{% endif %}
'''
struct = infer(template, config)
assert struct['yy'].order_nr < struct['zz'].order_nr
def test_basics_10():
template = '''
{% set items = data|dictsort %}
{% for x, y in items %}
{% endfor %}
'''
struct = infer(template)
expected_struct = Dictionary({
'data': Dictionary({}, label='data', linenos=[2]),
})
assert struct == expected_struct
def test_basics_10():
template = '''
{% set items = data|dictsort %}
{% for x, y in items %}
{% endfor %}
'''
struct = infer(template)
expected_struct = Dictionary({
'data': Dictionary({}, label='data', linenos=[2]),
})
assert struct == expected_struct
def test_hints_1(config):
template = """
<html>
<script src="{{ url_for(a, filename='vue.js') }}"></script>
</html>
"""
struct = infer(template, config)
expected_struct = Dictionary({
'a': String(label='a', linenos=[3]),
})
assert struct == expected_struct
def test_basics_12():
template = '''
{% for k, v in data|dictsort %}
{{ k }}
{{ v }}
{% endfor %}
'''
struct = infer(template)
expected_struct = Dictionary({
'data': Dictionary({}, label='data', linenos=[2]),
})
assert struct == expected_struct
def test_dict_get():
template = '''
{%if g.get(slswtch, 0) == 2 %}
<link rel=stylesheet type=text/css href="/css/select2.css')">
{%endif %}
{{ slswtch | safe }}
'''
struct = infer(template)
expected_struct = Dictionary(
{'slswtch': String(label="slswtch", linenos=[2, 5])})
assert struct == expected_struct
def test_basics_12():
template = '''
{% for k, v in data|dictsort %}
{{ k }}
{{ v }}
{% endfor %}
'''
struct = infer(template)
expected_struct = Dictionary({
'data': Dictionary({}, label='data', linenos=[2]),
})
assert struct == expected_struct
def test_macro_wrong_args():
macro_template = '''
{% macro format_hello(name, n, m='test', o='test') -%}
Hello, {{ name }}!
{{ n }}
{%- endmacro %}
'''
template = macro_template + '{{ format_hello() }}'
with pytest.raises(InvalidExpression) as e:
infer(template)
assert str(
e.value
) == 'line 6: incorrect usage of "format_hello". it takes exactly 2 positional arguments'
template = macro_template + '{{ format_hello(1, 2, "test", "test", 5) }}'
with pytest.raises(InvalidExpression) as e:
infer(template)
assert str(
e.value
) == 'line 6: incorrect usage of "format_hello". it takes exactly 2 positional arguments'
template = macro_template + '{{ format_hello(1, 2, missing=123) }}'
with pytest.raises(InvalidExpression) as e:
infer(template)
assert str(
e.value
) == 'line 6: incorrect usage of "format_hello". unknown keyword argument "missing" is passed'
def test_boolean_conditions_setting_1():
template = '''
{% if x %}
Hello!
{% endif %}
{{ 'Hello!' if y else '' }}
'''
config_1 = Config()
struct = infer(template, config_1)
expected_struct = Dictionary({
'x': Unknown(label='x', linenos=[2]),
'y': Unknown(label='y', linenos=[5]),
})
assert struct == expected_struct
infer('{% if [] %}{% endif %}', config_1) # make sure it doesn't raise
config_2 = Config(BOOLEAN_CONDITIONS=True)
struct = infer(template, config_2)
expected_struct = Dictionary({
'x': Boolean(label='x', linenos=[2]),
'y': Boolean(label='y', linenos=[5]),
})
assert struct == expected_struct
with pytest.raises(UnexpectedExpression) as e:
infer('{% if [] %}{% endif %}', config_2) # make sure this does raise
assert str(e.value) == ('conflict on the line 1\n'
'got: AST node jinja2.nodes.List of structure [<unknown>]\n'
'expected structure: <boolean>')
def test_boolean_conditions_setting_1():
template = '''
{% if x %}
Hello!
{% endif %}
{{ 'Hello!' if y else '' }}
'''
config_1 = Config()
struct = infer(template, config_1)
expected_struct = Dictionary({
'x': Unknown(label='x', linenos=[2]),
'y': Unknown(label='y', linenos=[5]),
})
assert struct == expected_struct
infer('{% if [] %}{% endif %}', config_1) # make sure it doesn't raise
config_2 = Config(BOOLEAN_CONDITIONS=True)
struct = infer(template, config_2)
expected_struct = Dictionary({
'x': Boolean(label='x', linenos=[2]),
'y': Boolean(label='y', linenos=[5]),
})
assert struct == expected_struct
with pytest.raises(UnexpectedExpression) as e:
infer('{% if [] %}{% endif %}', config_2) # make sure this does raise
assert str(e.value) == (
'conflict on the line 1\n'
'got: AST node jinja2.nodes.List of structure [<unknown>]\n'
'expected structure: <boolean>')
def test_extend_1(env, config):
struct = infer(env.loader.get_source(env, 'inner_extend.html')[0], config)
expected_struct = Dictionary({
'var':
Dictionary({'a': Scalar(label='a', linenos=[1])},
label='var',
linenos=[1]),
'some':
Scalar(label='some', linenos=[2]),
'extended':
Scalar(label='extended', linenos=[2]),
})
assert struct == expected_struct
def test_basics_7():
template = '''
{% for row in items|batch(3, ' ')|batch(1) %}
{{ row[1][1].name }}
{% endfor %}
'''
struct = infer(template)
expected_struct = Dictionary({
'items': List(Dictionary({
'name': Scalar(label='name', linenos=[3]),
}, linenos=[3]), label='items', linenos=[2, 3]),
})
assert struct == expected_struct
def test_boolean_conditions_setting_2():
config = Config(BOOLEAN_CONDITIONS=True)
template = '''
{% if x == 'test' %}
Hello!
{% endif %}
'''
struct = infer(template, config)
expected_struct = Dictionary({
'x': Unknown(label='x', linenos=[2]),
})
assert struct == expected_struct
def test_order_number_setting_1():
config = Config(ORDER_NUMBER=True)
template = '''
{{ x }}
{{ y }}
{{ z }}
{{ x }}
{{ x }}
'''
struct = infer(template, config)
assert struct['x'].order_nr < struct['y'].order_nr
assert struct['y'].order_nr < struct['z'].order_nr
def test_boolean_conditions_setting_2():
config = Config(BOOLEAN_CONDITIONS=True)
template = '''
{% if x == 'test' %}
Hello!
{% endif %}
'''
struct = infer(template, config)
expected_struct = Dictionary({
'x': Unknown(label='x', linenos=[2]),
})
assert struct == expected_struct
def test_assignment():
template = '''
{% set args = ['foo'] if foo else [] %}
{% set args = args + ['bar'] %}
{% set args = args + (['zork'] if zork else []) %}
f({{ args|join(sep) }});
'''
struct = infer(template)
expected_struct = Dictionary({
'foo': Unknown(label='foo', linenos=[2]),
'zork': Unknown(label='zork', linenos=[4]),
'sep': String(label='sep', linenos=[5])
})
assert struct == expected_struct
def test_2():
template = '''
{% if x is undefined %}
{% set x = 'atata' %}
{% endif %}
{{ x }}
{% if y is defined %}
{# pass #}
{% else %}
{% set y = 'atata' %}
{% endif %}
{{ y }}
'''
struct = infer(template)
expected_struct = Dictionary({
'x': String(linenos=[2, 3, 5], label='x',
constant=False, may_be_defined=True),
'y': String(linenos=[7, 10, 12], label='y',
constant=False, may_be_defined=True),
})
assert struct == expected_struct
template = '''
{{ x }}
{% if x is undefined %}
{% set x = 'atata' %}
{% endif %}
{{ x }}
'''
struct = infer(template)
expected_struct = Dictionary({
'x': String(linenos=[2, 3, 4, 6], label='x',
constant=False, may_be_defined=False),
})
assert struct == expected_struct
def test_basics_5():
template = '''
{% for row in items|batch(3, ' ') %}
{% for column in row %}
{{ column.x }}
{% endfor %}
{% endfor %}
'''
struct = infer(template)
expected_struct = Dictionary({
'items': List(Dictionary({
'x': Scalar(label='x', linenos=[4])
}, label='column', linenos=[4]), label='items', linenos=[2, 3]),
})
assert struct == expected_struct
def test_block_1():
config = Config()
template = '''
{% block test %}
{{ x }}
{{ y }}
{% endblock %}
'''
struct = infer(template, config)
expected_struct = Dictionary({
'x': Scalar(label='x', linenos=[3]),
'y': Scalar(label='y', linenos=[4]),
})
assert struct == expected_struct
def test_macro_call_2():
template = '''
{% macro user(login, name, is_active=True) %}
{{ login }} {{ name.first }} {{ name.last }} {{ is_active }}
{% endmacro %}
{{ user(data.login, data.name, is_active=data.is_active) }}
'''
struct = infer(template)
assert struct['data'] == Dictionary({
'login': Scalar(label='login', linenos=[2, 5]),
'is_active': Boolean(label='is_active', linenos=[2, 5]),
'name': Dictionary({
'first': Scalar(label='first', linenos=[3]),
'last': Scalar(label='last', linenos=[3]),
}, label='name', linenos=[2, 5]),
}, label='data', linenos=[5])
def test_order_number_setting_4():
config = Config(ORDER_NUMBER=True, ORDER_NUMBER_SUB_COUNTER=False)
template = '''
{% for a in aa %}
{{ ax }}
{% for b in bb %}
{{ bx }}
{% for c in cc %}
{{ cx }}
{% endfor %}
{% endfor %}
{% endfor %}
'''
struct = infer(template, config)
assert struct['ax'].order_nr != struct['bx'].order_nr != struct['cx'].order_nr
def test_for():
template = '''
{% for number in range(10 - users|length) %}
{{ number }}
{% endfor %}
'''
struct = infer(template)
expected_struct = Dictionary({
'users': List(Unknown(), label='users', linenos=[2]),
})
assert struct == expected_struct
template = '''
{% for number in range(10 - users|length) %}
{{ number.field }}
{% endfor %}
'''
with pytest.raises(MergeException):
infer(template)
template = '{{ range(10 - users|length) }}'
with pytest.raises(UnexpectedExpression):
infer(template)
template = '''
{% for number in lipsum(n=10) %}
{% endfor %}
'''
with pytest.raises(UnexpectedExpression):
infer(template)
template = '''
{% for k, v in data|dictsort %}
{{ k.x }}
{{ v }}
{% endfor %}
'''
with pytest.raises(UnexpectedExpression):
infer(template)
def test_order_number_setting_2():
config = Config(ORDER_NUMBER=True)
template = '''
{% for n in nx %}
{{ y }}
{{ z }}
{% endfor %}
{% if yy %}
{{ zz }}
{{ xx }}
{% endif %}
'''
struct = infer(template, config)
assert struct['y'].order_nr < struct['z'].order_nr
assert struct['nx'].order_nr < struct['yy'].order_nr
assert struct['zz'].order_nr < struct['xx'].order_nr
def test_3():
config = Config(BOOLEAN_CONDITIONS=True)
template = '''
{%- if new_configuration is undefined %}
{%- if production is defined and production %}
{% set new_configuration = 'prefix-' ~ timestamp %}
{%- else %}
{% set new_configuration = 'prefix-' ~ timestamp %}
{%- endif %}
{%- endif %}
'''
struct = infer(template, config)
expected_struct = Dictionary({
'new_configuration': String(label='new_configuration', may_be_defined=True, checked_as_undefined=True, linenos=[2, 4, 6]),
'production': Boolean(label='production', checked_as_defined=True, linenos=[3]),
'timestamp': String(label='timestamp', linenos=[4, 6]),
})
assert struct == expected_struct
def test_basics_14():
template = '''
{{ section.FILTERS.test }}
{%- for f in section.FILTERS.keys() %}
{{ section.GEO }}
{{ loop.index }}
{%- endfor %}
'''
struct = infer(template)
expected_struct = Dictionary({
'section': Dictionary({
'FILTERS': Dictionary({
'test': Scalar(label='test', linenos=[2])
}, label='FILTERS', linenos=[2, 3]),
'GEO': Scalar(label='GEO', linenos=[4]),
}, label='section', linenos=[2, 3, 4])
})
assert struct == expected_struct
def test_macro_call_1():
template = '''
{% macro format_hello(name, n, m='test', o='test', dict_arg={'field': 123}) -%}
Hello, {{ name }}!
{{ n }}
{{ dict_arg.field }}
{%- endmacro %}
{{ format_hello('Anton', 2, 'value', 'value') }}
{{ format_hello(name='Anton', n=2, m='value', o='value') }}
{{ format_hello('Anton', n=2, m='value', o='value') }}
{{ format_hello(name, 2, m='value', o='value') }}
{{ format_hello(name, 2, m='value', o='value', dict_arg=d) }}
'''
struct = infer(template)
assert struct == Dictionary({
'name': Scalar(label='name', linenos=[2, 11, 12]),
'd': Dictionary(label='d', linenos=[2, 12]),
})
def test_basics_13():
config = Config()
config.TYPE_OF_VARIABLE_INDEXED_WITH_INTEGER_TYPE = 'tuple'
template = '''
{% for x in xs %}
{{ x[2] }}
{{ x[3] }}
{% endfor %}
'''
struct = infer(template, config)
expected_struct = Dictionary({
'xs': List(Tuple((
Unknown(label=None, linenos=[]),
Unknown(label=None, linenos=[]),
Scalar(label=None, linenos=[3]),
Scalar(label=None, linenos=[4]),
), label='x', linenos=[3, 4]), label='xs', linenos=[2])
})
assert struct == expected_struct
def test_basics_4():
template = '''
{% set xys = [
('a', 0.3),
('b', 0.3),
] %}
{% if configuration is undefined %}
{% set configuration = 'prefix-' ~ timestamp %}
{% endif %}
queue: {{ queue if queue is defined else 'wizard' }}
description: >-
{% for x, y in xys %}
{{ loop.index }}:
{{ x }} {{ y }}
{% endfor %}
'''
struct = infer(template)
expected_struct = Dictionary({
'configuration': String(label='configuration',
may_be_defined=True, checked_as_undefined=True, constant=False, linenos=[6, 7]),
'queue': Scalar(label='queue', checked_as_defined=True, constant=False, linenos=[9]),
'timestamp': String(label='timestamp', constant=False, linenos=[7])
})
assert struct == expected_struct
def test_basics_11():
template = '''
{{ a|xmlattr }}
{{ a.attr1|join(',') }}
{{ a.attr2|default([])|first }}
{{ a.attr3|default('gsom') }}
{% for x in xs|rejectattr('is_active') %}
{{ x }}
{% endfor %}
'''
struct = infer(template)
expected_struct = Dictionary({
'a': Dictionary({
'attr1': List(String(), label='attr1', linenos=[3]),
'attr2': List(Scalar(linenos=[4]), label='attr2', linenos=[4], used_with_default=True),
'attr3': String(label='attr3', linenos=[5], used_with_default=True, value='gsom')
}, label='a', linenos=[2, 3, 4, 5]),
'xs': List(
Scalar(label='x', linenos=[7]), # TODO it should be Dictionary({'is_active': Unknown()})
label='xs',
linenos=[6]
),
})
assert struct == expected_struct