Add seasons tests

Author: koldakov

tests/conftest.py

@@ -58,6 +58,33 @@ def episode(faker: Faker, production_code):
     )
 
 
+@pytest.fixture
+def season(faker: Faker, production_code):
+    return SeasonModel(
+        id=faker.random_int(
+            min=1,
+            max=INT32,
+        ),
+        episodes=[
+            EpisodeModel(
+                id=faker.random_int(
+                    min=1,
+                    max=INT32,
+                ),
+                created_at=faker.date_time(),
+                name=faker.name(),
+                air_date=faker.date_object(),
+                duration=faker.random_int(),
+                production_code=production_code,
+                broadcast_number=faker.random_int(
+                    min=1,
+                    max=INT32,
+                ),
+            ),
+        ],
+    )
+
+
 @pytest.fixture
 def mock_session_manager(request):
     mock_context = AsyncMock()

tests/unit/routers/services/seasons/__init__.py

@@ -0,0 +1,46 @@
+from unittest.mock import MagicMock
+
+import pytest
+from faker import Faker
+from fastapi import HTTPException
+from sqlalchemy.exc import NoResultFound
+
+from futuramaapi.repositories import INT32
+from futuramaapi.repositories.models import SeasonModel
+from futuramaapi.routers.services.seasons.get_season import GetSeasonService
+
+
+class TestGetSeasonService:
+    @pytest.mark.asyncio
+    async def test_get_season_service_success(self, season: SeasonModel, mock_session_manager):
+        # Arrange
+        mock_result = MagicMock()
+        mock_result.scalars.return_value.one.return_value = season
+        mock_session_manager.execute.return_value = mock_result
+
+        service = GetSeasonService(pk=season.id)
+
+        # Act
+        result = await service()
+
+        # Assert
+        assert result.id == season.id
+        assert result.episodes[0].id == season.episodes[0].id
+
+    @pytest.mark.asyncio
+    async def test_get_season_service_not_found(self, faker: Faker, mock_session_manager):
+        # Arrange
+        mock_result = MagicMock()
+        mock_result.scalars.return_value.one.side_effect = NoResultFound
+        mock_session_manager.execute.return_value = mock_result
+
+        service = GetSeasonService(
+            pk=faker.random_int(
+                min=1,
+                max=INT32,
+            ),
+        )
+
+        # Act & Assert
+        with pytest.raises(HTTPException):
+            await service()

tests/unit/routers/services/seasons/test_get_season.py

@@ -0,0 +1,40 @@
+from unittest.mock import AsyncMock, MagicMock, patch
+
+import pytest
+
+from futuramaapi.routers.services.seasons.list_seasons import ListSeasonsService
+
+
+@pytest.fixture
+def mock_paginate(request):
+    patcher = patch(
+        "futuramaapi.routers.services.seasons.list_seasons.paginate",
+        new_callable=AsyncMock,
+    )
+    mocked = patcher.start()
+    request.addfinalizer(patcher.stop)
+    return mocked
+
+
+class TestListSeasonsService:
+    @pytest.mark.asyncio
+    async def test_list_seasons_success(
+        self,
+        mock_session_manager,
+        mock_paginate,
+    ):
+        # Arrange
+        page = MagicMock()
+        mock_paginate.return_value = page
+
+        service = ListSeasonsService()
+
+        # Act
+        result = await service()
+
+        # Assert
+        assert result is page
+        mock_paginate.assert_called_once()
+
+        args, _ = mock_paginate.call_args
+        assert args[0] is mock_session_manager