summaryrefslogtreecommitdiff
path: root/test
diff options
context:
space:
mode:
authorDana Powers <dana.powers@rd.io>2014-09-09 16:42:23 -0700
committerDana Powers <dana.powers@rd.io>2014-09-10 22:14:29 -0700
commit27e812e66bd04d3acf59ad6792b07d1c7056c036 (patch)
treea9960a0ec8d7fd4e1452a09f0ab7fcd69f6d0494 /test
parentf806c24d849928c95342d2b82a9d2d9507ba175a (diff)
downloadkafka-python-27e812e66bd04d3acf59ad6792b07d1c7056c036.tar.gz
Separate consumers/producers/partitioners
Diffstat (limited to 'test')
-rw-r--r--test/test_consumer_integration.py2
-rw-r--r--test/test_failover_integration.py2
-rw-r--r--test/test_producer.py2
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):