def test_scheduled_for_reminder_weekends(self):
h = Habit(user=self.user,
start=datetime.date(2013, 3, 4),
description='Do a thing. On a day.')
h.set_reminder_schedule(SCHEDULE_WEEKENDS, 0)
h.save()
self.assertEquals(len(self._scheduled(calendar.MONDAY, 0)), 0)
self.assertEquals(len(self._scheduled(calendar.SATURDAY, 6)), 0)
self.assertEquals(len(self._scheduled(calendar.SATURDAY, 0)), 1)
self.assertEquals(self._scheduled(calendar.SATURDAY, 0)[0], h)
def test_scheduled_for_reminder_mondays(self):
h = Habit(user=self.user,
start=datetime.date(2013, 3, 4),
description='Do a thing. On a day.')
h.set_reminder_schedule(SCHEDULE_MONDAYS, 12)
h.save()
self.assertEquals(len(self._scheduled(calendar.MONDAY, 12)), 1)
self.assertEquals(self._scheduled(calendar.MONDAY, 12)[0], h)
self.assertEquals(len(self._scheduled(calendar.MONDAY, 6)), 0)
self.assertEquals(len(self._scheduled(calendar.TUESDAY, 12)), 0)
def test_reminder_last_sent(self):
h = Habit(user=self.user,
start=datetime.date(2013, 3, 4),
description='Test my reminders')
d = timezone.now()
h.reminder_last_sent = d
with self.assertRaises(ValidationError):
h.save()
d = d.replace(minute=0, second=0, microsecond=0)
h.reminder_last_sent = d
h.save()
self.assertEqual(h.reminder_last_sent, d)
def test_scheduled_for_reminder_multiple_habits(self):
h1 = Habit(user=self.user,
start=datetime.date(2013, 3, 4),
description='Habit 1')
h1.set_reminder_schedule(SCHEDULE_MONDAYS, 16)
h1.save()
h2 = Habit(user=self.user,
start=datetime.date(2013, 3, 4),
description='Habit 2')
h2.set_reminder_schedule(SCHEDULE_MON_WED_FRI, 16)
h2.save()
self.assertEquals(len(self._scheduled(calendar.MONDAY, 0)), 0)
self.assertEquals(len(self._scheduled(calendar.MONDAY, 16)), 2)
self.assertEquals(len(self._scheduled(calendar.WEDNESDAY, 0)), 0)
self.assertEquals(len(self._scheduled(calendar.WEDNESDAY, 16)), 1)
