""" Tests for models listing router. """ import pytest from fastapi.testclient import TestClient @pytest.mark.unit def test_list_models(client: TestClient) -> None: """Test listing available models.""" response = client.get("/v1/models") assert response.status_code == 200 data = response.json() # Verify response structure assert data["object"] == "list" assert "data" in data assert isinstance(data["data"], list) assert len(data["data"]) > 0 # Verify model structure model = data["data"][0] assert model["object"] == "model" assert "id" in model assert model["id"] == "mistral-nemo:latest" assert "created" in model assert model["owned_by"] == "system" @pytest.mark.unit def test_models_endpoint_returns_json(client: TestClient) -> None: """Test that models endpoint returns valid JSON.""" response = client.get("/v1/models") assert response.status_code == 200 assert response.headers["content-type"] == "application/json" # Should be able to parse as JSON data = response.json() assert isinstance(data, dict)