0
0
mirror of https://github.com/django/django.git synced 2024-12-01 15:42:04 +01:00

Added tests for date_hierarchy template tag with models.DateTimeField.

This commit is contained in:
Alexander Holmbäck 2020-03-25 11:52:20 +01:00 committed by Mariusz Felisiak
parent 8bebb8f126
commit e43c34d10d

View File

@ -128,3 +128,45 @@ class DateHierarchyTests(TestCase):
]
expected_choices = [('?' + choice) if choice else '' for choice in expected_choices]
self.assertEqual(choices, expected_choices)
def test_choice_links_datetime(self):
modeladmin = ModelAdmin(Question, site)
modeladmin.date_hierarchy = 'expires'
Question.objects.bulk_create([
Question(question='q1', expires=datetime.datetime(2017, 10, 1)),
Question(question='q2', expires=datetime.datetime(2017, 10, 1)),
Question(question='q3', expires=datetime.datetime(2017, 12, 15)),
Question(question='q4', expires=datetime.datetime(2017, 12, 15)),
Question(question='q5', expires=datetime.datetime(2017, 12, 31)),
Question(question='q6', expires=datetime.datetime(2018, 2, 1)),
])
tests = [
({}, [['year=2017'], ['year=2018']]),
({'year': 2016}, []),
(
{'year': 2017}, [
['month=10', 'year=2017'],
['month=12', 'year=2017'],
],
),
({'year': 2017, 'month': 9}, []),
(
{'year': 2017, 'month': 12}, [
['day=15', 'month=12', 'year=2017'],
['day=31', 'month=12', 'year=2017'],
],
),
]
for query, expected_choices in tests:
with self.subTest(query=query):
query = {'expires__%s' % q: val for q, val in query.items()}
request = self.factory.get('/', query)
request.user = self.superuser
changelist = modeladmin.get_changelist_instance(request)
spec = date_hierarchy(changelist)
choices = [choice['link'] for choice in spec['choices']]
expected_choices = [
'?' + '&'.join('expires__%s' % c for c in choice)
for choice in expected_choices
]
self.assertEqual(choices, expected_choices)