diff options
Diffstat (limited to 'test/test_client.py')
| -rw-r--r-- | test/test_client.py | 11 |
1 files changed, 4 insertions, 7 deletions
diff --git a/test/test_client.py b/test/test_client.py index 6980434..42d7dbd 100644 --- a/test/test_client.py +++ b/test/test_client.py @@ -6,17 +6,14 @@ import six from . import unittest from kafka import SimpleClient -from kafka.common import ( - ProduceRequestPayload, - BrokerMetadata, - TopicPartition, KafkaUnavailableError, - LeaderNotAvailableError, UnknownTopicOrPartitionError, - KafkaTimeoutError, ConnectionError, FailedPayloadsError -) from kafka.conn import KafkaConnection +from kafka.errors import ( + KafkaUnavailableError, LeaderNotAvailableError, KafkaTimeoutError, + UnknownTopicOrPartitionError, ConnectionError, FailedPayloadsError) from kafka.future import Future from kafka.protocol import KafkaProtocol, create_message from kafka.protocol.metadata import MetadataResponse +from kafka.structs import ProduceRequestPayload, BrokerMetadata, TopicPartition from test.testutil import Timer |