55 lines
1.8 KiB
Python
55 lines
1.8 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
import logging
|
|
import re
|
|
from unittest.mock import AsyncMock
|
|
|
|
from fastapi.testclient import TestClient
|
|
|
|
from app.logging_config import JsonLogFormatter, parse_redact_fields
|
|
from app.main import app
|
|
from app.http.middleware import REQUEST_ID_HEADER, parse_skip_paths
|
|
|
|
|
|
def test_json_log_formatter_redacts_sensitive_fields() -> None:
|
|
formatter = JsonLogFormatter(redact_fields=parse_redact_fields("api_key"))
|
|
record = logging.makeLogRecord(
|
|
{
|
|
"name": "tests.logging",
|
|
"levelno": logging.INFO,
|
|
"levelname": "INFO",
|
|
"msg": "test.event",
|
|
"args": (),
|
|
"password": "very-secret",
|
|
"payload": {
|
|
"csrf_token": "token-value",
|
|
"safe": "ok",
|
|
},
|
|
"color_message": "ANSI-noise",
|
|
}
|
|
)
|
|
|
|
payload = json.loads(formatter.format(record))
|
|
|
|
assert payload["event"] == "test.event"
|
|
assert re.fullmatch(r"\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}Z", payload["timestamp"])
|
|
assert payload["password"] == "[REDACTED]"
|
|
assert payload["payload"]["csrf_token"] == "[REDACTED]"
|
|
assert payload["payload"]["safe"] == "ok"
|
|
assert "color_message" not in payload
|
|
|
|
|
|
def test_request_logging_middleware_sets_request_id_header(monkeypatch) -> None:
|
|
monkeypatch.setattr("app.main.check_database", AsyncMock(return_value=True))
|
|
client = TestClient(app)
|
|
response = client.get("/healthz", headers={REQUEST_ID_HEADER: "request-123"})
|
|
assert response.status_code == 200
|
|
assert response.headers[REQUEST_ID_HEADER] == "request-123"
|
|
|
|
|
|
def test_parse_skip_paths_trims_and_discards_empty_segments() -> None:
|
|
assert parse_skip_paths(" /healthz , , /api/v1/metrics ") == (
|
|
"/healthz",
|
|
"/api/v1/metrics",
|
|
)
|