@@ -685,9 +685,8 @@ def handler(event, context):
685
685
686
686
def test_clear_state_keeps_custom_keys (lambda_context , stdout , service_name ):
687
687
# GIVEN
688
- date_format = "%Y"
689
- location_format = "%(module)s:%(funcName)s:%(lineno)d"
690
- logger = Logger (service = service_name , stream = stdout , location = location_format , datefmt = date_format )
688
+ location_format = "%(module)s.%(funcName)s:clear_state"
689
+ logger = Logger (service = service_name , stream = stdout , location = location_format , custom_key = "foo" )
691
690
692
691
# WHEN clear_state is set
693
692
@logger .inject_lambda_context (clear_state = True )
@@ -699,11 +698,9 @@ def handler(event, context):
699
698
handler ({}, lambda_context )
700
699
701
700
first_log , second_log = capture_multiple_logging_statements_output (stdout )
702
-
703
- assert re .fullmatch ("[0-9]{4}" , first_log ["timestamp" ])
704
- assert re .fullmatch ("[0-9]{4}" , second_log ["timestamp" ])
705
- assert re .fullmatch (".+:.+:[0-9]+" , first_log ["location" ])
706
- assert re .fullmatch (".+:.+:[0-9]+" , second_log ["location" ])
701
+ for log in (first_log , second_log ):
702
+ assert "foo" == log ["custom_key" ]
703
+ assert "test_logger.handler:clear_state" == log ["location" ]
707
704
708
705
709
706
def test_clear_state_keeps_exception_keys (lambda_context , stdout , service_name ):
0 commit comments