diff options
Diffstat (limited to 'test/test_producer_legacy.py')
| -rw-r--r-- | test/test_producer_legacy.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/test/test_producer_legacy.py b/test/test_producer_legacy.py index 850cb80..9b87c76 100644 --- a/test/test_producer_legacy.py +++ b/test/test_producer_legacy.py @@ -9,12 +9,12 @@ from mock import MagicMock, patch from . import unittest from kafka import SimpleClient, SimpleProducer, KeyedProducer -from kafka.common import ( - AsyncProducerQueueFull, FailedPayloadsError, NotLeaderForPartitionError, - ProduceResponsePayload, RetryOptions, TopicPartition -) +from kafka.errors import ( + AsyncProducerQueueFull, FailedPayloadsError, NotLeaderForPartitionError) from kafka.producer.base import Producer, _send_upstream from kafka.protocol import CODEC_NONE +from kafka.structs import ( + ProduceResponsePayload, RetryOptions, TopicPartition) from six.moves import queue, xrange |