mocking

Author: liustve

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

@@ -497,7 +497,13 @@ def test_customize_logs_exporter_sigv4(self):
     @patch(
             "amazon.opentelemetry.distro.aws_opentelemetry_configurator._customize_logs_exporter"
         )
-    def test_init_logging(self, mock_get_logger, mock_logging_handler):
+    def test_init_logging(self, mock_customize_logs_exporter, mock_get_logger, mock_logging_handler):
+        def capture_exporter(*args, **kwargs):
+            nonlocal captured_exporter
+            result = _customize_logs_exporter(*args, **kwargs)
+            captured_exporter = result
+            return result
+        
         captured_exporter = None
 
         mock_logger_instance = MagicMock()
@@ -506,11 +512,9 @@ def test_init_logging(self, mock_get_logger, mock_logging_handler):
         mock_handler_instance = MagicMock()
         mock_logging_handler.return_value = mock_handler_instance
 
-        def capture_exporter(*args, **kwargs):
-            nonlocal captured_exporter
-            result = _customize_logs_exporter(*args, **kwargs)
-            captured_exporter = result
-            return result
+        mock_customize_logs_exporter = MagicMock()
+        mock_customize_logs_exporter.side_effect = capture_exporter
+
 
         test_cases = [
             [{"otlp": OTLPLogExporter}, OTLPLogExporter],