diff options
| -rw-r--r-- | kafka/common.py | 4 | ||||
| -rw-r--r-- | test/test_producer.py | 4 |
2 files changed, 4 insertions, 4 deletions
diff --git a/kafka/common.py b/kafka/common.py index 8c13798..2fdf7d2 100644 --- a/kafka/common.py +++ b/kafka/common.py @@ -169,9 +169,9 @@ class KafkaTimeoutError(KafkaError): class FailedPayloadsError(KafkaError): - def __init__(self, failed_payloads, *args): + def __init__(self, payload, *args): super(FailedPayloadsError, self).__init__(*args) - self.failed_payloads = failed_payloads + self.payload = payload class ConnectionError(KafkaError): diff --git a/test/test_producer.py b/test/test_producer.py index 258b9c3..85a5a2e 100644 --- a/test/test_producer.py +++ b/test/test_producer.py @@ -143,7 +143,7 @@ class TestKafkaProducerSendUpstream(unittest.TestCase): def send_side_effect(reqs, *args, **kwargs): if self.client.is_first_time: self.client.is_first_time = False - return [FailedPayloadsError(reqs)] + return [FailedPayloadsError(req) for req in reqs] return [] self.client.send_produce_request.side_effect = send_side_effect @@ -165,7 +165,7 @@ class TestKafkaProducerSendUpstream(unittest.TestCase): self.queue.put((TopicAndPartition("test", i), "msg %i" % i, "key %i" % i)) def send_side_effect(reqs, *args, **kwargs): - return [FailedPayloadsError(reqs)] + return [FailedPayloadsError(req) for req in reqs] self.client.send_produce_request.side_effect = send_side_effect |