diff options
Diffstat (limited to 'test/test_consumer.py')
| -rw-r--r-- | test/test_consumer.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/test/test_consumer.py b/test/test_consumer.py index 97110e2..0cab116 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, OffsetFetchResponse, FailedPayloadsError, OffsetAndMessage, NotLeaderForPartitionError, UnknownTopicOrPartitionError ) @@ -54,7 +54,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) @@ -75,7 +75,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) @@ -128,7 +128,7 @@ 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, + FetchResponsePayload(payloads[0].topic, payloads[0].partition, 0, 0, (OffsetAndMessage( payloads[0].offset + i, "msg %d" % (payloads[0].offset + i)) |