from fastapi.testclient import TestClient

from yuqei_ai_platform_api.config import AiPlatformSettings
from yuqei_ai_platform_api.main import create_app
from yuqei_ai_platform_api.repository import InMemoryAiPlatformRepository


def make_client() -> TestClient:
    repository = InMemoryAiPlatformRepository()
    repository.seed_defaults()
    app = create_app(AiPlatformSettings(environment="test"), repository=repository)
    return TestClient(app)


def test_model_config_can_be_hot_updated_and_listed() -> None:
    client = make_client()

    response = client.post(
        "/api/v1/model-configs",
        json={
            "provider": "openai-compatible",
            "model": "legal-pro",
            "display_name": "Legal Pro",
            "base_url": "https://ai.example.test/v1",
            "temperature": 0.1,
            "max_tokens": 8192,
            "enabled": True,
            "is_default": True,
        },
    )

    assert response.status_code == 200
    created = response.json()
    assert created["model"] == "legal-pro"
    assert created["is_default"] is True

    list_response = client.get("/api/v1/model-configs")
    assert list_response.status_code == 200
    configs = list_response.json()
    assert configs[0]["model"] == "legal-pro"
    assert sum(1 for config in configs if config["is_default"]) == 1


def test_knowledge_document_can_be_added_and_searched() -> None:
    client = make_client()

    add_response = client.post(
        "/api/v1/knowledge-documents",
        json={
            "knowledge_base_id": "enterprise",
            "title": "公司租赁审查规范",
            "source_type": "internal_policy",
            "reference": "租赁-001",
            "content": "房屋租赁合同必须明确押金返还条件。",
        },
    )
    assert add_response.status_code == 200

    search_response = client.get("/api/v1/knowledge-documents/search", params={"keyword": "押金"})
    assert search_response.status_code == 200
    assert search_response.json()[0]["title"] == "公司租赁审查规范"


def test_legal_qa_uses_knowledge_and_records_audit() -> None:
    client = make_client()
    client.post(
        "/api/v1/knowledge-documents",
        json={
            "knowledge_base_id": "laws-cn",
            "title": "中华人民共和国劳动合同法",
            "source_type": "law",
            "reference": "第十九条",
            "content": "试用期不得超过法定上限。",
        },
    )

    response = client.post(
        "/api/v1/legal/qa",
        json={
            "question": "劳动合同试用期怎么约定？",
            "knowledge_base_ids": ["laws-cn"],
        },
    )

    assert response.status_code == 200
    payload = response.json()
    assert payload["citations"][0]["reference"] == "第十九条"

    audit_response = client.get("/api/v1/audit/ai-runs")
    assert audit_response.status_code == 200
    audits = audit_response.json()
    assert audits[0]["operation"] == "legal_qa"
    assert audits[0]["citations_count"] >= 1
    assert audits[0]["status"] == "succeeded"