from __future__ import annotations

from fastapi.testclient import TestClient

from promptboost.main import app


def test_health_route(temp_db) -> None:
    client = TestClient(app)
    response = client.get("/health")

    assert response.status_code == 200
    assert response.json()["ok"] is True


def test_index_route(temp_db) -> None:
    client = TestClient(app)
    response = client.get("/")

    assert response.status_code == 200
    assert "PromptBoost" in response.text


def test_boost_and_runs_routes(temp_db) -> None:
    client = TestClient(app)
    response = client.post(
        "/boost",
        json={
            "raw_prompt": "Fix the API bug and add tests without unrelated refactors.",
            "target_harness": "codex",
            "target_model": "gpt",
        },
    )

    assert response.status_code == 200
    body = response.json()
    assert body["result"]["detected_task_type"] == "coding_patch"
    assert body["result"]["input_adequacy"]["status"] in {"adequate", "needs_clarification"}
    assert "no_new_facts_result" in body["result"]
    assert body["result"]["no_new_facts_passed"] is True
    assert body["run"]["id"] == 1

    runs = client.get("/api/runs").json()["runs"]
    assert len(runs) == 1


def test_promptgarage_samples_route(temp_db) -> None:
    client = TestClient(app)
    response = client.get("/api/promptgarage/samples?limit=2")

    assert response.status_code == 200
    assert "samples" in response.json()


def test_matrix_lab_page_and_deterministic_route(temp_db) -> None:
    client = TestClient(app)

    page = client.get("/matrix-lab")
    assert page.status_code == 200
    assert "Matrix Lab" in page.text

    response = client.post(
        "/api/matrix/deterministic",
        json={
            "raw_prompt": "Fix the API route, add tests, and avoid unrelated refactors.",
        },
    )

    assert response.status_code == 200
    body = response.json()
    assert len(body["harnesses"]) == 10
    assert len(body["models"]) == 10
    assert len(body["rows"]) == 100
    assert body["rows"][0]["metrics"]["words"] > 0