summaryrefslogtreecommitdiff
path: root/test/test_consumer.py
diff options
context:
space:
mode:
authorDana Powers <dana.powers@rd.io>2016-01-07 18:51:14 -0800
committerDana Powers <dana.powers@rd.io>2016-01-07 18:51:14 -0800
commit828377377da43749af0d27ee256ef31bf714cf17 (patch)
treefbad4d4381fc4d1ea2be7ce2009214d18fbeb674 /test/test_consumer.py
parent71e7568fcb8132899f366b37c32645fd5a40dc4b (diff)
parent9a8af1499ca425366d934487469d9977fae7fe5f (diff)
downloadkafka-python-828377377da43749af0d27ee256ef31bf714cf17.tar.gz
Merge branch '0.9'
Conflicts: kafka/codec.py kafka/version.py test/test_producer.py test/test_producer_integration.py
Diffstat (limited to 'test/test_consumer.py')
-rw-r--r--test/test_consumer.py18
1 files changed, 7 insertions, 11 deletions
diff --git a/test/test_consumer.py b/test/test_consumer.py
index df15115..2c9561b 100644
--- a/test/test_consumer.py
+++ b/test/test_consumer.py
@@ -4,7 +4,7 @@ from . import unittest
from kafka import SimpleConsumer, KafkaConsumer, MultiProcessConsumer
from kafka.common import (
- KafkaConfigurationError, FetchResponse, OffsetFetchResponse,
+ KafkaConfigurationError, FetchResponsePayload, OffsetFetchResponsePayload,
FailedPayloadsError, OffsetAndMessage,
NotLeaderForPartitionError, UnknownTopicOrPartitionError
)
@@ -15,10 +15,6 @@ class TestKafkaConsumer(unittest.TestCase):
with self.assertRaises(AssertionError):
SimpleConsumer(MagicMock(), 'group', 'topic', partitions = [ '0' ])
- def test_broker_list_required(self):
- with self.assertRaises(KafkaConfigurationError):
- KafkaConsumer()
-
class TestMultiProcessConsumer(unittest.TestCase):
def test_partition_list(self):
@@ -52,7 +48,7 @@ class TestSimpleConsumer(unittest.TestCase):
# Mock so that only the first request gets a valid response
def not_leader(request):
- return FetchResponse(request.topic, request.partition,
+ return FetchResponsePayload(request.topic, request.partition,
NotLeaderForPartitionError.errno, -1, ())
client.send_fetch_request.side_effect = self.fail_requests_factory(not_leader)
@@ -72,7 +68,7 @@ class TestSimpleConsumer(unittest.TestCase):
# Mock so that only the first request gets a valid response
def unknown_topic_partition(request):
- return FetchResponse(request.topic, request.partition,
+ return FetchResponsePayload(request.topic, request.partition,
UnknownTopicOrPartitionError.errno, -1, ())
client.send_fetch_request.side_effect = self.fail_requests_factory(unknown_topic_partition)
@@ -86,7 +82,7 @@ class TestSimpleConsumer(unittest.TestCase):
client.get_partition_ids_for_topic.return_value = [0, 1]
def mock_offset_fetch_request(group, payloads, **kwargs):
- return [OffsetFetchResponse(p.topic, p.partition, 0, b'', 0) for p in payloads]
+ return [OffsetFetchResponsePayload(p.topic, p.partition, 0, b'', 0) for p in payloads]
client.send_offset_fetch_request.side_effect = mock_offset_fetch_request
@@ -125,11 +121,11 @@ class TestSimpleConsumer(unittest.TestCase):
# Mock so that only the first request gets a valid response
def fail_requests(payloads, **kwargs):
responses = [
- FetchResponse(payloads[0].topic, payloads[0].partition, 0, 0,
- (OffsetAndMessage(
+ FetchResponsePayload(payloads[0].topic, payloads[0].partition, 0, 0,
+ [OffsetAndMessage(
payloads[0].offset + i,
"msg %d" % (payloads[0].offset + i))
- for i in range(10))),
+ for i in range(10)]),
]
for failure in payloads[1:]:
responses.append(error_factory(failure))