diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_consumer_integration.py | 2 | ||||
| -rw-r--r-- | test/test_failover_integration.py | 2 | ||||
| -rw-r--r-- | test/test_producer.py | 2 |
3 files changed, 3 insertions, 3 deletions
diff --git a/test/test_consumer_integration.py b/test/test_consumer_integration.py index 6895c0e..717eac5 100644 --- a/test/test_consumer_integration.py +++ b/test/test_consumer_integration.py @@ -4,7 +4,7 @@ from six.moves import xrange from kafka import SimpleConsumer, MultiProcessConsumer, create_message from kafka.common import ProduceRequest, ConsumerFetchSizeTooSmall -from kafka.consumer import MAX_FETCH_BUFFER_SIZE_BYTES +from kafka.consumer.base import MAX_FETCH_BUFFER_SIZE_BYTES from test.fixtures import ZookeeperFixture, KafkaFixture from test.testutil import ( diff --git a/test/test_failover_integration.py b/test/test_failover_integration.py index b572b4e..d307d41 100644 --- a/test/test_failover_integration.py +++ b/test/test_failover_integration.py @@ -6,7 +6,7 @@ from . import unittest from kafka import KafkaClient, SimpleConsumer from kafka.common import TopicAndPartition, FailedPayloadsError, ConnectionError -from kafka.producer import Producer +from kafka.producer.base import Producer from test.fixtures import ZookeeperFixture, KafkaFixture from test.testutil import ( diff --git a/test/test_producer.py b/test/test_producer.py index 1375525..caf8fe3 100644 --- a/test/test_producer.py +++ b/test/test_producer.py @@ -5,7 +5,7 @@ import logging from mock import MagicMock from . import unittest -from kafka.producer import Producer +from kafka.producer.base import Producer class TestKafkaProducer(unittest.TestCase): def test_producer_message_types(self): |