"""Tests del endpoint POST /predict."""

from unittest.mock import MagicMock

import pytest
from fastapi.testclient import TestClient

from src.api import main as api_main

PREDICT_RESPONSE_KEYS = {
    "text",
    "is_toxic",
    "probability",
    "labels",
    "model_used",
    "latency_ms",
}


@pytest.fixture
def client():
    mock_service = MagicMock()
    mock_service.predict.return_value = {
        "is_toxic": False,
        "probability": 0.12,
        "labels": [],
        "model_used": "LR + TF-IDF (local)",
    }

    with TestClient(api_main.app) as test_client:
        api_main._state["service"] = mock_service
        api_main._state["model_name"] = "LR + TF-IDF (local)"
        api_main._state["predictions_served"] = 0
        yield test_client

    api_main._state["service"] = None
    api_main._state["model_name"] = None


def test_predict_returns_correct_structure(client: TestClient):
    response = client.post(
        "/predict",
        json={"text": "This is a sample comment", "threshold": 0.5},
    )

    assert response.status_code == 200
    data = response.json()
    assert PREDICT_RESPONSE_KEYS <= set(data.keys())
    assert data["text"] == "This is a sample comment"
    assert isinstance(data["is_toxic"], bool)
    assert 0.0 <= data["probability"] <= 1.0
    assert isinstance(data["labels"], list)
    assert isinstance(data["model_used"], str)
    assert isinstance(data["latency_ms"], (int, float))


def test_predict_rejects_empty_text(client: TestClient):
    response = client.post("/predict", json={"text": "   "})

    assert response.status_code == 422