| from unittest.mock import MagicMock |
| from fastapi.testclient import TestClient |
|
|
| from backend.app.main import app |
|
|
|
|
| def _make_mock_engine(ready=True, n_products=100): |
| engine = MagicMock() |
| engine._is_ready = ready |
| engine.metadata = MagicMock() |
| engine.metadata.__len__ = MagicMock(return_value=n_products) |
| engine.audit.return_value = {"status": "ready", "products": n_products} |
| return engine |
|
|
|
|
| class TestHealthEndpoints: |
| def test_health_ok(self): |
| client = TestClient(app, raise_server_exceptions=False) |
| app.state.engine = _make_mock_engine() |
| response = client.get("/api/v1/health") |
| assert response.status_code == 200 |
| data = response.json() |
| assert data["status"] == "ok" |
| assert data["engine_ready"] is True |
| assert data["products"] == 100 |
|
|
| def test_health_not_ready(self): |
| client = TestClient(app, raise_server_exceptions=False) |
| app.state.engine = _make_mock_engine(ready=False) |
| response = client.get("/api/v1/health") |
| data = response.json() |
| assert data["engine_ready"] is False |
|
|
| def test_health_no_engine(self): |
| client = TestClient(app, raise_server_exceptions=False) |
| app.state.engine = None |
| response = client.get("/api/v1/health") |
| data = response.json() |
| assert data["engine_ready"] is False |
|
|
| def test_audit(self): |
| client = TestClient(app, raise_server_exceptions=False) |
| app.state.engine = _make_mock_engine() |
| response = client.get("/api/v1/audit") |
| assert response.status_code == 200 |
| data = response.json() |
| assert data["status"] == "ready" |
|
|
