mocking

Author: liustve

aws-opentelemetry-distro/src/amazon/opentelemetry/distro/aws_opentelemetry_configurator.py

@@ -179,8 +179,8 @@ def _init_logging(
 
     # Provides a default OTLP log exporter when none is specified.
     # This is the behavior for the logs exporters for other languages.
-    # if not exporters:
-    #     exporters = {"otlp": OTLPLogExporter}
+    if not exporters:
+        exporters = {"otlp": OTLPLogExporter}
 
     provider = LoggerProvider(resource=resource)
     set_logger_provider(provider)

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

@@ -492,8 +492,19 @@ def test_customize_logs_exporter_sigv4(self):
             _customize_logs_exporter(OTLPGrpcLogExporter(), Resource.get_empty()), OTLPGrpcLogExporter
         )
 
-    def test_init_logging(self):
+    @patch('amazon.opentelemetry.distro.aws_opentelemetry_configurator.LoggingHandler')
+    @patch('amazon.opentelemetry.distro.aws_opentelemetry_configurator.getLogger')
+    @patch(
+            "amazon.opentelemetry.distro.aws_opentelemetry_configurator._customize_logs_exporter"
+        )
+    def test_init_logging(self, mock_get_logger, mock_logging_handler):
         captured_exporter = None
+        
+        mock_logger_instance = MagicMock()
+        mock_get_logger.return_value = mock_logger_instance
+        
+        mock_handler_instance = MagicMock()
+        mock_logging_handler.return_value = mock_handler_instance
 
         def capture_exporter(*args, **kwargs):
             nonlocal captured_exporter
@@ -502,9 +513,9 @@ def capture_exporter(*args, **kwargs):
             return result
 
         test_cases = [
-            # [{"otlp": OTLPLogExporter}, OTLPLogExporter],
-            [{}, None],
-            # [{"grpc": OTLPGrpcLogExporter}, OTLPGrpcLogExporter],
+            [{"otlp": OTLPLogExporter}, OTLPLogExporter],
+            [{}, OTLPLogExporter],
+            [{"grpc": OTLPGrpcLogExporter}, OTLPGrpcLogExporter],
         ]
 
         os.environ[OTEL_EXPORTER_OTLP_LOGS_ENDPOINT] = "https://logs.us-east-1.amazonaws.com/v1/logs"
@@ -520,7 +531,7 @@ def capture_exporter(*args, **kwargs):
                 expected_exporter = tc[1]
                 _init_logging(exporter_dict, Resource.get_empty())
 
-                # self.assertIsInstance(captured_exporter, expected_exporter)
+                self.assertIsInstance(captured_exporter, expected_exporter)
 
         os.environ.pop(OTEL_EXPORTER_OTLP_LOGS_ENDPOINT)