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