multi threading fix

Author: liustve

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

@@ -492,19 +492,23 @@ def test_customize_logs_exporter_sigv4(self):
             _customize_logs_exporter(OTLPGrpcLogExporter(), Resource.get_empty()), OTLPGrpcLogExporter
         )
 
-    @patch("amazon.opentelemetry.distro.aws_opentelemetry_configurator.LoggingHandler")
-    @patch("amazon.opentelemetry.distro.aws_opentelemetry_configurator.getLogger")
+    # Need to patch all of these to prevent some weird multi-threading error with the LogProvider
+    @patch("amazon.opentelemetry.distro.aws_opentelemetry_configurator.LoggingHandler", return_value=MagicMock())
+    @patch("amazon.opentelemetry.distro.aws_opentelemetry_configurator.getLogger", return_value=MagicMock())
     @patch("amazon.opentelemetry.distro.aws_opentelemetry_configurator._customize_logs_exporter")
-    @patch("amazon.opentelemetry.distro.aws_opentelemetry_configurator.LoggerProvider")
-    @patch("amazon.opentelemetry.distro.aws_opentelemetry_configurator.BatchLogRecordProcessor")
-    def test_init_logging(self, mock_batch_processor, mock_logger_provider, 
-                        mock_customize_logs_exporter, mock_get_logger, mock_logging_handler):
-        # Create mock objects
-        mock_logger_provider.return_value = MagicMock()
-        mock_get_logger.return_value = MagicMock()
-        mock_logging_handler.return_value = MagicMock()
-        mock_batch_processor.return_value = MagicMock()
-        
+    @patch("amazon.opentelemetry.distro.aws_opentelemetry_configurator.LoggerProvider", return_value=MagicMock())
+    @patch(
+        "amazon.opentelemetry.distro.aws_opentelemetry_configurator.BatchLogRecordProcessor", return_value=MagicMock()
+    )
+    def test_init_logging(
+        self,
+        mock_batch_processor,
+        mock_logger_provider,
+        mock_customize_logs_exporter,
+        mock_get_logger,
+        mock_logging_handler,
+    ):
+
         captured_exporter = None
 
         def capture_exporter(*args, **kwargs):
@@ -513,8 +517,6 @@ def capture_exporter(*args, **kwargs):
             captured_exporter = result
             return result
 
-        mock_logging_handler.return_value = MagicMock()
-        mock_customize_logs_exporter = MagicMock()
         mock_customize_logs_exporter.side_effect = capture_exporter
 
         test_cases = [