def test_shift_stores_in_database(self):
user_id = 1
start_time = now().replace(microsecond=0)
end_time = start_time + timedelta(hours=1)
shift = Shift(user_id, start_time, end_time, job_name='dd')
shift.save_to_database()
temp_shift = Shift.query.get(shift.id)
self.assertEqual(temp_shift.start_time, shift.start_time)
db.session.delete(shift)
db.session.commit()
def setUp(self):
db.create_all()
self.week_start = now().replace(hour=5, minute=0, second=0, microsecond=0)
self.week_end = self.week_start.replace(hour=10)
