Trakt

"""Tests for related shows client module."""

import os
from unittest.mock import AsyncMock, MagicMock, patch

import pytest

from client.shows.related import RelatedShowsClient


@pytest.mark.asyncio
async def test_get_related_shows_auto_pagination():
    """Test get_related_shows with auto-pagination mode (page=None)."""
    mock_response = MagicMock()
    mock_response.json.return_value = [
        {
            "title": "Better Call Saul",
            "year": 2015,
            "overview": "A prequel to Breaking Bad.",
            "ids": {"trakt": 59660, "slug": "better-call-saul"},
        },
        {
            "title": "The Wire",
            "year": 2002,
            "overview": "A crime drama set in Baltimore.",
            "ids": {"trakt": 1234, "slug": "the-wire"},
        },
    ]
    mock_response.headers = {
        "X-Pagination-Page": "1",
        "X-Pagination-Limit": "10",
        "X-Pagination-Page-Count": "1",
        "X-Pagination-Item-Count": "2",
    }
    mock_response.raise_for_status = MagicMock()

    with (
        patch("httpx.AsyncClient") as mock_client,
        patch.dict(
            os.environ,
            {"TRAKT_CLIENT_ID": "test_id", "TRAKT_CLIENT_SECRET": "test_secret"},
        ),
    ):
        mock_instance = MagicMock()
        mock_instance.get = AsyncMock(return_value=mock_response)
        mock_instance.post = AsyncMock()
        mock_instance.aclose = AsyncMock()
        mock_client.return_value = mock_instance

        client = RelatedShowsClient()
        result = await client.get_related_shows("breaking-bad", limit=10)

        assert isinstance(result, list)
        assert len(result) == 2
        assert result[0]["title"] == "Better Call Saul"
        assert result[1]["title"] == "The Wire"

        mock_response.raise_for_status.assert_called()
        mock_instance.aclose.assert_awaited_once()


@pytest.mark.asyncio
async def test_get_related_shows_single_page():
    """Test get_related_shows with single-page mode (page=N)."""
    mock_response = MagicMock()
    mock_response.json.return_value = [
        {
            "title": "Better Call Saul",
            "year": 2015,
            "ids": {"trakt": 59660, "slug": "better-call-saul"},
        }
    ]
    mock_response.headers = {
        "X-Pagination-Page": "2",
        "X-Pagination-Limit": "10",
        "X-Pagination-Page-Count": "3",
        "X-Pagination-Item-Count": "25",
    }
    mock_response.raise_for_status = MagicMock()

    with (
        patch("httpx.AsyncClient") as mock_client,
        patch.dict(
            os.environ,
            {"TRAKT_CLIENT_ID": "test_id", "TRAKT_CLIENT_SECRET": "test_secret"},
        ),
    ):
        mock_instance = MagicMock()
        mock_instance.get = AsyncMock(return_value=mock_response)
        mock_instance.post = AsyncMock()
        mock_instance.aclose = AsyncMock()
        mock_client.return_value = mock_instance

        client = RelatedShowsClient()
        result = await client.get_related_shows("breaking-bad", limit=10, page=2)

        # Single-page mode returns PaginatedResponse with pagination metadata
        assert hasattr(result, "data")
        assert hasattr(result, "pagination")
        assert result.pagination.current_page == 2
        assert len(result.data) == 1
        assert result.data[0]["title"] == "Better Call Saul"

        mock_response.raise_for_status.assert_called()
        mock_instance.aclose.assert_awaited_once()


@pytest.mark.asyncio
async def test_get_related_shows_url_encoding():
    """Test that show_id with special characters is URL encoded."""
    mock_response = MagicMock()
    mock_response.json.return_value = []
    mock_response.headers = {
        "X-Pagination-Page": "1",
        "X-Pagination-Limit": "10",
        "X-Pagination-Page-Count": "1",
        "X-Pagination-Item-Count": "0",
    }
    mock_response.raise_for_status = MagicMock()

    with (
        patch("httpx.AsyncClient") as mock_client,
        patch.dict(
            os.environ,
            {"TRAKT_CLIENT_ID": "test_id", "TRAKT_CLIENT_SECRET": "test_secret"},
        ),
    ):
        mock_instance = MagicMock()
        mock_instance.get = AsyncMock(return_value=mock_response)
        mock_instance.post = AsyncMock()
        mock_instance.aclose = AsyncMock()
        mock_client.return_value = mock_instance

        client = RelatedShowsClient()
        # Test with a slug containing special characters that need encoding
        # / -> %2F, : -> %3A, & -> %26
        await client.get_related_shows("show/with:special&chars", limit=5)

        # Verify the request was made
        mock_instance.get.assert_called()
        call_args = mock_instance.get.call_args
        # The endpoint should contain the URL-encoded show_id
        assert "show%2Fwith%3Aspecial%26chars" in str(call_args)

        mock_instance.aclose.assert_awaited_once()


@pytest.mark.asyncio
async def test_get_related_shows_empty_results():
    """Test get_related_shows with no related shows found."""
    mock_response = MagicMock()
    mock_response.json.return_value = []
    mock_response.headers = {
        "X-Pagination-Page": "1",
        "X-Pagination-Limit": "10",
        "X-Pagination-Page-Count": "1",
        "X-Pagination-Item-Count": "0",
    }
    mock_response.raise_for_status = MagicMock()

    with (
        patch("httpx.AsyncClient") as mock_client,
        patch.dict(
            os.environ,
            {"TRAKT_CLIENT_ID": "test_id", "TRAKT_CLIENT_SECRET": "test_secret"},
        ),
    ):
        mock_instance = MagicMock()
        mock_instance.get = AsyncMock(return_value=mock_response)
        mock_instance.post = AsyncMock()
        mock_instance.aclose = AsyncMock()
        mock_client.return_value = mock_instance

        client = RelatedShowsClient()
        result = await client.get_related_shows("obscure-show", limit=10)

        assert isinstance(result, list)
        assert len(result) == 0

        mock_instance.aclose.assert_awaited_once()