diff --git a/src/daq_queuing_service/api/api.py b/src/daq_queuing_service/api/api.py index 3fde18a..50c0b83 100644 --- a/src/daq_queuing_service/api/api.py +++ b/src/daq_queuing_service/api/api.py @@ -7,7 +7,7 @@ UnknownPlanError, ) from blueapi.service.model import TaskRequest -from fastapi import APIRouter, Request +from fastapi import APIRouter, Request, Response from pydantic import BaseModel from daq_queuing_service.task import ExperimentDefinition, Status, Task @@ -71,6 +71,10 @@ def create_api_router( ) -> APIRouter: router = APIRouter() + @router.get("/healthz") + async def healthz(): + return Response() + @router.get("/") def read_root(request: Request): base_url = str(request.base_url) diff --git a/tests/unit_tests/test_api.py b/tests/unit_tests/test_api.py index f97dc96..63493b5 100644 --- a/tests/unit_tests/test_api.py +++ b/tests/unit_tests/test_api.py @@ -69,6 +69,11 @@ def test_read_root_returns_expected_string(test_client: TestClient): ) +def test_healthz_returns_healthy(test_client: TestClient): + response = test_client.get("/healthz") + assert response.status_code == 200 + + def test_get_queue_state_returns_queue_state(test_client: TestClient): response = test_client.get("/queue/state") assert response.status_code == 200