def test_incorrect_date_object_passed(self):
"""Test handling incorrect value for 'date' argument."""
with self.assertRaises(ValueError) as exc_info:
get_next_lottery_date({})
self.assertEqual(exc_info.exception.args[0],
'Argument \'date\' must be a valid datetime object')
def test_next_date_at_current_date_exactly(self):
"""Test passed date as current lottery draw date."""
date = datetime(2018, 4, 11, 20)
result_date = get_next_lottery_date(date)
self.assertEqual(result_date, datetime(2018, 4, 14, 20))
self.assertEqual(result_date.strftime('%a %I%p'), 'Sat 08PM')
date = datetime(2018, 4, 14, 20)
result_date = get_next_lottery_date(date)
self.assertEqual(result_date, datetime(2018, 4, 18, 20))
self.assertEqual(result_date.strftime('%a %I%p'), 'Wed 08PM')
def test_current_date_as_default(self):
"""Test default value for a date evaluates to the current date."""
with freeze_time('2018-04-12'):
result_date = get_next_lottery_date()
self.assertEqual(result_date, datetime(2018, 4, 14, 20))
self.assertEqual(result_date.strftime('%a %I%p'), 'Sat 08PM')
def test_next_date_at_current_date_but_not_reached(self):
"""Test passed date as current lottery draw date."""
date = datetime(2018, 4, 11, 19, 59, 59)
result_date = get_next_lottery_date(date)
self.assertEqual(result_date, datetime(2018, 4, 11, 20))
self.assertEqual(result_date.strftime('%a %I%p'), 'Wed 08PM')
def test_next_date_at_next_month(self):
"""Test next date at the next month."""
date = datetime(2018, 4, 29)
result_date = get_next_lottery_date(date)
self.assertEqual(result_date, datetime(2018, 5, 2, 20))
self.assertEqual(result_date.strftime('%a %I%p'), 'Wed 08PM')
def test_next_date_at_next_week(self):
"""Test next date at the next week."""
date = datetime(2018, 4, 14, 20)
result_date = get_next_lottery_date(date)
self.assertEqual(result_date, datetime(2018, 4, 18, 20))
self.assertEqual(result_date.strftime('%a %I%p'), 'Wed 08PM')
def test_next_date_within_current_week(self):
"""Test next date within the current week."""
date = datetime(2018, 4, 12, 21, 10, 10)
result_date = get_next_lottery_date(date)
self.assertEqual(result_date, datetime(2018, 4, 14, 20))
self.assertEqual(result_date.strftime('%a %I%p'), 'Sat 08PM')