66from unittest import TestCase
77from unittest .mock import MagicMock , patch
88
9- from amazon .opentelemetry .exporters .otlp .udp import (
9+ from amazon .distro . opentelemetry .exporters .otlp .udp import (
1010 DEFAULT_ENDPOINT ,
1111 FORMAT_OTEL_SAMPLED_TRACES_BINARY_PREFIX ,
1212 FORMAT_OTEL_UNSAMPLED_TRACES_BINARY_PREFIX ,
1919
2020class TestUdpExporter (TestCase ):
2121
22- @patch ("amazon.opentelemetry.exporters.otlp.udp.exporter.socket.socket" )
22+ @patch ("amazon.distro. opentelemetry.exporters.otlp.udp.exporter.socket.socket" )
2323 def test_udp_exporter_init_default (self , mock_socket ):
2424 exporter = UdpExporter ()
2525 self .assertEqual (exporter ._endpoint , DEFAULT_ENDPOINT )
@@ -28,7 +28,7 @@ def test_udp_exporter_init_default(self, mock_socket):
2828 mock_socket .assert_called_once_with (socket .AF_INET , socket .SOCK_DGRAM )
2929 mock_socket ().setblocking .assert_called_once_with (False )
3030
31- @patch ("amazon.opentelemetry.exporters.otlp.udp.exporter.socket.socket" )
31+ @patch ("amazon.distro. opentelemetry.exporters.otlp.udp.exporter.socket.socket" )
3232 def test_udp_exporter_init_with_endpoint (self , mock_socket ):
3333 exporter = UdpExporter (endpoint = "localhost:5000" )
3434 self .assertNotEqual (exporter ._endpoint , DEFAULT_ENDPOINT )
@@ -37,13 +37,13 @@ def test_udp_exporter_init_with_endpoint(self, mock_socket):
3737 mock_socket .assert_called_once_with (socket .AF_INET , socket .SOCK_DGRAM )
3838 mock_socket ().setblocking .assert_called_once_with (False )
3939
40- @patch ("amazon.opentelemetry.exporters.otlp.udp.exporter.socket.socket" )
40+ @patch ("amazon.distro. opentelemetry.exporters.otlp.udp.exporter.socket.socket" )
4141 def test_udp_exporter_init_invalid_endpoint (self , mock_socket ):
4242 with self .assertRaises (ValueError ):
4343 UdpExporter (endpoint = "invalidEndpoint:port" )
4444
4545 # pylint: disable=no-self-use
46- @patch ("amazon.opentelemetry.exporters.otlp.udp.exporter.socket.socket" )
46+ @patch ("amazon.distro. opentelemetry.exporters.otlp.udp.exporter.socket.socket" )
4747 def test_send_data (self , mock_socket ):
4848 mock_socket_instance = mock_socket .return_value
4949 exporter = UdpExporter ()
@@ -53,7 +53,7 @@ def test_send_data(self, mock_socket):
5353 expected_message = PROTOCOL_HEADER + "signal_prefix" + encoded_bytes .decode ("utf-8" )
5454 mock_socket_instance .sendto .assert_called_once_with (expected_message .encode ("utf-8" ), ("127.0.0.1" , 2000 ))
5555
56- @patch ("amazon.opentelemetry.exporters.otlp.udp.exporter.socket.socket" )
56+ @patch ("amazon.distro. opentelemetry.exporters.otlp.udp.exporter.socket.socket" )
5757 def test_shutdown (self , mock_socket ):
5858 mock_socket_instance = mock_socket .return_value
5959 exporter = UdpExporter ()
@@ -63,8 +63,8 @@ def test_shutdown(self, mock_socket):
6363
6464class TestOTLPUdpSpanExporter (unittest .TestCase ):
6565
66- @patch ("amazon.opentelemetry.exporters.otlp.udp.exporter.encode_spans" )
67- @patch ("amazon.opentelemetry.exporters.otlp.udp.exporter.UdpExporter" )
66+ @patch ("amazon.distro. opentelemetry.exporters.otlp.udp.exporter.encode_spans" )
67+ @patch ("amazon.distro. opentelemetry.exporters.otlp.udp.exporter.UdpExporter" )
6868 def test_export_unsampled_span (self , mock_udp_exporter , mock_encode_spans ):
6969 mock_udp_exporter_instance = mock_udp_exporter .return_value
7070 mock_encoded_data = MagicMock ()
@@ -76,8 +76,8 @@ def test_export_unsampled_span(self, mock_udp_exporter, mock_encode_spans):
7676 )
7777 self .assertEqual (result , SpanExportResult .SUCCESS )
7878
79- @patch ("amazon.opentelemetry.exporters.otlp.udp.exporter.encode_spans" )
80- @patch ("amazon.opentelemetry.exporters.otlp.udp.exporter.UdpExporter" )
79+ @patch ("amazon.distro. opentelemetry.exporters.otlp.udp.exporter.encode_spans" )
80+ @patch ("amazon.distro. opentelemetry.exporters.otlp.udp.exporter.UdpExporter" )
8181 def test_export_sampled_span (self , mock_udp_exporter , mock_encode_spans ):
8282 mock_udp_exporter_instance = mock_udp_exporter .return_value
8383 mock_encoded_data = MagicMock ()
@@ -89,8 +89,8 @@ def test_export_sampled_span(self, mock_udp_exporter, mock_encode_spans):
8989 )
9090 self .assertEqual (result , SpanExportResult .SUCCESS )
9191
92- @patch ("amazon.opentelemetry.exporters.otlp.udp.exporter.encode_spans" )
93- @patch ("amazon.opentelemetry.exporters.otlp.udp.exporter.UdpExporter" )
92+ @patch ("amazon.distro. opentelemetry.exporters.otlp.udp.exporter.encode_spans" )
93+ @patch ("amazon.distro. opentelemetry.exporters.otlp.udp.exporter.UdpExporter" )
9494 def test_export_with_exception (self , mock_udp_exporter , mock_encode_spans ):
9595 mock_udp_exporter_instance = mock_udp_exporter .return_value
9696 mock_encoded_data = MagicMock ()
@@ -101,7 +101,7 @@ def test_export_with_exception(self, mock_udp_exporter, mock_encode_spans):
101101 self .assertEqual (result , SpanExportResult .FAILURE )
102102
103103 # pylint: disable=no-self-use
104- @patch ("amazon.opentelemetry.exporters.otlp.udp.exporter.UdpExporter" )
104+ @patch ("amazon.distro. opentelemetry.exporters.otlp.udp.exporter.UdpExporter" )
105105 def test_shutdown (self , mock_udp_exporter ):
106106 mock_udp_exporter_instance = mock_udp_exporter .return_value
107107 exporter = OTLPUdpSpanExporter ()
0 commit comments