mocked log provider

Author: liustve

aws-opentelemetry-distro/tests/amazon/opentelemetry/distro/test_aws_opentelementry_configurator.py

@@ -1,6 +1,5 @@
 # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 # SPDX-License-Identifier: Apache-2.0
-import logging
 import os
 import time
 from unittest import TestCase
@@ -88,18 +87,6 @@ def setUpClass(cls):
     def tearDown(self):
         os.environ.pop("OTEL_AWS_APPLICATION_SIGNALS_ENABLED", None)
         os.environ.pop("OTEL_AWS_APPLICATION_SIGNALS_RUNTIME_ENABLED", None)
-        root_logger = logging.getLogger()
-        for handler in root_logger.handlers[:]:
-            handler.close()
-            root_logger.removeHandler(handler)
-
-        # Shut down any active exporters
-        try:
-            from opentelemetry.sdk._logs import _internal
-
-            _internal.get_log_provider().shutdown()
-        except (ImportError, AttributeError):
-            pass
 
     # The probability of this passing once without correct IDs is low, 20 times is inconceivable.
     def test_provide_generate_xray_ids(self):
@@ -524,9 +511,18 @@ def capture_exporter(*args, **kwargs):
 
         with patch(
             "amazon.opentelemetry.distro.aws_opentelemetry_configurator._customize_logs_exporter"
-        ) as mock_customize_logs_exporter:
+        ) as mock_customize_logs_exporter, patch(
+            "opentelemetry.sdk._logs.LoggerProvider"
+        ) as mock_logger_provider, patch(
+            "opentelemetry.sdk._logs.export.BatchLogRecordProcessor"
+        ) as mock_batch_processor:
 
+            # Set up the mocks
             mock_customize_logs_exporter.side_effect = capture_exporter
+            mock_provider_instance = MagicMock()
+            mock_logger_provider.return_value = mock_provider_instance
+            mock_batch_processor_instance = MagicMock()
+            mock_batch_processor.return_value = mock_batch_processor_instance
 
             for tc in test_cases:
                 exporter_dict = tc[0]