diff options
Diffstat (limited to 'test/test_producer.py')
| -rw-r--r-- | test/test_producer.py | 4 |
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 |