feat: add /healthz endpoint

PrivateAIM · Feb 22, 2024 · 7a0bc99 · 7a0bc99
1 parent 8de419a
commit 7a0bc99
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/project/server.py b/project/server.py
@@ -25,6 +25,12 @@ async def lifespan(app: FastAPI):
 
 app = FastAPI(lifespan=lifespan)
 
+
+@app.get("/healthz")
+async def do_healthcheck():
+    return {"status": "ok"}
+
+
 app.include_router(
     upload.router,
     prefix="/upload",

diff --git a/tests/test_health.py b/tests/test_health.py
@@ -0,0 +1,5 @@
+def test_200_healthcheck(test_client):
+    r = test_client.get("/healthz")
+
+    assert r.status_code == 200
+    assert r.json() == {"status": "ok"}