lint + added unit test case for better coverage

Author: srprash

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

@@ -40,7 +40,6 @@
 from opentelemetry.sdk.extension.aws.resource.ecs import AwsEcsResourceDetector
 from opentelemetry.sdk.extension.aws.resource.eks import AwsEksResourceDetector
 from opentelemetry.sdk.metrics import MeterProvider
-from opentelemetry.sdk.metrics._internal.export import ConsoleMetricExporter
 from opentelemetry.sdk.metrics._internal.instrument import (
     Counter,
     Histogram,

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

@@ -74,6 +74,17 @@ def test_export(self, mock_udp_exporter, mock_encode_metrics):
         mock_udp_exporter_instance.send_data.assert_called_once_with(data=mock_encoded_data, signal_format_prefix="M1")
         self.assertEqual(result, MetricExportResult.SUCCESS)
 
+    @patch("amazon.opentelemetry.distro.otlp_udp_exporter.encode_metrics")
+    @patch("amazon.opentelemetry.distro.otlp_udp_exporter.UdpExporter")
+    def test_export_with_exception(self, mock_udp_exporter, mock_encode_metrics):
+        mock_udp_exporter_instance = mock_udp_exporter.return_value
+        mock_encoded_data = MagicMock()
+        mock_encode_metrics.return_value.SerializeToString.return_value = mock_encoded_data
+        mock_udp_exporter_instance.send_data.side_effect = Exception("Something went wrong")
+        exporter = OTLPUdpMetricExporter()
+        result = exporter.export(MagicMock())
+        self.assertEqual(result, MetricExportResult.FAILURE)
+
     # pylint: disable=no-self-use
     @patch("amazon.opentelemetry.distro.otlp_udp_exporter.UdpExporter")
     def test_shutdown(self, mock_udp_exporter):
@@ -96,6 +107,17 @@ def test_export(self, mock_udp_exporter, mock_encode_spans):
         mock_udp_exporter_instance.send_data.assert_called_once_with(data=mock_encoded_data, signal_format_prefix="T1")
         self.assertEqual(result, SpanExportResult.SUCCESS)
 
+    @patch("amazon.opentelemetry.distro.otlp_udp_exporter.encode_spans")
+    @patch("amazon.opentelemetry.distro.otlp_udp_exporter.UdpExporter")
+    def test_export_with_exception(self, mock_udp_exporter, mock_encode_spans):
+        mock_udp_exporter_instance = mock_udp_exporter.return_value
+        mock_encoded_data = MagicMock()
+        mock_encode_spans.return_value.SerializeToString.return_value = mock_encoded_data
+        mock_udp_exporter_instance.send_data.side_effect = Exception("Something went wrong")
+        exporter = OTLPUdpSpanExporter()
+        result = exporter.export(MagicMock())
+        self.assertEqual(result, SpanExportResult.FAILURE)
+
     # pylint: disable=no-self-use
     @patch("amazon.opentelemetry.distro.otlp_udp_exporter.UdpExporter")
     def test_shutdown(self, mock_udp_exporter):