add liveness probe router

papermerge/app.py

@@ -18,6 +18,7 @@
 from papermerge.core.features.document.router_document_version import (
     router as document_versions_router,
 )
+from papermerge.core.features.liveness_probe.router import router as probe_router
 from papermerge.search.routers.search import router as search_router
 
 from papermerge.core.version import __version__
@@ -47,6 +48,7 @@
 app.include_router(usr_router, prefix=prefix)
 app.include_router(tags_router, prefix=prefix)
 app.include_router(groups_router, prefix=prefix)
+app.include_router(probe_router, prefix=prefix)
 
 if config.papermerge__search__url:
     app.include_router(search_router, prefix=prefix)

papermerge/core/features/conftest.py

@@ -35,6 +35,7 @@
 from papermerge.core.features.groups import router as groups_router
 from papermerge.core.features.tags import router as tags_router
 from papermerge.core.features.users import router as usr_router
+from papermerge.core.features.liveness_probe import router as probe_router
 from papermerge.core import orm, dbapi, schema
 from papermerge.core import utils
 from papermerge.core.tests.types import AuthTestClient
@@ -276,6 +277,7 @@ def get_app_with_routes():
     app.include_router(thumbnails_router.router, prefix="")
     app.include_router(usr_router.router, prefix="")
     app.include_router(tags_router.router, prefix="")
+    app.include_router(probe_router.router, prefix="")
 
     return app
 

papermerge/core/routers/probe.py → papermerge/core/features/liveness_probe/router.py

@@ -1,7 +1,7 @@
 from fastapi import APIRouter, Response
 from sqlalchemy import text
 
-from papermerge.core.db import get_engine
+from papermerge.core.db.engine import Session
 
 router = APIRouter(
     prefix="/probe",
@@ -12,8 +12,7 @@
 @router.get("/")
 def probe_endpoint():
     """Liveness probe endpoint"""
-    engine = get_engine()
-    with engine.connect() as connection:
-        connection.execute(text("select 1"))
+    with Session() as db_session:
+        db_session.execute(text("select 1"))
 
     return Response()

papermerge/core/features/liveness_probe/tests/test_router.py

@@ -0,0 +1,3 @@
+def test_liveness_probe(api_client):
+    response = api_client.get("/probe")
+    assert response.status_code == 200, response.json()