Fix test case

open-telemetry · Feb 14, 2024 · 41449c2 · 41449c2
1 parent 1ddd505
commit 41449c2
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/opentelemetry-sdk/tests/logs/test_handler.py b/opentelemetry-sdk/tests/logs/test_handler.py
@@ -80,11 +80,18 @@ def test_log_record_emit_noop(self):
         handler_mock._translate.assert_not_called()
 
     def test_log_flush_noop(self):
-        emitter_provider_mock = Mock(spec=NoOpLoggerProvider)
-        logger = get_logger(logger_provider=emitter_provider_mock)
+
+        no_op_logger_provider = NoOpLoggerProvider()
+        no_op_logger_provider.force_flush = Mock()
+
+        logger = get_logger(logger_provider=NoOpLoggerProvider())
+
         with self.assertLogs(level=logging.WARNING):
             logger.warning("Warning message")
 
+        logger.handlers[0].flush()
+        no_op_logger_provider.force_flush.assert_not_called()
+
     def test_log_record_no_span_context(self):
         emitter_provider_mock = Mock(spec=LoggerProvider)
         emitter_mock = APIGetLogger(