1717import unittest
1818from unittest .mock import Mock , patch
1919
20- from opentelemetry ._logs import LogRecord as APILogRecord
21- from opentelemetry ._logs import SeverityNumber
20+ from opentelemetry ._logs import LogRecord , SeverityNumber
2221from opentelemetry .context import get_current
2322from opentelemetry .sdk ._logs import (
2423 Logger ,
@@ -178,18 +177,18 @@ def test_can_emit_api_logrecord(self):
178177 self .assertTrue (isinstance (log_record , LogRecord ))
179178 self .assertEqual (log_record .timestamp , None )
180179 self .assertEqual (log_record .observed_timestamp , 0 )
181- self .assertEqual (log_record .context , {} )
180+ self .assertIsNotNone (log_record .context )
182181 self .assertEqual (log_record .severity_number , None )
183182 self .assertEqual (log_record .severity_text , None )
184183 self .assertEqual (log_record .body , "a log line" )
185184 self .assertEqual (log_record .attributes , {})
186185 self .assertEqual (log_record .event_name , None )
187- self .assertEqual (log_record .resource , logger .resource )
186+ self .assertEqual (log_data .resource , logger .resource )
188187
189188 def test_can_emit_with_keywords_arguments (self ):
190189 logger , log_record_processor_mock = self ._get_logger ()
191190
192- logger . emit (
191+ log_record = LogRecord (
193192 timestamp = 100 ,
194193 observed_timestamp = 101 ,
195194 context = get_current (),
@@ -199,16 +198,19 @@ def test_can_emit_with_keywords_arguments(self):
199198 attributes = {"some" : "attributes" },
200199 event_name = "event_name" ,
201200 )
201+ logger .emit (log_record )
202202 log_record_processor_mock .on_emit .assert_called_once ()
203203 log_data = log_record_processor_mock .on_emit .call_args .args [0 ]
204- log_record = log_data .log_record
205- self .assertTrue (isinstance (log_record , LogRecord ))
206- self .assertEqual (log_record .timestamp , 100 )
207- self .assertEqual (log_record .observed_timestamp , 101 )
208- self .assertEqual (log_record .context , {})
209- self .assertEqual (log_record .severity_number , SeverityNumber .WARN )
210- self .assertEqual (log_record .severity_text , "warn" )
211- self .assertEqual (log_record .body , "a body" )
212- self .assertEqual (log_record .attributes , {"some" : "attributes" })
213- self .assertEqual (log_record .event_name , "event_name" )
214- self .assertEqual (log_record .resource , logger .resource )
204+ result_log_record = log_data .log_record
205+ self .assertTrue (isinstance (result_log_record , LogRecord ))
206+ self .assertEqual (result_log_record .timestamp , 100 )
207+ self .assertEqual (result_log_record .observed_timestamp , 101 )
208+ self .assertIsNotNone (result_log_record .context )
209+ self .assertEqual (
210+ result_log_record .severity_number , SeverityNumber .WARN
211+ )
212+ self .assertEqual (result_log_record .severity_text , "warn" )
213+ self .assertEqual (result_log_record .body , "a body" )
214+ self .assertEqual (result_log_record .attributes , {"some" : "attributes" })
215+ self .assertEqual (result_log_record .event_name , "event_name" )
216+ self .assertEqual (log_data .resource , logger .resource )
0 commit comments