diff options
| author | Dana Powers <dana.powers@rd.io> | 2015-12-10 17:24:39 -0800 |
|---|---|---|
| committer | Dana Powers <dana.powers@rd.io> | 2015-12-10 18:37:03 -0800 |
| commit | 3662cb6c50f1a0f893cf23e18f4d14509b25dfa3 (patch) | |
| tree | 5886fe647e729c65de540edaa3f8f46252806fe7 /test/test_consumer_integration.py | |
| parent | 8411c169ec4d24007e313bb71aa51049a5a42102 (diff) | |
| download | kafka-python-3662cb6c50f1a0f893cf23e18f4d14509b25dfa3.tar.gz | |
Fix consumer integration test that assumed a single broker
Diffstat (limited to 'test/test_consumer_integration.py')
| -rw-r--r-- | test/test_consumer_integration.py | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/test/test_consumer_integration.py b/test/test_consumer_integration.py index a5878d1..eb87b00 100644 --- a/test/test_consumer_integration.py +++ b/test/test_consumer_integration.py @@ -193,13 +193,14 @@ class TestConsumerIntegration(KafkaIntegrationTestCase): self.assert_message_count(messages, 0) self.assertGreaterEqual(t.interval, 1) - self.send_messages(0, range(0, 10)) + self.send_messages(0, range(0, 5)) + self.send_messages(1, range(5, 10)) # Ask for 5 messages, 10 in queue. Get 5 back, no blocking with Timer() as t: - messages = consumer.get_messages(count=5, block=True, timeout=5) + messages = consumer.get_messages(count=5, block=True, timeout=3) self.assert_message_count(messages, 5) - self.assertLessEqual(t.interval, 1) + self.assertLess(t.interval, 3) # Ask for 10 messages, get 5 back, block 1 second with Timer() as t: @@ -209,7 +210,8 @@ class TestConsumerIntegration(KafkaIntegrationTestCase): # Ask for 10 messages, 5 in queue, ask to block for 1 message or 1 # second, get 5 back, no blocking - self.send_messages(0, range(0, 5)) + self.send_messages(0, range(0, 3)) + self.send_messages(1, range(3, 5)) with Timer() as t: messages = consumer.get_messages(count=10, block=1, timeout=1) self.assert_message_count(messages, 5) |