"""Contract tests for the FastAPI endpoints without loading model weights."""

import unittest

from fastapi.testclient import TestClient

import api.main as api


class FakeAnalyzer:
    model_id = "fake/model"
    device = "cpu"
    is_loaded = True

    def predict(self, text):
        return {
            "label": "positif",
            "confidence": 0.9,
            "probabilities": {"negatif": 0.1, "positif": 0.9},
        }


class ApiContractTests(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        api.analyzer = FakeAnalyzer()
        cls.client = TestClient(api.app)

    def test_health(self):
        response = self.client.get("/health")
        self.assertEqual(response.status_code, 200)
        self.assertTrue(response.json()["model_loaded"])

    def test_prediction(self):
        response = self.client.post("/predict", json={"text": "  Super   film  "})
        self.assertEqual(response.status_code, 200)
        self.assertEqual(response.json()["label"], "positif")

    def test_invalid_texts(self):
        self.assertEqual(self.client.post("/predict", json={"text": " "}).status_code, 400)
        self.assertEqual(
            self.client.post("/predict", json={"text": "x" * 5001}).status_code,
            400,
        )
        self.assertEqual(self.client.post("/predict", json={}).status_code, 422)


if __name__ == "__main__":
    unittest.main()