class TestWorkout(unittest.TestCase):
def setUp(self):
self.date = datetime.date(2020, 12, 16)
self.time = datetime.time(6, 15)
self.workout_1 = Workout("Just Skate", "freestyle", self.date,
self.time, 3)
def test_workout_has_name(self):
self.assertEqual("Just Skate", self.workout_1.name)
def test_workout_has_category(self):
self.assertEqual("freestyle", self.workout_1.category)
def test_workout_has_no_members_booked(self):
self.assertEqual(0, self.workout_1.booked)
def test_workout_has_capacity(self):
self.assertEqual(3, self.workout_1.capacity)
# Date and time tests
def test_workout_has_date(self):
self.assertEqual(self.date, self.workout_1.date)
def test_workout_has_start_time(self):
self.assertEqual(self.time, self.workout_1.start_time)
def test_date_time__returns_date_and_time(self):
self.assertEqual("2020-12-16 06:15:00", self.workout_1.date_time())
def test_check_upcoming__returns_true(self):
self.assertEqual(True, self.workout_1.check_upcoming())
def test_check_upcoming__returns_false(self):
self.workout_1.date = datetime.date(2020, 12, 13)
self.assertEqual(False, self.workout_1.check_upcoming())
# Capacity tests
def test_increment_booked__returns_1(self):
self.workout_1.increment_booked()
self.assertEqual(1, self.workout_1.booked)
def test_decrease_booked__returns_0(self):
self.workout_1.increment_booked()
self.workout_1.decrease_booked()
self.assertEqual(0, self.workout_1.booked)
def test_check_capacity__returns_true(self):
self.assertEqual(True, self.workout_1.check_capacity())
def test_check_capacity__returns_false(self):
self.workout_1.increment_booked()
self.workout_1.increment_booked()
self.workout_1.increment_booked()
self.assertEqual(False, self.workout_1.check_capacity())
def select_upcoming():
workouts = []
sql = "SELECT * FROM workouts ORDER BY date ASC"
results = run_sql(sql)
for row in results:
workout = Workout(row['name'], row['category'], row['date'], row['start_time'], row['capacity'], row['booked'], row['id'])
if workout.check_upcoming():
workouts.append(workout)
return workouts