import pytest


@pytest.mark.parametrize("case", list(range(120)))
def test_list_models_cases(api_client, case):
    resp = api_client.get("/v1/models", headers={"x-case": str(case)})
    assert resp.status_code == 200
    payload = resp.json()
    assert payload["object"] == "list"
    assert isinstance(payload["data"], list)


@pytest.mark.parametrize("model_id", [f"model-a.gguf" for _ in range(120)])
def test_get_model_ok(api_client, model_id):
    resp = api_client.get(f"/v1/models/{model_id}")
    assert resp.status_code == 200
    payload = resp.json()
    assert payload["id"] == model_id


@pytest.mark.parametrize("model_id", [f"missing-{i}" for i in range(120)])
def test_get_model_not_found(api_client, model_id):
    resp = api_client.get(f"/v1/models/{model_id}")
    assert resp.status_code == 404