Skip to content

Commit e16541e

Browse files
author
Dana Powers
committed
FailedPayloadsError now only has a single payload; use .payload attribute
.failed_payloads attribute has been removed.
1 parent 207499b commit e16541e

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

kafka/common.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -169,9 +169,9 @@ class KafkaTimeoutError(KafkaError):
169169

170170

171171
class FailedPayloadsError(KafkaError):
172-
def __init__(self, failed_payloads, *args):
172+
def __init__(self, payload, *args):
173173
super(FailedPayloadsError, self).__init__(*args)
174-
self.failed_payloads = failed_payloads
174+
self.payload = payload
175175

176176

177177
class ConnectionError(KafkaError):

test/test_producer.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,7 @@ def test_first_send_failed(self):
143143
def send_side_effect(reqs, *args, **kwargs):
144144
if self.client.is_first_time:
145145
self.client.is_first_time = False
146-
return [FailedPayloadsError(reqs)]
146+
return [FailedPayloadsError(req) for req in reqs]
147147
return []
148148

149149
self.client.send_produce_request.side_effect = send_side_effect
@@ -165,7 +165,7 @@ def test_with_limited_retries(self):
165165
self.queue.put((TopicAndPartition("test", i), "msg %i" % i, "key %i" % i))
166166

167167
def send_side_effect(reqs, *args, **kwargs):
168-
return [FailedPayloadsError(reqs)]
168+
return [FailedPayloadsError(req) for req in reqs]
169169

170170
self.client.send_produce_request.side_effect = send_side_effect
171171

0 commit comments

Comments
 (0)