summaryrefslogtreecommitdiff
path: root/test
diff options
context:
space:
mode:
authorDana Powers <dana.powers@gmail.com>2014-09-10 23:31:33 -0700
committerDana Powers <dana.powers@gmail.com>2014-09-10 23:31:33 -0700
commit931670ff2d13e1e4953bbcfa4fff45d7dcb94533 (patch)
tree46a3873a5d9d653e09aaf5c7b5fc60167ea22a8c /test
parent04dbd0e7912c43e0d5cf32b29f0250dc67937df7 (diff)
parent27e812e66bd04d3acf59ad6792b07d1c7056c036 (diff)
downloadkafka-python-931670ff2d13e1e4953bbcfa4fff45d7dcb94533.tar.gz
Merge pull request #232 from dpkp/directory_layout
Separate Consumer/Producer/Partitioner modules
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):