threading error

Author: liustve

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

@@ -492,6 +492,38 @@ def test_customize_logs_exporter_sigv4(self):
 
         self.assertIsInstance(_customize_logs_exporter(OTLPGrpcLogExporter()), OTLPGrpcLogExporter)
 
+    def test_init_logging(self):
+        captured_exporter = None
+
+        def capture_exporter(*args, **kwargs):
+            nonlocal captured_exporter
+            result = _customize_logs_exporter(*args, **kwargs)
+            captured_exporter = result
+            return result
+
+        test_cases = [
+            [{"otlp": OTLPLogExporter}, OTLPLogExporter],
+            [{}, OTLPLogExporter],
+            [{"grpc": OTLPGrpcLogExporter}, OTLPGrpcLogExporter],
+        ]
+
+        os.environ[OTEL_EXPORTER_OTLP_LOGS_ENDPOINT] = "https://logs.us-east-1.amazonaws.com/v1/logs"
+
+        with patch(
+            "amazon.opentelemetry.distro.aws_opentelemetry_configurator._customize_logs_exporter"
+        ) as mock_customize_logs_exporter:
+
+            mock_customize_logs_exporter.side_effect = capture_exporter
+
+            for tc in test_cases:
+                exporter_dict = tc[0]
+                expected_exporter = tc[1]
+                _init_logging(exporter_dict, Resource.get_empty())
+
+                self.assertIsInstance(captured_exporter, expected_exporter)
+
+        os.environ.pop(OTEL_EXPORTER_OTLP_LOGS_ENDPOINT)
+
     def test_customize_span_processors(self):
         mock_tracer_provider: TracerProvider = MagicMock()
         _customize_span_processors(mock_tracer_provider, Resource.get_empty())