mirror of
https://github.com/django/django.git
synced 2024-11-18 07:26:04 +00:00
219 lines
11 KiB
Python
219 lines
11 KiB
Python
from django.template import TemplateSyntaxError
|
|
from django.template.defaulttags import ForNode
|
|
from django.test import SimpleTestCase
|
|
|
|
from ..utils import setup
|
|
|
|
|
|
class ForTagTests(SimpleTestCase):
|
|
libraries = {'custom': 'template_tests.templatetags.custom'}
|
|
|
|
@setup({'for-tag01': '{% for val in values %}{{ val }}{% endfor %}'})
|
|
def test_for_tag01(self):
|
|
output = self.engine.render_to_string('for-tag01', {'values': [1, 2, 3]})
|
|
self.assertEqual(output, '123')
|
|
|
|
@setup({'for-tag02': '{% for val in values reversed %}{{ val }}{% endfor %}'})
|
|
def test_for_tag02(self):
|
|
output = self.engine.render_to_string('for-tag02', {'values': [1, 2, 3]})
|
|
self.assertEqual(output, '321')
|
|
|
|
@setup({'for-tag-vars01': '{% for val in values %}{{ forloop.counter }}{% endfor %}'})
|
|
def test_for_tag_vars01(self):
|
|
output = self.engine.render_to_string('for-tag-vars01', {'values': [6, 6, 6]})
|
|
self.assertEqual(output, '123')
|
|
|
|
@setup({'for-tag-vars02': '{% for val in values %}{{ forloop.counter0 }}{% endfor %}'})
|
|
def test_for_tag_vars02(self):
|
|
output = self.engine.render_to_string('for-tag-vars02', {'values': [6, 6, 6]})
|
|
self.assertEqual(output, '012')
|
|
|
|
@setup({'for-tag-vars03': '{% for val in values %}{{ forloop.revcounter }}{% endfor %}'})
|
|
def test_for_tag_vars03(self):
|
|
output = self.engine.render_to_string('for-tag-vars03', {'values': [6, 6, 6]})
|
|
self.assertEqual(output, '321')
|
|
|
|
@setup({'for-tag-vars04': '{% for val in values %}{{ forloop.revcounter0 }}{% endfor %}'})
|
|
def test_for_tag_vars04(self):
|
|
output = self.engine.render_to_string('for-tag-vars04', {'values': [6, 6, 6]})
|
|
self.assertEqual(output, '210')
|
|
|
|
@setup({'for-tag-vars05': '{% for val in values %}'
|
|
'{% if forloop.first %}f{% else %}x{% endif %}{% endfor %}'})
|
|
def test_for_tag_vars05(self):
|
|
output = self.engine.render_to_string('for-tag-vars05', {'values': [6, 6, 6]})
|
|
self.assertEqual(output, 'fxx')
|
|
|
|
@setup({'for-tag-vars06': '{% for val in values %}'
|
|
'{% if forloop.last %}l{% else %}x{% endif %}{% endfor %}'})
|
|
def test_for_tag_vars06(self):
|
|
output = self.engine.render_to_string('for-tag-vars06', {'values': [6, 6, 6]})
|
|
self.assertEqual(output, 'xxl')
|
|
|
|
@setup({'for-tag-unpack01': '{% for key,value in items %}{{ key }}:{{ value }}/{% endfor %}'})
|
|
def test_for_tag_unpack01(self):
|
|
output = self.engine.render_to_string('for-tag-unpack01', {'items': (('one', 1), ('two', 2))})
|
|
self.assertEqual(output, 'one:1/two:2/')
|
|
|
|
@setup({'for-tag-unpack03': '{% for key, value in items %}{{ key }}:{{ value }}/{% endfor %}'})
|
|
def test_for_tag_unpack03(self):
|
|
output = self.engine.render_to_string('for-tag-unpack03', {'items': (('one', 1), ('two', 2))})
|
|
self.assertEqual(output, 'one:1/two:2/')
|
|
|
|
@setup({'for-tag-unpack04': '{% for key , value in items %}{{ key }}:{{ value }}/{% endfor %}'})
|
|
def test_for_tag_unpack04(self):
|
|
output = self.engine.render_to_string('for-tag-unpack04', {'items': (('one', 1), ('two', 2))})
|
|
self.assertEqual(output, 'one:1/two:2/')
|
|
|
|
@setup({'for-tag-unpack05': '{% for key ,value in items %}{{ key }}:{{ value }}/{% endfor %}'})
|
|
def test_for_tag_unpack05(self):
|
|
output = self.engine.render_to_string('for-tag-unpack05', {'items': (('one', 1), ('two', 2))})
|
|
self.assertEqual(output, 'one:1/two:2/')
|
|
|
|
@setup({'for-tag-unpack06': '{% for key value in items %}{{ key }}:{{ value }}/{% endfor %}'})
|
|
def test_for_tag_unpack06(self):
|
|
msg = "'for' tag received an invalid argument: for key value in items"
|
|
with self.assertRaisesMessage(TemplateSyntaxError, msg):
|
|
self.engine.render_to_string('for-tag-unpack06', {'items': (('one', 1), ('two', 2))})
|
|
|
|
@setup({'for-tag-unpack07': '{% for key,,value in items %}{{ key }}:{{ value }}/{% endfor %}'})
|
|
def test_for_tag_unpack07(self):
|
|
msg = "'for' tag received an invalid argument: for key,,value in items"
|
|
with self.assertRaisesMessage(TemplateSyntaxError, msg):
|
|
self.engine.render_to_string('for-tag-unpack07', {'items': (('one', 1), ('two', 2))})
|
|
|
|
@setup({'for-tag-unpack08': '{% for key,value, in items %}{{ key }}:{{ value }}/{% endfor %}'})
|
|
def test_for_tag_unpack08(self):
|
|
msg = "'for' tag received an invalid argument: for key,value, in items"
|
|
with self.assertRaisesMessage(TemplateSyntaxError, msg):
|
|
self.engine.render_to_string('for-tag-unpack08', {'items': (('one', 1), ('two', 2))})
|
|
|
|
@setup({'double-quote': '{% for "k" in items %}{{ "k" }}/{% endfor %}'})
|
|
def test_unpack_double_quote(self):
|
|
msg = """'for' tag received an invalid argument: for "k" in items"""
|
|
with self.assertRaisesMessage(TemplateSyntaxError, msg):
|
|
self.engine.render_to_string('double-quote', {'items': (1, 2)})
|
|
|
|
@setup({'single-quote': "{% for 'k' in items %}{{ k }}/{% endfor %}"})
|
|
def test_unpack_single_quote(self):
|
|
msg = """'for' tag received an invalid argument: for 'k' in items"""
|
|
with self.assertRaisesMessage(TemplateSyntaxError, msg):
|
|
self.engine.render_to_string('single-quote', {'items': (1, 2)})
|
|
|
|
@setup({'vertical-bar': '{% for k|upper in items %}{{ k|upper }}/{% endfor %}'})
|
|
def test_unpack_vertical_bar(self):
|
|
msg = "'for' tag received an invalid argument: for k|upper in items"
|
|
with self.assertRaisesMessage(TemplateSyntaxError, msg):
|
|
self.engine.render_to_string('vertical-bar', {'items': (1, 2)})
|
|
|
|
@setup({'for-tag-unpack09': '{% for val in items %}{{ val.0 }}:{{ val.1 }}/{% endfor %}'})
|
|
def test_for_tag_unpack09(self):
|
|
"""
|
|
A single loopvar doesn't truncate the list in val.
|
|
"""
|
|
output = self.engine.render_to_string('for-tag-unpack09', {'items': (('one', 1), ('two', 2))})
|
|
self.assertEqual(output, 'one:1/two:2/')
|
|
|
|
@setup({'for-tag-unpack13': '{% for x,y,z in items %}{{ x }}:{{ y }},{{ z }}/{% endfor %}'})
|
|
def test_for_tag_unpack13(self):
|
|
output = self.engine.render_to_string(
|
|
'for-tag-unpack13', {'items': (('one', 1, 'carrot'), ('two', 2, 'cheese'))}
|
|
)
|
|
if self.engine.string_if_invalid:
|
|
self.assertEqual(output, 'one:1,carrot/two:2,cheese/')
|
|
else:
|
|
self.assertEqual(output, 'one:1,carrot/two:2,cheese/')
|
|
|
|
@setup({'for-tag-empty01': '{% for val in values %}{{ val }}{% empty %}empty text{% endfor %}'})
|
|
def test_for_tag_empty01(self):
|
|
output = self.engine.render_to_string('for-tag-empty01', {'values': [1, 2, 3]})
|
|
self.assertEqual(output, '123')
|
|
|
|
@setup({'for-tag-empty02': '{% for val in values %}{{ val }}{% empty %}values array empty{% endfor %}'})
|
|
def test_for_tag_empty02(self):
|
|
output = self.engine.render_to_string('for-tag-empty02', {'values': []})
|
|
self.assertEqual(output, 'values array empty')
|
|
|
|
@setup({'for-tag-empty03': '{% for val in values %}'
|
|
'{{ val }}{% empty %}values array not found{% endfor %}'})
|
|
def test_for_tag_empty03(self):
|
|
output = self.engine.render_to_string('for-tag-empty03')
|
|
self.assertEqual(output, 'values array not found')
|
|
|
|
@setup({'for-tag-filter-ws': "{% load custom %}{% for x in s|noop:'x y' %}{{ x }}{% endfor %}"})
|
|
def test_for_tag_filter_ws(self):
|
|
"""
|
|
#19882
|
|
"""
|
|
output = self.engine.render_to_string('for-tag-filter-ws', {'s': 'abc'})
|
|
self.assertEqual(output, 'abc')
|
|
|
|
@setup({'for-tag-unpack-strs': '{% for x,y in items %}{{ x }}:{{ y }}/{% endfor %}'})
|
|
def test_for_tag_unpack_strs(self):
|
|
output = self.engine.render_to_string('for-tag-unpack-strs', {'items': ('ab', 'ac')})
|
|
self.assertEqual(output, 'a:b/a:c/')
|
|
|
|
@setup({'for-tag-unpack10': '{% for x,y in items %}{{ x }}:{{ y }}/{% endfor %}'})
|
|
def test_for_tag_unpack10(self):
|
|
with self.assertRaisesMessage(ValueError, 'Need 2 values to unpack in for loop; got 3.'):
|
|
self.engine.render_to_string(
|
|
'for-tag-unpack10',
|
|
{'items': (('one', 1, 'carrot'), ('two', 2, 'orange'))},
|
|
)
|
|
|
|
@setup({'for-tag-unpack11': '{% for x,y,z in items %}{{ x }}:{{ y }},{{ z }}/{% endfor %}'})
|
|
def test_for_tag_unpack11(self):
|
|
with self.assertRaisesMessage(ValueError, 'Need 3 values to unpack in for loop; got 2.'):
|
|
self.engine.render_to_string(
|
|
'for-tag-unpack11',
|
|
{'items': (('one', 1), ('two', 2))},
|
|
)
|
|
|
|
@setup({'for-tag-unpack12': '{% for x,y,z in items %}{{ x }}:{{ y }},{{ z }}/{% endfor %}'})
|
|
def test_for_tag_unpack12(self):
|
|
with self.assertRaisesMessage(ValueError, 'Need 3 values to unpack in for loop; got 2.'):
|
|
self.engine.render_to_string(
|
|
'for-tag-unpack12',
|
|
{'items': (('one', 1, 'carrot'), ('two', 2))}
|
|
)
|
|
|
|
@setup({'for-tag-unpack14': '{% for x,y in items %}{{ x }}:{{ y }}/{% endfor %}'})
|
|
def test_for_tag_unpack14(self):
|
|
with self.assertRaisesMessage(ValueError, 'Need 2 values to unpack in for loop; got 1.'):
|
|
self.engine.render_to_string('for-tag-unpack14', {'items': (1, 2)})
|
|
|
|
@setup({
|
|
'main': '{% with alpha=alpha.values %}{% include "base" %}{% endwith %}_'
|
|
'{% with alpha=alpha.extra %}{% include "base" %}{% endwith %}',
|
|
'base': '{% for x, y in alpha %}{{ x }}:{{ y }},{% endfor %}'
|
|
})
|
|
def test_for_tag_context(self):
|
|
"""
|
|
ForNode.render() pops the values it pushes to the context (#28001).
|
|
"""
|
|
output = self.engine.render_to_string('main', {
|
|
'alpha': {
|
|
'values': [('two', 2), ('four', 4)],
|
|
'extra': [('six', 6), ('eight', 8)],
|
|
},
|
|
})
|
|
self.assertEqual(output, 'two:2,four:4,_six:6,eight:8,')
|
|
|
|
@setup({'invalid_for_loop': '{% for x items %}{{ x }}{% endfor %}'})
|
|
def test_invalid_arg(self):
|
|
msg = "'for' statements should have at least four words: for x items"
|
|
with self.assertRaisesMessage(TemplateSyntaxError, msg):
|
|
self.engine.render_to_string('invalid_for_loop', {'items': (1, 2)})
|
|
|
|
@setup({'invalid_for_loop': '{% for x from items %}{{ x }}{% endfor %}'})
|
|
def test_invalid_in_keyword(self):
|
|
msg = "'for' statements should use the format 'for x in y': for x from items"
|
|
with self.assertRaisesMessage(TemplateSyntaxError, msg):
|
|
self.engine.render_to_string('invalid_for_loop', {'items': (1, 2)})
|
|
|
|
|
|
class ForNodeTests(SimpleTestCase):
|
|
def test_repr(self):
|
|
node = ForNode('x', 'sequence', is_reversed=True, nodelist_loop=['val'], nodelist_empty=['val2'])
|
|
self.assertEqual(repr(node), '<ForNode: for x in sequence, tail_len: 1 reversed>')
|