diff --git a/kafka/cluster.py b/kafka/cluster.py index 28b71c9d1..41695496b 100644 --- a/kafka/cluster.py +++ b/kafka/cluster.py @@ -9,7 +9,7 @@ from kafka.vendor import six from kafka import errors as Errors -from kafka.conn import collect_hosts, dns_lookup +from kafka.conn import collect_hosts from kafka.future import Future from kafka.structs import BrokerMetadata, PartitionMetadata, TopicPartition @@ -189,7 +189,7 @@ def request_update(self): with self._lock: self._need_update = True if not self._future or self._future.is_done: - self._future = Future() + self._future = Future() return self._future def topics(self, exclude_internal_topics=True): diff --git a/test/test_client_async.py b/test/test_client_async.py index 2132c8e4c..77f6b6b7c 100644 --- a/test/test_client_async.py +++ b/test/test_client_async.py @@ -17,15 +17,13 @@ from kafka.conn import ConnectionStates import kafka.errors as Errors from kafka.future import Future -from kafka.protocol.metadata import MetadataResponse, MetadataRequest +from kafka.protocol.metadata import MetadataRequest from kafka.protocol.produce import ProduceRequest from kafka.structs import BrokerMetadata @pytest.fixture def cli(mocker, conn): - mocker.patch('kafka.cluster.dns_lookup', - return_value=[(socket.AF_INET, None, None, None, ('localhost', 9092))]) client = KafkaClient(api_version=(0, 9)) client.poll(future=client.cluster.request_update()) return client @@ -33,8 +31,6 @@ def cli(mocker, conn): def test_bootstrap(mocker, conn): conn.state = ConnectionStates.CONNECTED - mocker.patch('kafka.cluster.dns_lookup', - return_value=[(socket.AF_INET, None, None, None, ('localhost', 9092))]) cli = KafkaClient(api_version=(0, 9)) future = cli.cluster.request_update() cli.poll(future=future)