chore: add test to reproduce clear state bug with custom keys

Author: eliasbrange

tests/functional/test_logger.py

@@ -683,6 +683,29 @@ def handler(event, context):
         assert key in second_log
 
 
+def test_clear_state_keeps_custom_keys(lambda_context, stdout, service_name):
+    # GIVEN
+    date_format = "%Y"
+    location_format = "%(module)s:%(funcName)s:%(lineno)d"
+    logger = Logger(service=service_name, stream=stdout, location=location_format, datefmt=date_format)
+
+    # WHEN clear_state is set
+    @logger.inject_lambda_context(clear_state=True)
+    def handler(event, context):
+        logger.info("Foo")
+
+    # THEN all standard keys should be available as usual
+    handler({}, lambda_context)
+    handler({}, lambda_context)
+
+    first_log, second_log = capture_multiple_logging_statements_output(stdout)
+
+    assert re.fullmatch("[0-9]{4}", first_log["timestamp"])
+    assert re.fullmatch("[0-9]{4}", second_log["timestamp"])
+    assert re.fullmatch(".+:.+:[0-9]+", first_log["location"])
+    assert re.fullmatch(".+:.+:[0-9]+", second_log["location"])
+
+
 def test_clear_state_keeps_exception_keys(lambda_context, stdout, service_name):
     # GIVEN
     logger = Logger(service=service_name, stream=stdout)