@@ -42,7 +42,8 @@ def test_handler_default_log_level(self):
4242 logger .debug ("Debug message" )
4343 self .assertEqual (emitter_mock .emit .call_count , 0 )
4444 # Assert emit gets called for warning message
45- logger .warning ("Warning message" )
45+ with self .assertLogs (level = logging .WARNING ):
46+ logger .warning ("Warning message" )
4647 self .assertEqual (emitter_mock .emit .call_count , 1 )
4748
4849 def test_handler_custom_log_level (self ):
@@ -53,11 +54,14 @@ def test_handler_custom_log_level(self):
5354 logger = get_logger (
5455 level = logging .ERROR , logger_provider = emitter_provider_mock
5556 )
56- logger .warning ("Warning message test custom log level" )
57+ with self .assertLogs (level = logging .WARNING ):
58+ logger .warning ("Warning message test custom log level" )
5759 # Make sure any log with level < ERROR is ignored
5860 self .assertEqual (emitter_mock .emit .call_count , 0 )
59- logger .error ("Mumbai, we have a major problem" )
60- logger .critical ("No Time For Caution" )
61+ with self .assertLogs (level = logging .ERROR ):
62+ logger .error ("Mumbai, we have a major problem" )
63+ with self .assertLogs (level = logging .CRITICAL ):
64+ logger .critical ("No Time For Caution" )
6165 self .assertEqual (emitter_mock .emit .call_count , 2 )
6266
6367 def test_log_record_no_span_context (self ):
@@ -67,7 +71,8 @@ def test_log_record_no_span_context(self):
6771 )
6872 logger = get_logger (logger_provider = emitter_provider_mock )
6973 # Assert emit gets called for warning message
70- logger .warning ("Warning message" )
74+ with self .assertLogs (level = logging .WARNING ):
75+ logger .warning ("Warning message" )
7176 args , _ = emitter_mock .emit .call_args_list [0 ]
7277 log_record = args [0 ]
7378
@@ -86,7 +91,8 @@ def test_log_record_user_attributes(self):
8691 )
8792 logger = get_logger (logger_provider = emitter_provider_mock )
8893 # Assert emit gets called for warning message
89- logger .warning ("Warning message" , extra = {"http.status_code" : 200 })
94+ with self .assertLogs (level = logging .WARNING ):
95+ logger .warning ("Warning message" , extra = {"http.status_code" : 200 })
9096 args , _ = emitter_mock .emit .call_args_list [0 ]
9197 log_record = args [0 ]
9298
@@ -104,7 +110,8 @@ def test_log_record_exception(self):
104110 try :
105111 raise ZeroDivisionError ("division by zero" )
106112 except ZeroDivisionError :
107- logger .exception ("Zero Division Error" )
113+ with self .assertLogs (level = logging .ERROR ):
114+ logger .exception ("Zero Division Error" )
108115 args , _ = emitter_mock .emit .call_args_list [0 ]
109116 log_record = args [0 ]
110117
@@ -137,7 +144,8 @@ def test_log_exc_info_false(self):
137144 try :
138145 raise ZeroDivisionError ("division by zero" )
139146 except ZeroDivisionError :
140- logger .error ("Zero Division Error" , exc_info = False )
147+ with self .assertLogs (level = logging .ERROR ):
148+ logger .error ("Zero Division Error" , exc_info = False )
141149 args , _ = emitter_mock .emit .call_args_list [0 ]
142150 log_record = args [0 ]
143151
@@ -160,7 +168,8 @@ def test_log_record_trace_correlation(self):
160168
161169 tracer = trace .TracerProvider ().get_tracer (__name__ )
162170 with tracer .start_as_current_span ("test" ) as span :
163- logger .critical ("Critical message within span" )
171+ with self .assertLogs (level = logging .CRITICAL ):
172+ logger .critical ("Critical message within span" )
164173
165174 args , _ = emitter_mock .emit .call_args_list [0 ]
166175 log_record = args [0 ]
0 commit comments