def test_one_shift_in_range_returns_none_for_other_days(self): user = User('*****@*****.**', 'u3', username='******', _id='u3') user.save_to_database() # USER HAS SHIFT FOT ONLY MONDAY start = beginning_of_week() end = start + timedelta(seconds=1) # create 1 second shift shift = Shift(user._id, start, end) shift.save_to_database() schedule = Schedule(user._id) # Monday is not None self.assertEqual(type(schedule.shifts[0]), type(shift)) self.assertEqual(schedule.row[1], None) self.assertEqual(schedule.row[2], None) self.assertEqual(schedule.row[3], None) self.assertEqual(schedule.row[4], None) self.assertEqual(schedule.row[5], None) self.assertEqual(schedule.row[6], None) Database.remove('users', {'user_id': user._id}) Database.remove('shifts', {'user_id': user._id})
def test_monday_and_wednesday_day_are_shifts_rest_are_none(self): user = User('*****@*****.**', 'u4', username='******', _id='u4') user.save_to_database() # USER HAS SHIFT FOT ONLY MONDAY monday = beginning_of_week() wednesday = beginning_of_week(days=2) # create 1 second shift shift = Shift(user._id, monday, monday+timedelta(seconds=1)) shift.save_to_database() shift = Shift(user._id, wednesday, wednesday+timedelta(seconds=1)) shift.save_to_database() schedule = Schedule(user._id) # Monday is not None self.assertEqual(type(schedule.row[0]), type(shift)) self.assertEqual(schedule.row[1], None) self.assertEqual(type(schedule.row[2]), type(shift)) Database.remove('users', {'user_id': user._id}) Database.remove('shifts', {'user_id': user._id})