feat: added db health check (#9)

* feat: added db health check

* feat: added error message to health check

* test: fixed health check test

Author: JanssenBrm

app/routers/health.py

@@ -1,8 +1,37 @@
-from fastapi import APIRouter
+from fastapi import APIRouter, Depends
+from loguru import logger
+from sqlalchemy.orm import Session
+
+from app.database.db import get_db
+from app.database.models.processing_job import ProcessingJobRecord
 
 router = APIRouter()
 
 
+def check_db_status(db: Session) -> dict:
+    try:
+        if db.is_active:
+            db.query(ProcessingJobRecord).limit(1).all()
+            db_status = "ok"
+            db_message = None
+        else:
+            raise Exception("Database session is not active")
+    except Exception as e:
+        logger.exception("Database connection check failed")
+        db_status = "error"
+        db_message = str(e)
+
+    status = {
+        "status": db_status,
+    }
+    if db_message:
+        status["message"] = db_message
+    return status
+
+
 @router.get("/health")
-async def health():
-    return {"status": "ok"}
+async def health(db: Session = Depends(get_db)):
+    return {
+        "status": "ok",
+        "database": check_db_status(db),
+    }

tests/routers/test_health.py

@@ -1,4 +1,10 @@
-def test_health(client):
+from unittest.mock import patch
+
+
+@patch("app.routers.health.check_db_status")
+def test_health(mock_db_status, client):
+    db_status = {"status": "error", "message": "Database connection failed"}
+    mock_db_status.return_value = db_status
     r = client.get("/health")
     assert r.status_code == 200
-    assert r.json() == {"status": "ok"}
+    assert r.json() == {"status": "ok", "database": db_status}