"""Tests for FastAPI application"""

from unittest.mock import MagicMock, call, patch

import pytest
from fastapi.testclient import TestClient

from api.src.main import app, lifespan


@pytest.fixture
def test_client():
    """Create a test client"""
    return TestClient(app)


def test_health_check(test_client):
    """Test health check endpoint"""
    response = test_client.get("/health")
    assert response.status_code == 200
    assert response.json() == {"status": "healthy"}


@pytest.mark.asyncio
@patch("api.src.main.TTSModel")
@patch("api.src.main.logger")
async def test_lifespan_successful_warmup(mock_logger, mock_tts_model):
    """Test successful model warmup in lifespan"""
    # Mock file system for voice counting
    mock_tts_model.VOICES_DIR = "/mock/voices"

    # Create async mock
    async def async_setup():
        return 3

    mock_tts_model.setup = MagicMock()
    mock_tts_model.setup.side_effect = async_setup
    mock_tts_model.get_device.return_value = "cuda"

    with patch("os.listdir", return_value=["voice1.pt", "voice2.pt", "voice3.pt"]):
        # Create an async generator from the lifespan context manager
        async_gen = lifespan(MagicMock())
        # Start the context manager
        await async_gen.__aenter__()

        # Verify the expected logging sequence
        mock_logger.info.assert_any_call("Loading TTS model and voice packs...")

        # Check for the startup message containing the required info
        startup_calls = [call[0][0] for call in mock_logger.info.call_args_list]
        startup_msg = next(msg for msg in startup_calls if "Model warmed up on" in msg)
        assert "Model warmed up on" in startup_msg
        assert "3 voice packs loaded" in startup_msg

        # Verify model setup was called
        mock_tts_model.setup.assert_called_once()

        # Clean up
        await async_gen.__aexit__(None, None, None)


@pytest.mark.asyncio
@patch("api.src.main.TTSModel")
@patch("api.src.main.logger")
async def test_lifespan_failed_warmup(mock_logger, mock_tts_model):
    """Test failed model warmup in lifespan"""
    # Mock the model setup to fail
    mock_tts_model.setup.side_effect = RuntimeError("Failed to initialize model")

    # Create an async generator from the lifespan context manager
    async_gen = lifespan(MagicMock())

    # Verify the exception is raised
    with pytest.raises(RuntimeError, match="Failed to initialize model"):
        await async_gen.__aenter__()

    # Verify the expected logging sequence
    mock_logger.info.assert_called_with("Loading TTS model and voice packs...")

    # Clean up
    await async_gen.__aexit__(None, None, None)


@pytest.mark.asyncio
@patch("api.src.main.TTSModel")
async def test_lifespan_cuda_warmup(mock_tts_model):
    """Test model warmup specifically on CUDA"""
    # Mock file system for voice counting
    mock_tts_model.VOICES_DIR = "/mock/voices"

    # Create async mock
    async def async_setup():
        return 2

    mock_tts_model.setup = MagicMock()
    mock_tts_model.setup.side_effect = async_setup
    mock_tts_model.get_device.return_value = "cuda"

    with patch("os.listdir", return_value=["voice1.pt", "voice2.pt"]):
        # Create an async generator from the lifespan context manager
        async_gen = lifespan(MagicMock())
        await async_gen.__aenter__()

        # Verify model setup was called
        mock_tts_model.setup.assert_called_once()

        # Clean up
        await async_gen.__aexit__(None, None, None)