From 9b59825c20240a5d2fdb4a575f3b1e9b9df07b3b Mon Sep 17 00:00:00 2001 From: Jeff Widman Date: Wed, 22 May 2019 18:06:32 -0700 Subject: [PATCH] Remove unused imports --- kafka/cluster.py | 4 ++-- test/test_client_async.py | 6 +----- 2 files changed, 3 insertions(+), 7 deletions(-) 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)