diff options
Diffstat (limited to 'test/test_producer_integration.py')
| -rw-r--r-- | test/test_producer_integration.py | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/test/test_producer_integration.py b/test/test_producer_integration.py index d631402..176c99e 100644 --- a/test/test_producer_integration.py +++ b/test/test_producer_integration.py @@ -10,11 +10,9 @@ from kafka import ( RoundRobinPartitioner, HashedPartitioner ) from kafka.codec import has_snappy -from kafka.common import ( - FetchRequestPayload, ProduceRequestPayload, - UnknownTopicOrPartitionError, LeaderNotAvailableError -) +from kafka.errors import UnknownTopicOrPartitionError, LeaderNotAvailableError from kafka.producer.base import Producer +from kafka.structs import FetchRequestPayload, ProduceRequestPayload from test.fixtures import ZookeeperFixture, KafkaFixture from test.testutil import KafkaIntegrationTestCase, kafka_versions |