summaryrefslogtreecommitdiff
path: root/test
diff options
context:
space:
mode:
authorDana Powers <dana.powers@rd.io>2015-06-04 13:26:09 -0700
committerDana Powers <dana.powers@rd.io>2015-06-06 16:46:36 -0700
commite16541e3cbfb9501099cf02dc237237e7519d637 (patch)
tree3d19b6196133f24b94c3bc239a79651b49d89cbe /test
parent207499b26d5ce6c7a72771250b8e4365944aa1bd (diff)
downloadkafka-python-e16541e3cbfb9501099cf02dc237237e7519d637.tar.gz
FailedPayloadsError now only has a single payload; use .payload attribute
.failed_payloads attribute has been removed.
Diffstat (limited to 'test')
-rw-r--r--test/test_producer.py4
1 files changed, 2 insertions, 2 deletions
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