class GetSafeDate(Redirector):
def setUp(self):
super(GetSafeDate, self).setUp()
schedule_lines_test = [
'2013/06/29 lightning energy',
' ;; schedule ; monthly ; 12th ; ; auto'
]
ScheduleThing.do_file_config = False
self.schedule_thing = ScheduleThing(schedule_lines_test)
def test_date_is_fine(self):
expected = date(2013, 8, 31)
actual = self.schedule_thing._get_safe_date(expected, 31)
self.assertEqual(expected, actual)
def test_day_is_too_many(self):
expected = date(2013, 8, 31)
actual = self.schedule_thing._get_safe_date(
date(2013, 8, 31),
99
)
self.assertEqual(expected, actual)
