def test_shift(self):
agenda = AgendaController()
shift = agenda.add_shift(9, 14)
shift = agenda.get_shift(shift.key)
self.assertEqual(shift.interval, Interval(9, 14))
shifts = list(agenda.get_shifts_itervalues())
self.assertEqual([s.key for s in shifts], [shift.key])
self.assertEqual([s.interval for s in shifts], [shift.interval])
agenda.del_shift(shift.key)
shifts = list(agenda.get_shifts_itervalues())
self.assertEqual(shifts, [])
agenda.destroy()
def test_appointment(self):
agenda = AgendaController()
shift = agenda.add_shift(9, 14)
with self.assertRaises(NotAvailableSlotError):
agenda.add_appointment(8, 9)
app1 = agenda.add_appointment(9, 10)
app1 = agenda.get_appointment(app1.key)
self.assertEqual(app1.interval, Interval(9, 10))
agenda.del_appointment(app1.key)
with self.assertRaises(KeyError):
app1 = agenda.get_appointment(app1.key)
app2 = agenda.add_appointment(9, 10)
app2 = agenda.get_appointment(app2.key)
self.assertEqual(app2.interval, Interval(9, 10))
appos = list(agenda.get_appointments_in_shift_iteritems(shift.key))
print appos
with self.assertRaises(NotAvailableSlotError):
agenda.add_appointment(9, 11)
with self.assertRaises(NotAvailableSlotError):
agenda.add_appointment(9, 11)
with self.assertRaises(ShiftNotEmptyError):
agenda.del_shift(shift.key)
agenda.del_appointment(app2.key)
with self.assertRaises(KeyError):
app1 = agenda.get_appointment(app1)
agenda.del_shift(shift.key)
agenda.destroy()
