diff options
author | Dana Powers <dana.powers@rd.io> | 2014-09-16 22:31:19 -0700 |
---|---|---|
committer | Dana Powers <dana.powers@rd.io> | 2014-12-15 12:42:56 -0800 |
commit | 6b44828685e75117d9def8a80b165e6154f78e11 (patch) | |
tree | 8cddd82fbc54c5f99fc00768735010d6df9f8afa /test | |
parent | 6e6ae533272ed32d150221534a16d588e42f9c51 (diff) | |
download | kafka-python-6b44828685e75117d9def8a80b165e6154f78e11.tar.gz |
Move KafkaConsumer to kafka.consumer.kafka module; make available for import from kafka at top-level
Diffstat (limited to 'test')
-rw-r--r-- | test/test_consumer.py | 3 | ||||
-rw-r--r-- | test/test_consumer_integration.py | 3 |
2 files changed, 2 insertions, 4 deletions
diff --git a/test/test_consumer.py b/test/test_consumer.py index b33e537..7b8f370 100644 --- a/test/test_consumer.py +++ b/test/test_consumer.py @@ -2,8 +2,7 @@ from mock import MagicMock from . import unittest -from kafka.consumer import SimpleConsumer -from kafka.consumer.new import KafkaConsumer +from kafka import SimpleConsumer, KafkaConsumer from kafka.common import KafkaConfigurationError class TestKafkaConsumer(unittest.TestCase): diff --git a/test/test_consumer_integration.py b/test/test_consumer_integration.py index b4af70b..c97081b 100644 --- a/test/test_consumer_integration.py +++ b/test/test_consumer_integration.py @@ -3,12 +3,11 @@ import os from six.moves import xrange -from kafka import SimpleConsumer, MultiProcessConsumer, create_message +from kafka import SimpleConsumer, MultiProcessConsumer, KafkaConsumer, create_message from kafka.common import ( ProduceRequest, ConsumerFetchSizeTooSmall, ConsumerTimeout ) from kafka.consumer.base import MAX_FETCH_BUFFER_SIZE_BYTES -from kafka.consumer.new import KafkaConsumer from test.fixtures import ZookeeperFixture, KafkaFixture from test.testutil import ( |