Changed test host/ip settings for registry to avoid accidental ipv6 u…

…sage.
tomerfiliba-org · Jan 11, 2024 · 1b071da · 1b071da
1 parent bd0e4d0
commit 1b071da
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/test_registry.py b/tests/test_registry.py
@@ -79,15 +79,16 @@ def test_listing(self):
 
 class TestTcpRegistry(BaseRegistryTest, unittest.TestCase):
     def _get_server(self):
-        return TCPRegistryServer(pruning_timeout=PRUNING_TIMEOUT, allow_listing=True)
+        return TCPRegistryServer(host="127.0.0.1", pruning_timeout=PRUNING_TIMEOUT, allow_listing=True)
 
     def _get_client(self):
-        return TCPRegistryClient("localhost")
+        return TCPRegistryClient(ip="127.0.0.1")
 
 
 class TestUdpRegistry(BaseRegistryTest, unittest.TestCase):
+    """ May fail due to iptables/packet-drops. """
     def _get_server(self):
-        return UDPRegistryServer(pruning_timeout=PRUNING_TIMEOUT, allow_listing=True)
+        return UDPRegistryServer(host="0.0.0.0", pruning_timeout=PRUNING_TIMEOUT, allow_listing=True)
 
     def _get_client(self):
         return UDPRegistryClient()