Update test

Author: tpoliaw

tests/unit_tests/service/test_main.py

@@ -1,5 +1,5 @@
 from unittest import mock
-from unittest.mock import Mock
+from unittest.mock import Mock, call
 
 import pytest
 from fastapi import FastAPI, Request
@@ -29,31 +29,36 @@ async def root():
     assert response.headers["X-BlueAPI-VERSION"] == __version__
 
 
-async def test_log_request_details():
+@pytest.mark.parametrize("path,level", [("/", "info"), ("/healthz", "debug")])
+async def test_log_request_details(path: str, level: str):
     with mock.patch("blueapi.service.main.LOGGER") as logger:
         app = FastAPI()
         app.middleware("http")(log_request_details)
 
-        @app.post("/")
+        @app.post(path)
         async def root():
             return {"message": "Hello World"}
 
         client = TestClient(app)
-        response = client.post("/", content="foo")
+        response = client.post(path, content="foo")
 
         assert response.status_code == 200
-        logger.debug.assert_called_once_with(
-            "testclient:50000 POST /",
-            extra={
-                "request_body": b"foo",
-            },
-        )
-
-        logger.info.assert_called_once_with(
-            "testclient:50000 POST / 200",
-            extra={
-                "request_body": b"foo",
-            },
+        log_level = getattr(logger, level)
+        log_level.assert_has_calls(
+            [
+                call(
+                    f"testclient:50000 POST {path}",
+                    extra={
+                        "request_body": b"foo",
+                    },
+                ),
+                call(
+                    f"testclient:50000 POST {path} 200",
+                    extra={
+                        "request_body": b"foo",
+                    },
+                ),
+            ]
         )