"""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()