diff options
author | Dana Powers <dana.powers@rd.io> | 2016-01-07 18:51:14 -0800 |
---|---|---|
committer | Dana Powers <dana.powers@rd.io> | 2016-01-07 18:51:14 -0800 |
commit | 828377377da43749af0d27ee256ef31bf714cf17 (patch) | |
tree | fbad4d4381fc4d1ea2be7ce2009214d18fbeb674 /test/test_consumer.py | |
parent | 71e7568fcb8132899f366b37c32645fd5a40dc4b (diff) | |
parent | 9a8af1499ca425366d934487469d9977fae7fe5f (diff) | |
download | kafka-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.py | 18 |
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)) |