From 5c0bf5f377f15e301ad8ac011854db54ab0f6f92 Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Wed, 9 Dec 2015 15:48:30 -0800 Subject: Update to Payload namedtuples in consumer tests --- test/test_consumer.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'test/test_consumer.py') 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)) -- cgit v1.2.1