import fastapi
import pytest
from fastapi.testclient import TestClient

import api


def test_health_readiness_and_bearer_auth(monkeypatch):
    client = TestClient(api.app)
    assert client.get("/healthz").status_code == 200
    assert client.get("/readyz").status_code == 503
    monkeypatch.setenv("NAYARA_API_KEY", "test-secret")
    assert client.get("/v1/models").status_code == 401
    response = client.get("/v1/models", headers={"Authorization": "Bearer test-secret"})
    assert response.status_code == 200
    assert response.json()["data"][0]["id"] == "nayara-v1"


def test_chat_prompt_requires_user_message():
    with pytest.raises(fastapi.HTTPException):
        api._chat_prompt([api.ChatMessage(role="system", content="Be concise")])
    prompt = api._chat_prompt([api.ChatMessage(role="user", content="hello")])
    assert prompt.endswith("<|assistant|>\n")