104 lines
3.8 KiB
Python
104 lines
3.8 KiB
Python
|
from datetime import date
|
||
|
|
|
||
|
|
from django.template.base import TemplateSyntaxError
|
||
|
|
from django.template.loader import get_template
|
||
|
|
from django.test import TestCase
|
||
|
|
|
||
|
|
from .utils import render, setup
|
||
|
|
|
||
|
|
|
||
|
|
class RegroupTagTests(TestCase):
|
||
|
|
|
||
|
|
@setup({'regroup01': ''
|
||
|
|
'{% regroup data by bar as grouped %}'
|
||
|
|
'{% for group in grouped %}'
|
||
|
|
'{{ group.grouper }}:'
|
||
|
|
'{% for item in group.list %}'
|
||
|
|
'{{ item.foo }}'
|
||
|
|
'{% endfor %},'
|
||
|
|
'{% endfor %}'})
|
||
|
|
def test_regroup01(self):
|
||
|
|
output = render('regroup01', {
|
||
|
|
'data': [{'foo': 'c', 'bar': 1},
|
||
|
|
{'foo': 'd', 'bar': 1},
|
||
|
|
{'foo': 'a', 'bar': 2},
|
||
|
|
{'foo': 'b', 'bar': 2},
|
||
|
|
{'foo': 'x', 'bar': 3}],
|
||
|
|
})
|
||
|
|
self.assertEqual(output, '1:cd,2:ab,3:x,')
|
||
|
|
|
||
|
|
@setup({'regroup02': ''
|
||
|
|
'{% regroup data by bar as grouped %}'
|
||
|
|
'{% for group in grouped %}'
|
||
|
|
'{{ group.grouper }}:'
|
||
|
|
'{% for item in group.list %}'
|
||
|
|
'{{ item.foo }}'
|
||
|
|
'{% endfor %}'
|
||
|
|
'{% endfor %}'})
|
||
|
|
def test_regroup02(self):
|
||
|
|
"""
|
||
|
|
Test for silent failure when target variable isn't found
|
||
|
|
"""
|
||
|
|
output = render('regroup02', {})
|
||
|
|
self.assertEqual(output, '')
|
||
|
|
|
||
|
|
@setup({'regroup03': ''
|
||
|
|
'{% regroup data by at|date:"m" as grouped %}'
|
||
|
|
'{% for group in grouped %}'
|
||
|
|
'{{ group.grouper }}:'
|
||
|
|
'{% for item in group.list %}'
|
||
|
|
'{{ item.at|date:"d" }}'
|
||
|
|
'{% endfor %},'
|
||
|
|
'{% endfor %}'})
|
||
|
|
def test_regroup03(self):
|
||
|
|
"""
|
||
|
|
Regression tests for #17675
|
||
|
|
The date template filter has expects_localtime = True
|
||
|
|
"""
|
||
|
|
output = render('regroup03', {
|
||
|
|
'data': [{'at': date(2012, 2, 14)},
|
||
|
|
{'at': date(2012, 2, 28)},
|
||
|
|
{'at': date(2012, 7, 4)}],
|
||
|
|
})
|
||
|
|
self.assertEqual(output, '02:1428,07:04,')
|
||
|
|
|
||
|
|
@setup({'regroup04': ''
|
||
|
|
'{% regroup data by bar|join:"" as grouped %}'
|
||
|
|
'{% for group in grouped %}'
|
||
|
|
'{{ group.grouper }}:'
|
||
|
|
'{% for item in group.list %}'
|
||
|
|
'{{ item.foo|first }}'
|
||
|
|
'{% endfor %},'
|
||
|
|
'{% endfor %}'})
|
||
|
|
def test_regroup04(self):
|
||
|
|
"""
|
||
|
|
The join template filter has needs_autoescape = True
|
||
|
|
"""
|
||
|
|
output = render('regroup04', {
|
||
|
|
'data': [{'foo': 'x', 'bar': ['ab', 'c']},
|
||
|
|
{'foo': 'y', 'bar': ['a', 'bc']},
|
||
|
|
{'foo': 'z', 'bar': ['a', 'd']}],
|
||
|
|
})
|
||
|
|
self.assertEqual(output, 'abc:xy,ad:z,')
|
||
|
|
|
||
|
|
# Test syntax errors
|
||
|
|
@setup({'regroup05': '{% regroup data by bar as %}'})
|
||
|
|
def test_regroup05(self):
|
||
|
|
with self.assertRaises(TemplateSyntaxError):
|
||
|
|
get_template('regroup05')
|
||
|
|
|
||
|
|
@setup({'regroup06': '{% regroup data by bar thisaintright grouped %}'})
|
||
|
|
def test_regroup06(self):
|
||
|
|
with self.assertRaises(TemplateSyntaxError):
|
||
|
|
get_template('regroup06')
|
||
|
|
|
||
|
|
@setup({'regroup07': '{% regroup data thisaintright bar as grouped %}'})
|
||
|
|
def test_regroup07(self):
|
||
|
|
with self.assertRaises(TemplateSyntaxError):
|
||
|
|
get_template('regroup07')
|
||
|
|
|
||
|
|
@setup({'regroup08': '{% regroup data by bar as grouped toomanyargs %}'})
|
||
|
|
def test_regroup08(self):
|
||
|
|
with self.assertRaises(TemplateSyntaxError):
|
||
|
|
get_template('regroup08')
|