"""Coach Modelfile-builder endpoints + the separate builder page."""

from __future__ import annotations

from fastapi.testclient import TestClient

from mindxtrain.operator.app import app

client = TestClient(app)


def test_modelfile_page_served():
    r = client.get("/coach/modelfile")
    assert r.status_code == 200
    assert "Modelfile builder" in r.text
    assert "/coach/static/modelfile.js" in r.text


def test_modelfile_params_endpoint():
    r = client.get("/coach/api/modelfile/params")
    assert r.status_code == 200
    params = r.json()["parameters"]
    names = {p["name"] for p in params}
    assert {"num_ctx", "temperature", "top_k", "repeat_penalty"} <= names
    assert all("type" in p and "description" in p for p in params)


def test_modelfile_build_renders_text():
    r = client.post("/coach/api/modelfile/build", json={
        "from_model": "qwen3:0.6b",
        "system": "You are Codephreak.",
        "parameters": {"temperature": 0.7, "num_ctx": 4096},
        "stop": ["<|im_end|>"],
        "messages": [{"role": "user", "content": "who are you?"}],
    })
    assert r.status_code == 200, r.text
    mf = r.json()["modelfile"]
    assert mf.startswith("FROM qwen3:0.6b")
    assert "PARAMETER temperature 0.7" in mf
    assert "PARAMETER num_ctx 4096" in mf
    assert 'PARAMETER stop "<|im_end|>"' in mf
    assert 'SYSTEM """You are Codephreak."""' in mf
    assert "MESSAGE user who are you?" in mf


def test_modelfile_build_requires_from():
    r = client.post("/coach/api/modelfile/build", json={"system": "x"})
    assert r.status_code == 422


def test_modelfile_create_requires_tag():
    r = client.post("/coach/api/modelfile/create", json={"spec": {"from_model": "m"}})
    assert r.status_code == 422


def test_coach_index_links_modelfile_builder():
    r = client.get("/coach/")
    assert 'id="open-modelfile-btn"' in r.text
    js = client.get("/coach/static/coach.js").text
    assert "openModelfileBuilder" in js
    assert "/coach/modelfile" in js