Fix lint and unit tests

Author: wangzlei

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

@@ -805,6 +805,7 @@ def _create_emf_exporter():
         if _is_lambda_environment() and not log_header_setting.is_valid():
             # pylint: disable=import-outside-toplevel
             from amazon.opentelemetry.distro.exporter.aws.metrics.console_emf_exporter import ConsoleEmfExporter
+
             return ConsoleEmfExporter(namespace=log_header_setting.namespace)
 
         # For non-Lambda environment or Lambda with valid headers - use CloudWatch EMF exporter

aws-opentelemetry-distro/tests/amazon/opentelemetry/distro/exporter/aws/metrics/test_aws_cloudwatch_emf_exporter.py

@@ -635,19 +635,19 @@ def test_shutdown(self, mock_force_flush):
         mock_force_flush.assert_called_once_with(5000)
 
     # pylint: disable=broad-exception-caught
-    def test_send_log_event_method_exists(self):
-        """Test that _send_log_event method exists and can be called."""
+    def test_export_method_exists(self):
+        """Test that _export method exists and can be called."""
         # Just test that the method exists and doesn't crash with basic input
         log_event = {"message": "test message", "timestamp": 1234567890}
 
         # Mock the log client to avoid actual AWS calls
         with patch.object(self.exporter.log_client, "send_log_event") as mock_send:
             # Should not raise an exception
             try:
-                self.exporter._send_log_event(log_event)
+                self.exporter._export(log_event)
                 mock_send.assert_called_once_with(log_event)
             except Exception as error:
-                self.fail(f"_send_log_event raised an exception: {error}")
+                self.fail(f"_export raised an exception: {error}")
 
 
 if __name__ == "__main__":

aws-opentelemetry-distro/tests/amazon/opentelemetry/distro/exporter/aws/metrics/test_console_emf_exporter.py

@@ -35,13 +35,10 @@ def test_send_log_event(self):
             '"Dimensions":[["Service"]],"Metrics":[{"Name":"TestMetric","Unit":"Count"}]}]},'
             '"Service":"test-service","TestMetric":42}'
         )
-        log_event = {
-            "message": test_message,
-            "timestamp": 1640995200000
-        }
+        log_event = {"message": test_message, "timestamp": 1640995200000}
 
         # Capture stdout to verify the output
-        with patch('sys.stdout', new_callable=StringIO) as mock_stdout:
+        with patch("sys.stdout", new_callable=StringIO) as mock_stdout:
             exporter._export(log_event)
 
         # Verify the message was printed to stdout

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

@@ -508,6 +508,7 @@ def test_customize_span_exporter_sigv4(self):
             bad_configs.append(config)
 
         for config in good_configs:
+            _clear_logs_header_cache()
             self.customize_exporter_test(
                 config,
                 _customize_span_exporter,
@@ -519,6 +520,7 @@ def test_customize_span_exporter_sigv4(self):
             )
 
         for config in bad_configs:
+            _clear_logs_header_cache()
             self.customize_exporter_test(
                 config,
                 _customize_span_exporter,
@@ -616,6 +618,7 @@ def test_customize_logs_exporter_sigv4(self):
             bad_configs.append(config)
 
         for config in good_configs:
+            _clear_logs_header_cache()
             self.customize_exporter_test(
                 config,
                 _customize_logs_exporter,
@@ -626,6 +629,7 @@ def test_customize_logs_exporter_sigv4(self):
             )
 
         for config in bad_configs:
+            _clear_logs_header_cache()
             self.customize_exporter_test(
                 config, _customize_logs_exporter, OTLPLogExporter(), OTLPLogExporter, Session, Compression.NoCompression
             )