From 0f78d57c604e864fab51f7cfb8fa69c9c4e623c7 Mon Sep 17 00:00:00 2001 From: Tim Evens Date: Wed, 30 Mar 2016 15:32:05 -0700 Subject: 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). --- test/test_client_async.py | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) (limited to 'test/test_client_async.py') diff --git a/test/test_client_async.py b/test/test_client_async.py index 2e0d9b4..e0b98c4 100644 --- a/test/test_client_async.py +++ b/test/test_client_async.py @@ -1,4 +1,5 @@ import time +import socket import pytest @@ -12,11 +13,11 @@ from kafka.protocol.produce import ProduceRequest @pytest.mark.parametrize("bootstrap,expected_hosts", [ - (None, [('localhost', 9092)]), - ('foobar:1234', [('foobar', 1234)]), - ('fizzbuzz', [('fizzbuzz', 9092)]), - ('foo:12,bar:34', [('foo', 12), ('bar', 34)]), - (['fizz:56', 'buzz'], [('fizz', 56), ('buzz', 9092)]), + (None, [('localhost', 9092, socket.AF_INET)]), + ('foobar:1234', [('foobar', 1234, socket.AF_INET)]), + ('fizzbuzz', [('fizzbuzz', 9092, socket.AF_INET)]), + ('foo:12,bar:34', [('foo', 12, socket.AF_INET), ('bar', 34, socket.AF_INET)]), + (['fizz:56', 'buzz'], [('fizz', 56, socket.AF_INET), ('buzz', 9092, socket.AF_INET)]), ]) def test_bootstrap_servers(mocker, bootstrap, expected_hosts): mocker.patch.object(KafkaClient, '_bootstrap') @@ -47,7 +48,7 @@ def conn(mocker): def test_bootstrap_success(conn): conn.state = ConnectionStates.CONNECTED cli = KafkaClient() - conn.assert_called_once_with('localhost', 9092, **cli.config) + conn.assert_called_once_with('localhost', 9092, socket.AF_INET, **cli.config) conn.connect.assert_called_with() conn.send.assert_called_once_with(MetadataRequest([])) assert cli._bootstrap_fails == 0 @@ -57,7 +58,7 @@ def test_bootstrap_success(conn): def test_bootstrap_failure(conn): conn.state = ConnectionStates.DISCONNECTED cli = KafkaClient() - conn.assert_called_once_with('localhost', 9092, **cli.config) + conn.assert_called_once_with('localhost', 9092, socket.AF_INET, **cli.config) conn.connect.assert_called_with() conn.close.assert_called_with() assert cli._bootstrap_fails == 1 -- cgit v1.2.1