Tests for logger

Author: patrick91

src/fastapi_cli/utils/cli.py

@@ -12,7 +12,7 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
         self.toolkit = get_rich_toolkit()
 
     def formatMessage(self, record: logging.LogRecord) -> str:
-        return self.toolkit.print_as_string(record.message, tag=record.levelname)
+        return self.toolkit.print_as_string(record.getMessage(), tag=record.levelname)
 
 
 def get_uvicorn_log_config() -> Dict[str, Any]:
@@ -68,6 +68,7 @@ def get_rich_toolkit() -> RichToolkit:
             "result": "grey85",
             "progress": "on #007166",
             "error": "red",
+            "log.info": "black on blue",
         },
     )
 

tests/test_utils_cli.py

@@ -0,0 +1,35 @@
+import logging
+
+from fastapi_cli.utils.cli import CustomFormatter, get_uvicorn_log_config
+
+
+def test_get_uvicorn_config_uses_custom_formatter() -> None:
+    config = get_uvicorn_log_config()
+
+    assert config["formatters"]["default"]["()"] is CustomFormatter
+    assert config["formatters"]["access"]["()"] is CustomFormatter
+
+
+def test_custom_formatter() -> None:
+    formatter = CustomFormatter()
+
+    record = logging.LogRecord(
+        name="uvicorn.access",
+        level=logging.INFO,
+        pathname="",
+        lineno=0,
+        msg="%(client_addr)s - '%(request_line)s' %(status_code)s",
+        args={
+            "client_addr": "127.0.0.1",
+            "request_line": "GET / HTTP/1.1",
+            "status_code": 200,
+        },
+        exc_info=None,
+    )
+
+    formatted = formatter.formatMessage(record)
+
+    assert "INFO" in formatted
+    assert "127.0.0.1" in formatted
+    assert "GET / HTTP/1.1" in formatted
+    assert "200" in formatted