diff options
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()) + |