Initial commit

tests/test_models.py

@@ -0,0 +1,24 @@
+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