Merge pull request #19 from PrivateAIM/18-add-healthcheck-endpoint

Add healthcheck endpoint
PrivateAIM · Feb 22, 2024 · 6e919c5 · 6e919c5
2 parents 8de419a + 7a0bc99
commit 6e919c5
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"}