diff options
Diffstat (limited to 'test/test_conn_legacy.py')
| -rw-r--r-- | test/test_conn_legacy.py | 16 |
1 files changed, 9 insertions, 7 deletions
diff --git a/test/test_conn_legacy.py b/test/test_conn_legacy.py index 347588e..820c4e7 100644 --- a/test/test_conn_legacy.py +++ b/test/test_conn_legacy.py @@ -48,12 +48,12 @@ class ConnTest(unittest.TestCase): self.MockCreateConn.reset_mock() def test_collect_hosts__happy_path(self): - hosts = "localhost:1234,localhost" + hosts = "127.0.0.1:1234,127.0.0.1" results = collect_hosts(hosts) self.assertEqual(set(results), set([ - ('localhost', 1234, socket.AF_INET), - ('localhost', 9092, socket.AF_INET), + ('127.0.0.1', 1234, socket.AF_INET), + ('127.0.0.1', 9092, socket.AF_INET), ])) def test_collect_hosts__ipv6(self): @@ -72,16 +72,18 @@ class ConnTest(unittest.TestCase): 'localhost', '[localhost]', '2001::1', + '[2001::1]', '[2001::1]:1234', ] results = collect_hosts(hosts) self.assertEqual(set(results), set([ - ('localhost', 1234, socket.AF_INET), - ('localhost', 9092, socket.AF_INET), + ('localhost', 1234, socket.AF_UNSPEC), + ('localhost', 9092, socket.AF_UNSPEC), ('localhost', 9092, socket.AF_INET6), ('2001::1', 9092, socket.AF_INET6), + ('2001::1', 9092, socket.AF_INET6), ('2001::1', 1234, socket.AF_INET6), ])) @@ -90,8 +92,8 @@ class ConnTest(unittest.TestCase): results = collect_hosts(hosts) self.assertEqual(set(results), set([ - ('localhost', 1234, socket.AF_INET), - ('localhost', 9092, socket.AF_INET), + ('localhost', 1234, socket.AF_UNSPEC), + ('localhost', 9092, socket.AF_UNSPEC), ])) |