11"""Unit tests for logger module."""
22
3+ import logging
34from collections .abc import Mapping
45from unittest .mock import Mock
56
@@ -155,7 +156,7 @@ def test_logger_from_log_info_full():
155156 expected_extra = {
156157 "execution_arn" : "arn:aws:test" ,
157158 "parent_id" : "parent123" ,
158- "name " : "test_name" ,
159+ "operation_name " : "test_name" ,
159160 "attempt" : 5 ,
160161 }
161162 assert logger ._default_extra == expected_extra # noqa: SLF001
@@ -175,7 +176,7 @@ def test_logger_from_log_info_partial_fields():
175176 # Test with name but no parent_id or attempt
176177 log_info = LogInfo ("arn:aws:test" , None , "test_name" )
177178 logger = Logger .from_log_info (mock_logger , log_info )
178- expected_extra = {"execution_arn" : "arn:aws:test" , "name " : "test_name" }
179+ expected_extra = {"execution_arn" : "arn:aws:test" , "operation_name " : "test_name" }
179180 assert logger ._default_extra == expected_extra # noqa: SLF001
180181
181182 # Test with attempt but no parent_id or name
@@ -207,7 +208,7 @@ def test_logger_with_log_info():
207208 expected_extra = {
208209 "execution_arn" : "arn:aws:new" ,
209210 "parent_id" : "parent2" ,
210- "name " : "new_name" ,
211+ "operation_name " : "new_name" ,
211212 }
212213 assert new_logger ._default_extra == expected_extra # noqa: SLF001
213214 assert new_logger ._logger is mock_logger # noqa: SLF001
@@ -325,3 +326,13 @@ def test_logger_extra_override():
325326 "new_field" : "value" ,
326327 }
327328 mock_logger .info .assert_called_once_with ("test" , extra = expected_extra )
329+
330+
331+ def test_logger_without_mocked_logger ():
332+ """Test Logger methods without mocking the underlying logger."""
333+ log_info = LogInfo ("arn:aws:test" , "parent123" , "test_name" , 5 )
334+ logger = Logger .from_log_info (logging .getLogger (), log_info )
335+
336+ logger .info ("test" , extra = {"execution_arn" : "overridden" , "new_field" : "value" })
337+ logger .warning ("test" , extra = {"execution_arn" : "overridden" , "new_field" : "value" })
338+ logger .error ("test" , extra = {"execution_arn" : "overridden" , "new_field" : "value" })
0 commit comments