def test_basic_entry(self): schedule_lines = [ '2013/06/13 lightning energy', ' ;; schedule ; monthly', ' blah blah blah', ] schedule_thing = ScheduleThing(schedule_lines) schedule_thing.thing_date = date(2013, 7, 1) expected = [ '2013/07/01 lightning energy', ' blah blah blah', ] actual = schedule_thing._get_entry_thing().get_lines() self.assertEqual(expected, actual)
def test_two_entries_next_date(self): testdate = util.get_date_string(date.today()) schedule_lines = [ '{date} lightning energy'.format(date=testdate), ' ;; schedule ; yearly', ' blah blah blah', ] schedule_thing = ScheduleThing(schedule_lines) schedule_thing.thing_date = date.today() - relativedelta( years=1 ) schedule_thing.get_scheduled_entries() expected = date.today() + relativedelta(years=1) actual = schedule_thing.thing_date self.assertEqual(expected, actual)