def test_initial_order(self):
"""Checklists are ordered by date then location."""
today = timezone.now().date()
yesterday = today - timedelta(days=1)
ChecklistFactory(date=today, location__name="C")
ChecklistFactory(date=yesterday, location__name="B")
ChecklistFactory(date=yesterday, location__name="A")
expected = ['C', 'A', 'B']
result = checklists_between(yesterday, today, order=self.order)
actual = [obj.location.name for obj in result]
self.assertEqual(expected, actual)
def test_only_included(self):
"""Only checklists marked as included are fetched"""
today = timezone.now().date()
yesterday = today - timedelta(days=1)
earlier = today - timedelta(days=2)
ChecklistFactory(date=today, include=True)
ChecklistFactory(date=yesterday, include=False)
ChecklistFactory(date=earlier, include=True)
expected = [today, earlier]
result = checklists_between(earlier, today, order=self.order)
actual = [checklist.date for checklist in result]
self.assertEqual(expected, actual)
def test_end_date(self):
"""Only checklists on or before the end date are fetched."""
today = timezone.now().date()
yesterday = today - timedelta(days=1)
earlier = today - timedelta(days=2)
ChecklistFactory(date=today)
ChecklistFactory(date=yesterday)
ChecklistFactory(date=earlier)
expected = [yesterday, earlier]
result = checklists_between(earlier, yesterday, order=self.order)
actual = [obj.date for obj in result]
self.assertEqual(expected, actual)
