96 lines
3.7 KiB
Python
96 lines
3.7 KiB
Python
"""Tests for the ExerciseService class."""
|
|
|
|
from sqlmodel import SQLModel, Session, create_engine
|
|
|
|
from app.models.exercise import Exercise
|
|
from app.models.warmup import Warmup
|
|
from app.models.workout_day import WorkoutDay
|
|
from app.services.exercise_service import ExerciseService
|
|
|
|
|
|
class TestExerciseService:
|
|
"""Tests for exercise and warmup data access."""
|
|
|
|
def _setup(self):
|
|
"""Create an in-memory DB with seed data and return the service."""
|
|
engine = create_engine("sqlite:///:memory:")
|
|
SQLModel.metadata.create_all(engine)
|
|
session = Session(engine)
|
|
service = ExerciseService(session)
|
|
|
|
# Seed workout days
|
|
for i, (name, desc) in enumerate([
|
|
("Push", "Chest, shoulders, triceps"),
|
|
("Pull", "Back, biceps, traps"),
|
|
("Lower", "Quads, hamstrings, glutes, calves"),
|
|
("Full Body", "Compound movements"),
|
|
], start=1):
|
|
session.add(WorkoutDay(name=name, day_number=i, description=desc))
|
|
|
|
# Seed a few exercises
|
|
session.add(Exercise(name="DB Chest Press", muscle_group="Chest", workout_day="Push", sets=3, tempo="3-1-2", form_cues="..."))
|
|
session.add(Exercise(name="DB Row", muscle_group="Back", workout_day="Pull", sets=3, tempo="3-1-2", form_cues="..."))
|
|
|
|
# Seed warmups
|
|
session.add(Warmup(name="Cat / Cow", type="Thoracic Mob", reps="8 reps", form_cues="...", sort_order=1))
|
|
session.add(Warmup(name="Fire Hydrant", type="Hip Mobility", reps="8 each", form_cues="...", sort_order=2))
|
|
|
|
session.commit()
|
|
return session, service
|
|
|
|
def test_list_exercises_all(self) -> None:
|
|
"""list_exercises should return all exercises."""
|
|
session, service = self._setup()
|
|
exercises = service.list_exercises()
|
|
assert len(exercises) == 2
|
|
session.close()
|
|
|
|
def test_list_exercises_by_workout_day(self) -> None:
|
|
"""list_exercises should filter by workout day."""
|
|
session, service = self._setup()
|
|
exercises = service.list_exercises(workout_day="Push")
|
|
assert len(exercises) == 1
|
|
assert exercises[0].name == "DB Chest Press"
|
|
session.close()
|
|
|
|
def test_list_exercises_by_muscle_group(self) -> None:
|
|
"""list_exercises should filter by muscle group."""
|
|
session, service = self._setup()
|
|
exercises = service.list_exercises(muscle_group="Back")
|
|
assert len(exercises) == 1
|
|
session.close()
|
|
|
|
def test_get_exercise_by_id(self) -> None:
|
|
"""get_exercise_by_id should return the correct exercise."""
|
|
session, service = self._setup()
|
|
exercises = service.list_exercises()
|
|
found = service.get_exercise_by_id(exercises[0].id)
|
|
assert found is not None
|
|
session.close()
|
|
|
|
def test_get_exercise_by_name(self) -> None:
|
|
"""get_exercise_by_name should return the correct exercise."""
|
|
session, service = self._setup()
|
|
found = service.get_exercise_by_name("DB Chest Press")
|
|
assert found is not None
|
|
assert found.muscle_group == "Chest"
|
|
session.close()
|
|
|
|
def test_list_warmups_ordered(self) -> None:
|
|
"""list_warmups should return warmups in sort_order."""
|
|
session, service = self._setup()
|
|
warmups = service.list_warmups()
|
|
assert len(warmups) == 2
|
|
assert warmups[0].name == "Cat / Cow"
|
|
assert warmups[1].name == "Fire Hydrant"
|
|
session.close()
|
|
|
|
def test_list_workout_days(self) -> None:
|
|
"""list_workout_days should return all 4 days in order."""
|
|
session, service = self._setup()
|
|
days = service.list_workout_days()
|
|
assert len(days) == 4
|
|
assert days[0].name == "Push"
|
|
assert days[3].name == "Full Body"
|
|
session.close()
|