add db healthcheck endpoint

Author: qorexdevs

app/routers/api.py

@@ -4,7 +4,8 @@
 import qrcode
 from fastapi import APIRouter, Depends, HTTPException
 from fastapi.responses import RedirectResponse, StreamingResponse
-from sqlalchemy.exc import IntegrityError
+from sqlalchemy import text
+from sqlalchemy.exc import IntegrityError, SQLAlchemyError
 from sqlalchemy.ext.asyncio import AsyncSession
 
 from app.config import BASE_URL, MAX_TTL_HOURS
@@ -77,6 +78,14 @@ async def shorten_url(data: ShortenRequest, session: AsyncSession = Depends(get_
         short_code=code,
     )
 
+@router.get("/api/health")
+async def health(session: AsyncSession = Depends(get_session)):
+    try:
+        await session.execute(text("SELECT 1"))
+    except SQLAlchemyError:
+        raise HTTPException(status_code=503, detail="database unavailable")
+    return {"status": "ok"}
+
 @router.get("/api/stats/{code}", response_model=LinkStats)
 async def get_stats(code: str, session: AsyncSession = Depends(get_session)):
     link = await find_link(session, code)

tests/test_api.py

@@ -8,6 +8,13 @@
 from conftest import test_session as db_session_factory
 
 
+@pytest.mark.asyncio
+async def test_health_ok(client):
+    res = await client.get("/api/health")
+    assert res.status_code == 200
+    assert res.json() == {"status": "ok"}
+
+
 @pytest.mark.asyncio
 async def test_shorten_url(client):
     res = await client.post("/api/shorten", json={"url": "https://example.com"})