def test_delete_schedule(self): model = ModelService() start = "15/06 12:00" end = "15/06 13:00" formated_start = datetime.strptime(start, "%d/%m %H:%M") formated_end = datetime.strptime(end, "%d/%m %H:%M") user1_uid = 111111111 user1_username = '******' user1_description = "I love python" model.add_schedule(user1_uid, formated_start, formated_end) self.assertEqual(model.delete_schedule(uid=user1_uid, start=formated_start, end=formated_end), None)
def test_read_schedule(self): model = ModelService() start = "15/06 12:00" end = "15/06 13:00" formated_start = datetime.strptime(start, "%d/%m %H:%M") formated_end = datetime.strptime(end, "%d/%m %H:%M") user1_uid = 1111111121112 user1_username = '******' user1_description = "I love python" schedule = Schedule(UID=user1_uid, StartTime=formated_start, EndTime=formated_end) model.add_schedule(user1_uid, formated_start, formated_end) self.assertEqual(model.read_schedule(uid=user1_uid, start=formated_start, end=formated_end).UID, schedule.UID) model.delete_schedule(uid=user1_uid, start=formated_start, end=formated_end)
def test_find_meetings(self): model = ModelService() user1_uid = 9999 start = "15/06 12:00" end = "15/06 13:00" formated_start = datetime.strptime(start, "%d/%m %H:%M") formated_end = datetime.strptime(end, "%d/%m %H:%M") length = timedelta(hours=1) model.add_schedule(user1_uid, formated_start, formated_end) result = [] result.append({ 'user': user1_uid, 'start': formated_start, 'end': formated_end }) self.assertEqual(model.find_meetings(start=formated_start, end=formated_end, meeting_length=length)[0], result[0]) model.delete_schedule(uid=user1_uid, start=formated_start, end=formated_end)
def test_update_schedule(self): model = ModelService() start = "15/06 12:00" end = "15/06 13:00" formated_start = datetime.strptime(start, "%d/%m %H:%M") formated_end = datetime.strptime(end, "%d/%m %H:%M") user1_uid = 111111112111221 user1_username = '******' user1_description = "I love python" model.add_schedule(user1_uid, formated_start, formated_end) updated_start = "15/06 15:00" updated_end = "15/06 19:00" formated_updated_start = datetime.strptime(updated_start, "%d/%m %H:%M") formated_updated_end = datetime.strptime(updated_end, "%d/%m %H:%M") new_schedule = Schedule(UID=user1_uid, StartTime=formated_updated_start, EndTime=formated_updated_end) model.update_schedule(user1_uid, formated_start, formated_end, formated_updated_start, formated_updated_end) self.assertEqual(model.get_user_schedules(uid=user1_uid)[0].UID, new_schedule.UID) model.delete_schedule(uid=user1_uid, start=formated_updated_start, end=formated_updated_end)