def test_exercise_model(self): """Test basic exercise model.""" # Create test exercises e1 = Exercise(name="Bench Press", description="Chest exercise", category_id=10) e1.id = 500 e2 = Exercise(name="Triceps Pushdown", description="Triceps exercise", category_id=11) e2.id = 501 db.session.add_all([e1, e2]) db.session.commit() e = Exercise.query.all() self.assertEqual(len(e), 2) self.assertEqual(e[0].id, e1.id) self.assertEqual(e[0].name, "Bench Press") self.assertEqual(e[0].description, "Chest exercise") self.assertEqual(e[1].id, e2.id) self.assertEqual(e[1].name, "Triceps Pushdown") self.assertEqual(e[1].description, "Triceps exercise")
def test_user_exercise_model(self): """Test adding exercises to a user's exercises functionality.""" # Create test exercises e1 = Exercise(name="Bench Press", description="Chest exercise", category_id=10) e1.id = 500 e2 = Exercise(name="Triceps Pushdown", description="Triceps exercise", category_id=11) e2.id = 501 # Create test user u = User.register("testing1", "password", "*****@*****.**", "John", "Doe", None) u.id = 777 db.session.add_all([e1, e2]) ue1 = UserExercise(user_id=u.id, exercise_id=e1.id) ue2 = UserExercise(user_id=u.id, exercise_id=e2.id) db.session.add_all([ue1, ue2]) db.session.commit() exc = UserExercise.query.filter_by(user_id=u.id) self.assertEqual(exc.count(), 2) self.assertEqual(exc[0].exercise_id, e1.id) self.assertEqual(exc[1].exercise_id, e2.id)
def test_exercise_comment_model(self): """Test user commentting functionality on an exercise.""" # Create test exercises e1 = Exercise(name="Bench Press", description="Chest exercise", category_id=10) e1.id = 500 e2 = Exercise(name="Triceps Pushdown", description="Triceps exercise", category_id=11) e2.id = 501 # Create test user u = User.register("testing1", "password", "*****@*****.**", "John", "Doe", None) u.id = 777 db.session.add_all([e1, e2]) uc1 = ExerciseComment(content="Test1", user_id=u.id, exercise_id=e1.id) uc2 = ExerciseComment(content="Test2", user_id=u.id, exercise_id=e2.id) db.session.add_all([uc1, uc2]) db.session.commit() ucs = ExerciseComment.query.filter_by(user_id=u.id) self.assertEqual(ucs.count(), 2) self.assertEqual(ucs[0].exercise_id, e1.id) self.assertEqual(ucs[0].user_id, u.id) self.assertEqual(ucs[1].exercise_id, e2.id) self.assertEqual(ucs[1].user_id, u.id)