context filter for adding additional data to logs

tests/utils/test_logging.py

@@ -1,10 +1,13 @@
 from io import StringIO
 import json
 import logging
+from uuid import uuid4
 
 import pytest
 
-from atst.utils.logging import JsonFormatter
+from atst.utils.logging import JsonFormatter, RequestContextFilter
+
+from tests.factories import UserFactory
 
 
 @pytest.fixture
@@ -30,6 +33,8 @@ def logger(log_stream):
     logHandler = logging.StreamHandler(log_stream)
     formatter = JsonFormatter()
     logHandler.setFormatter(formatter)
+    logger.setLevel(logging.INFO)
+    logger.addFilter(RequestContextFilter())
     logger.addHandler(logHandler)
 
     return logger
@@ -55,3 +60,15 @@ def test_json_formatter_for_exceptions(logger, log_stream_content):
     log = json.loads(log_stream_content())
     assert log["severity"] == "ERROR"
     assert log.get("details")
+
+
+def test_request_context_filter(logger, log_stream_content, request_ctx):
+    user = UserFactory.create()
+    uuid = str(uuid4())
+
+    request_ctx.g.current_user = user
+    request_ctx.request.environ["HTTP_X_REQUEST_ID"] = uuid
+    logger.info("this user is doing something")
+    log = json.loads(log_stream_content())
+    assert log["user_id"] == str(user.id)
+    assert log["request_id"] == uuid