From a3ec9bd8e8c730c9f6715b536c0c590230fc2e28 Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Wed, 9 Dec 2015 15:37:17 -0800 Subject: Update references to kafka.common Request/Response (now Payload) --- test/test_producer.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) (limited to 'test/test_producer.py') diff --git a/test/test_producer.py b/test/test_producer.py index 31282bf..cbc1773 100644 --- a/test/test_producer.py +++ b/test/test_producer.py @@ -10,7 +10,7 @@ from . import unittest from kafka import KafkaClient, SimpleProducer, KeyedProducer from kafka.common import ( AsyncProducerQueueFull, FailedPayloadsError, NotLeaderForPartitionError, - ProduceResponse, RetryOptions, TopicAndPartition + ProduceResponsePayload, RetryOptions, TopicAndPartition ) from kafka.producer.base import Producer, _send_upstream from kafka.protocol import CODEC_NONE @@ -186,7 +186,7 @@ class TestKafkaProducerSendUpstream(unittest.TestCase): offset = offsets[req.topic][req.partition] offsets[req.topic][req.partition] += len(req.messages) responses.append( - ProduceResponse(req.topic, req.partition, 0, offset) + ProduceResponsePayload(req.topic, req.partition, 0, offset) ) return responses @@ -234,8 +234,8 @@ class TestKafkaProducerSendUpstream(unittest.TestCase): def send_side_effect(reqs, *args, **kwargs): if self.client.is_first_time: self.client.is_first_time = False - return [ProduceResponse(req.topic, req.partition, - NotLeaderForPartitionError.errno, -1) + return [ProduceResponsePayload(req.topic, req.partition, + NotLeaderForPartitionError.errno, -1) for req in reqs] responses = [] @@ -243,7 +243,7 @@ class TestKafkaProducerSendUpstream(unittest.TestCase): offset = offsets[req.topic][req.partition] offsets[req.topic][req.partition] += len(req.messages) responses.append( - ProduceResponse(req.topic, req.partition, 0, offset) + ProduceResponsePayload(req.topic, req.partition, 0, offset) ) return responses -- cgit v1.2.1