@@ -42,7 +42,8 @@ def test_handler_default_log_level(self):
42
42
logger .debug ("Debug message" )
43
43
self .assertEqual (emitter_mock .emit .call_count , 0 )
44
44
# Assert emit gets called for warning message
45
- logger .warning ("Warning message" )
45
+ with self .assertLogs (level = logging .WARNING ):
46
+ logger .warning ("Warning message" )
46
47
self .assertEqual (emitter_mock .emit .call_count , 1 )
47
48
48
49
def test_handler_custom_log_level (self ):
@@ -53,11 +54,14 @@ def test_handler_custom_log_level(self):
53
54
logger = get_logger (
54
55
level = logging .ERROR , logger_provider = emitter_provider_mock
55
56
)
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" )
57
59
# Make sure any log with level < ERROR is ignored
58
60
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" )
61
65
self .assertEqual (emitter_mock .emit .call_count , 2 )
62
66
63
67
def test_log_record_no_span_context (self ):
@@ -67,7 +71,8 @@ def test_log_record_no_span_context(self):
67
71
)
68
72
logger = get_logger (logger_provider = emitter_provider_mock )
69
73
# Assert emit gets called for warning message
70
- logger .warning ("Warning message" )
74
+ with self .assertLogs (level = logging .WARNING ):
75
+ logger .warning ("Warning message" )
71
76
args , _ = emitter_mock .emit .call_args_list [0 ]
72
77
log_record = args [0 ]
73
78
@@ -86,7 +91,8 @@ def test_log_record_user_attributes(self):
86
91
)
87
92
logger = get_logger (logger_provider = emitter_provider_mock )
88
93
# 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 })
90
96
args , _ = emitter_mock .emit .call_args_list [0 ]
91
97
log_record = args [0 ]
92
98
@@ -104,7 +110,8 @@ def test_log_record_exception(self):
104
110
try :
105
111
raise ZeroDivisionError ("division by zero" )
106
112
except ZeroDivisionError :
107
- logger .exception ("Zero Division Error" )
113
+ with self .assertLogs (level = logging .ERROR ):
114
+ logger .exception ("Zero Division Error" )
108
115
args , _ = emitter_mock .emit .call_args_list [0 ]
109
116
log_record = args [0 ]
110
117
@@ -137,7 +144,8 @@ def test_log_exc_info_false(self):
137
144
try :
138
145
raise ZeroDivisionError ("division by zero" )
139
146
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 )
141
149
args , _ = emitter_mock .emit .call_args_list [0 ]
142
150
log_record = args [0 ]
143
151
@@ -160,7 +168,8 @@ def test_log_record_trace_correlation(self):
160
168
161
169
tracer = trace .TracerProvider ().get_tracer (__name__ )
162
170
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" )
164
173
165
174
args , _ = emitter_mock .emit .call_args_list [0 ]
166
175
log_record = args [0 ]
0 commit comments