diff options
| author | Tim Evens <tievens@cisco.com> | 2016-03-30 15:32:05 -0700 |
|---|---|---|
| committer | Tim Evens <tievens@cisco.com> | 2016-03-30 15:32:25 -0700 |
| commit | 0f78d57c604e864fab51f7cfb8fa69c9c4e623c7 (patch) | |
| tree | 649c0953a56f39719761751901da5da11b9589f0 /test/test_client.py | |
| parent | c6c862ad29ec5d0ae61d635c2020fb925b405c44 (diff) | |
| download | kafka-python-0f78d57c604e864fab51f7cfb8fa69c9c4e623c7.tar.gz | |
Kafka IPv6 Support.
IPv6 address without port can be defined as the IPv6 address. If the address
is a hostname or if a port is included, then the address MUST be wrapped
in brackets [] (E.g. [somehost]:1234 or [fd00:1001::2]:1234).
Diffstat (limited to 'test/test_client.py')
| -rw-r--r-- | test/test_client.py | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/test/test_client.py b/test/test_client.py index a53fce1..6980434 100644 --- a/test/test_client.py +++ b/test/test_client.py @@ -41,7 +41,7 @@ class TestSimpleClient(unittest.TestCase): client = SimpleClient(hosts=['kafka01:9092', 'kafka02:9092', 'kafka03:9092']) self.assertEqual( - sorted([('kafka01', 9092), ('kafka02', 9092), ('kafka03', 9092)]), + sorted([('kafka01', 9092, socket.AF_INET), ('kafka02', 9092, socket.AF_INET), ('kafka03', 9092, socket.AF_INET)]), sorted(client.hosts)) def test_init_with_csv(self): @@ -49,7 +49,7 @@ class TestSimpleClient(unittest.TestCase): client = SimpleClient(hosts='kafka01:9092,kafka02:9092,kafka03:9092') self.assertEqual( - sorted([('kafka01', 9092), ('kafka02', 9092), ('kafka03', 9092)]), + sorted([('kafka01', 9092, socket.AF_INET), ('kafka02', 9092, socket.AF_INET), ('kafka03', 9092, socket.AF_INET)]), sorted(client.hosts)) def test_init_with_unicode_csv(self): @@ -57,7 +57,7 @@ class TestSimpleClient(unittest.TestCase): client = SimpleClient(hosts=u'kafka01:9092,kafka02:9092,kafka03:9092') self.assertEqual( - sorted([('kafka01', 9092), ('kafka02', 9092), ('kafka03', 9092)]), + sorted([('kafka01', 9092, socket.AF_INET), ('kafka02', 9092, socket.AF_INET), ('kafka03', 9092, socket.AF_INET)]), sorted(client.hosts)) @patch.object(SimpleClient, '_get_conn') @@ -70,7 +70,7 @@ class TestSimpleClient(unittest.TestCase): for val in mocked_conns.values(): mock_conn(val, success=False) - def mock_get_conn(host, port): + def mock_get_conn(host, port, afi): return mocked_conns[(host, port)] conn.side_effect = mock_get_conn @@ -98,7 +98,7 @@ class TestSimpleClient(unittest.TestCase): mocked_conns[('kafka02', 9092)].send.return_value = future mocked_conns[('kafka02', 9092)].recv.side_effect = lambda: future.success('valid response') - def mock_get_conn(host, port): + def mock_get_conn(host, port, afi): return mocked_conns[(host, port)] # patch to avoid making requests before we want it @@ -409,3 +409,4 @@ class TestSimpleClient(unittest.TestCase): self.assertEqual(big_num + 1, client._next_id()) self.assertEqual(big_num + 2, client._next_id()) self.assertEqual(0, client._next_id()) + |