Jeroen SchweitzerandClaude Opus 4.5
c1f16d44e5
- Remove src/models/ollama_client.py, embeddings.py, embeddings_ollama.py - Remove model aliases and AI config from settings (both config.py files) - Update health endpoints to only check database connectivity - Update tests to reflect database-only health checks - Update README, .env.example, and OIDC docstrings 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
155 lines
4.9 KiB
Python
155 lines
4.9 KiB
Python
"""Tests for health endpoints."""
|
|
import pytest
|
|
from fastapi.testclient import TestClient
|
|
from unittest.mock import patch, AsyncMock
|
|
|
|
from src.main import app
|
|
|
|
|
|
@pytest.fixture
|
|
def client():
|
|
"""Create a test client."""
|
|
return TestClient(app)
|
|
|
|
|
|
class TestRootEndpoint:
|
|
"""Test root endpoint."""
|
|
|
|
def test_root_returns_200(self, client):
|
|
"""Root endpoint should return 200."""
|
|
response = client.get("/")
|
|
assert response.status_code == 200
|
|
|
|
def test_root_returns_service_info(self, client):
|
|
"""Root endpoint should return service information."""
|
|
response = client.get("/")
|
|
data = response.json()
|
|
|
|
assert "service" in data
|
|
assert "version" in data
|
|
assert "status" in data
|
|
assert data["service"] == "Core Code API"
|
|
assert data["status"] == "healthy"
|
|
|
|
def test_root_returns_docs_link(self, client):
|
|
"""Root endpoint should return docs link."""
|
|
response = client.get("/")
|
|
data = response.json()
|
|
|
|
assert "docs" in data
|
|
assert data["docs"] == "/docs"
|
|
|
|
|
|
class TestHealthEndpoint:
|
|
"""Test /health endpoint."""
|
|
|
|
def test_health_returns_200(self, client):
|
|
"""Health endpoint should return 200."""
|
|
response = client.get("/health")
|
|
assert response.status_code == 200
|
|
|
|
def test_health_returns_status(self, client):
|
|
"""Health endpoint should return status information."""
|
|
response = client.get("/health")
|
|
data = response.json()
|
|
|
|
assert "status" in data
|
|
assert "version" in data
|
|
assert data["status"] == "healthy"
|
|
|
|
|
|
class TestOpenAPIEndpoint:
|
|
"""Test OpenAPI documentation endpoints."""
|
|
|
|
def test_openapi_spec_available(self, client):
|
|
"""OpenAPI spec should be available."""
|
|
response = client.get("/openapi.json")
|
|
assert response.status_code == 200
|
|
|
|
data = response.json()
|
|
assert "openapi" in data
|
|
assert "info" in data
|
|
|
|
def test_swagger_ui_available(self, client):
|
|
"""Swagger UI should be available."""
|
|
response = client.get("/docs")
|
|
assert response.status_code == 200
|
|
|
|
|
|
class TestFullHealthCheck:
|
|
"""Test /health/full endpoint."""
|
|
|
|
@patch("src.shared.database.Database.health_check")
|
|
def test_full_health_returns_200_when_healthy(self, mock_db_health, client):
|
|
"""Full health should return 200 when database is healthy."""
|
|
mock_db_health.return_value = True
|
|
|
|
response = client.get("/health/full")
|
|
assert response.status_code == 200
|
|
|
|
@patch("src.shared.database.Database.health_check")
|
|
def test_full_health_returns_503_when_unhealthy(self, mock_db_health, client):
|
|
"""Full health should return 503 when database is unhealthy."""
|
|
mock_db_health.return_value = False
|
|
|
|
response = client.get("/health/full")
|
|
assert response.status_code == 503
|
|
|
|
@patch("src.shared.database.Database.health_check")
|
|
def test_full_health_returns_components_status(self, mock_db_health, client):
|
|
"""Full health should return component status."""
|
|
mock_db_health.return_value = True
|
|
|
|
response = client.get("/health/full")
|
|
data = response.json()
|
|
|
|
assert "status" in data
|
|
assert "components" in data
|
|
assert "database" in data["components"]
|
|
assert "response_time_ms" in data
|
|
|
|
@patch("src.shared.database.Database.health_check")
|
|
def test_full_health_handles_database_error(self, mock_db_health, client):
|
|
"""Full health should handle database errors gracefully."""
|
|
mock_db_health.side_effect = Exception("Connection error")
|
|
|
|
response = client.get("/health/full")
|
|
data = response.json()
|
|
|
|
assert response.status_code == 503
|
|
assert data["status"] == "unhealthy"
|
|
assert "error" in data["components"]["database"]
|
|
|
|
|
|
class TestDiagnosticsEndpoint:
|
|
"""Test /health/diagnostics endpoint."""
|
|
|
|
def test_diagnostics_returns_200(self, client):
|
|
"""Diagnostics should return 200."""
|
|
response = client.get("/health/diagnostics")
|
|
assert response.status_code == 200
|
|
|
|
def test_diagnostics_returns_service_info(self, client):
|
|
"""Diagnostics should return service information."""
|
|
response = client.get("/health/diagnostics")
|
|
data = response.json()
|
|
|
|
assert "service" in data
|
|
assert "name" in data["service"]
|
|
assert "version" in data["service"]
|
|
|
|
def test_diagnostics_returns_configuration(self, client):
|
|
"""Diagnostics should return configuration info."""
|
|
response = client.get("/health/diagnostics")
|
|
data = response.json()
|
|
|
|
assert "configuration" in data
|
|
|
|
def test_diagnostics_returns_response_time(self, client):
|
|
"""Diagnostics should return response time."""
|
|
response = client.get("/health/diagnostics")
|
|
data = response.json()
|
|
|
|
assert "response_time_ms" in data
|
|
assert isinstance(data["response_time_ms"], int)
|