def testGetBlackedOutElectivePeriods(self):
wa = ElectiveAlerts()
periods = self.sesshun.period_set.all()
now = datetime(2009, 4, 7, 12)
stats = wa.getBlackedOutElectivePeriods(now, self.sesshun.elective_set.all())
exp = [(self.elective1, [periods[2]]), (self.elective2, [periods[5]])]
self.assertEquals(exp, stats)
now = datetime(2009, 4, 8, 12)
stats = wa.getBlackedOutElectivePeriods(now, self.sesshun.elective_set.all())
exp = [(self.elective2, [periods[5]])]
self.assertEquals(exp, stats)
def testFindBlackoutAlerts(self):
wa = ElectiveAlerts()
now = datetime(2009, 3, 24, 0)
periods = self.sesshun.period_set.all()
# stage 1 check on 3/24 reveals blacked-out period on 4/14
alerts = wa.findBlackoutAlerts(1, now, self.sesshun.elective_set.all())
exp = [(self.elective2, [periods[5]], 1)]
self.assertEquals(exp, alerts)
# stage 2 check on 3/24 reveals blacked-out period on 4/7
alerts = wa.findBlackoutAlerts(2, now, self.sesshun.elective_set.all())
exp = [(self.elective1, [periods[2]], 2)]
self.assertEquals(exp, alerts)
