diff options
| -rw-r--r-- | test/test_consumer_integration.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/test/test_consumer_integration.py b/test/test_consumer_integration.py index 9c89190..7dd4a79 100644 --- a/test/test_consumer_integration.py +++ b/test/test_consumer_integration.py @@ -63,6 +63,8 @@ class TestConsumerIntegration(KafkaIntegrationTestCase): if consumer_class == SimpleConsumer: kwargs.setdefault('iter_timeout', 0) + elif consumer_class == MultiProcessConsumer: + kwargs.setdefault('simple_consumer_options', {'iter_timeout': 0}) return consumer_class(self.client, group, topic, **kwargs) @@ -243,7 +245,8 @@ class TestConsumerIntegration(KafkaIntegrationTestCase): self.send_messages(0, range(0, 10)) self.send_messages(1, range(10, 20)) - consumer = MultiProcessConsumer(self.client, "group1", self.topic, auto_commit=False) + consumer = MultiProcessConsumer(self.client, "group1", self.topic, auto_commit=False, + simple_consumer_options={'iter_timeout': 0}) self.assertEqual(consumer.pending(), 20) self.assertEqual(consumer.pending(partitions=[0]), 10) |