summaryrefslogtreecommitdiff
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
parent207499b26d5ce6c7a72771250b8e4365944aa1bd (diff)
downloadkafka-python-e16541e3cbfb9501099cf02dc237237e7519d637.tar.gz
FailedPayloadsError now only has a single payload; use .payload attribute
.failed_payloads attribute has been removed.
-rw-r--r--kafka/common.py4
-rw-r--r--test/test_producer.py4
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