"""Promotes the ad-hoc TestClient checks used to verify /draft and
/escalate/check into pytest -- roadmap.md Phase 1 §2.5."""
from fastapi.testclient import TestClient

from app.main import app

client = TestClient(app)


def test_health():
    resp = client.get("/health")
    assert resp.status_code == 200
    assert resp.json() == {"status": "ok"}


def test_escalate_check_money():
    resp = client.post("/escalate/check", json={"text": "계좌번호 알려줄래?"})
    assert resp.status_code == 200
    assert resp.json() == {"escalate": True, "reason": "금전"}


def test_escalate_check_benign():
    resp = client.post("/escalate/check", json={"text": "오늘 저녁에 뭐 먹을래?"})
    assert resp.status_code == 200
    assert resp.json() == {"escalate": False, "reason": ""}


def test_escalate_check_requires_text_field():
    resp = client.post("/escalate/check", json={})
    assert resp.status_code == 422


def test_draft_with_style_examples_no_key(monkeypatch):
    # No GEMINI_API_KEY in this test environment -- assert the deterministic
    # no_key path rather than hitting the real Gemini API.
    monkeypatch.delenv("GEMINI_API_KEY", raising=False)
    resp = client.post(
        "/draft",
        json={
            "context_lines": ["상대: 오늘 저녁에 뭐 먹을래?"],
            "style_examples": ["ㅇㅇ 좋지", "나도 궁금하네ㅋㅋ"],
        },
    )
    assert resp.status_code == 200
    assert resp.json()["status"] == "no_key"


def test_draft_escalates_without_calling_style_source_logic():
    resp = client.post(
        "/draft",
        json={
            "context_lines": ["상대: 계좌번호 좀 알려줘"],
            "style_examples": ["ㅇㅇ 알겠어"],
        },
    )
    assert resp.status_code == 200
    assert resp.json() == {"status": "escalate", "text": "금전"}


def test_draft_with_history_uses_retrieval(monkeypatch):
    monkeypatch.delenv("GEMINI_API_KEY", raising=False)
    resp = client.post(
        "/draft",
        json={
            "context_lines": ["상대: 오늘 저녁에 뭐 먹을래?"],
            "history": ["어제 저녁엔 라면 먹었어", "핀란드는 교육이 좋대"],
            "k": 1,
        },
    )
    assert resp.status_code == 200
    body = resp.json()
    assert body["status"] == "no_key"
    # 검색된 스타일 예시가 프롬프트에 반영됐는지는 no_key 응답의 text(프롬프트
    # 원문)로 확인 -- retrieve()가 실제로 호출·반영됐다는 증거.
    assert "어제 저녁엔 라면 먹었어" in body["text"] or "핀란드는 교육이 좋대" in body["text"]


def test_draft_rejects_both_style_sources():
    resp = client.post(
        "/draft",
        json={
            "context_lines": ["상대: 안녕"],
            "style_examples": ["ㅇㅇ"],
            "history": ["ㅇㅇ"],
        },
    )
    assert resp.status_code == 422


def test_draft_rejects_neither_style_source():
    resp = client.post("/draft", json={"context_lines": ["상대: 안녕"]})
    assert resp.status_code == 422