Fixed integration tests

exasol · Nov 22, 2024 · 8df7675 · 8df7675
1 parent 4914f33
commit 8df7675
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/exasol/analytics/udf/communication/discovery/localhost/communicator.py b/exasol/analytics/udf/communication/discovery/localhost/communicator.py
@@ -1,4 +1,5 @@
-from exasol.analytics.udf.communication.discovery import localhost
+from exasol.analytics.udf.communication.discovery.localhost.discovery_socket import DiscoverySocketFactory
+from exasol.analytics.udf.communication.discovery.localhost.discovery_strategy import DiscoveryStrategy 
 from exasol.analytics.udf.communication.ip_address import IPAddress, Port
 from exasol.analytics.udf.communication.peer_communicator import PeerCommunicator
 from exasol.analytics.udf.communication.peer_communicator.forward_register_peer_config import (
@@ -20,7 +21,7 @@ def create(
         listen_ip: IPAddress,
         discovery_port: Port,
         socket_factory: SocketFactory,
-        discovery_socket_factory: localhost.DiscoverySocketFactory,
+        discovery_socket_factory: DiscoverySocketFactory,
     ) -> PeerCommunicator:
         peer_communicator = PeerCommunicator(
             name=name,
@@ -35,7 +36,7 @@ def create(
             ),
             socket_factory=socket_factory,
         )
-        discovery = localhost.DiscoveryStrategy(
+        discovery = DiscoveryStrategy(
             port=discovery_port,
             timeout_in_seconds=120,
             time_between_ping_messages_in_seconds=1,

diff --git a/exasol/analytics/udf/communication/discovery/localhost/discovery_strategy.py b/exasol/analytics/udf/communication/discovery/localhost/discovery_strategy.py
@@ -3,7 +3,7 @@
 from typing import Optional, cast
 
 from exasol.analytics.udf.communication import messages
-from exasol.analytics.udf.communication.discovery.localhost import (
+from exasol.analytics.udf.communication.discovery.localhost.discovery_socket import (
     DiscoverySocket,
     DiscoverySocketFactory,
 )