From fa28b2a9f4738730bdb62b03bd6dd09e8c7ba152 Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Fri, 18 Dec 2015 08:52:21 -0800 Subject: Retry on FailedPayloadsError from consumer.pending() in failover tests --- test/test_failover_integration.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) (limited to 'test') diff --git a/test/test_failover_integration.py b/test/test_failover_integration.py index 28d671a..ae5cc51 100644 --- a/test/test_failover_integration.py +++ b/test/test_failover_integration.py @@ -214,11 +214,12 @@ class TestFailover(KafkaIntegrationTestCase): iter_timeout=timeout) started_at = time.time() - pending = consumer.pending(partitions) - - # Keep checking if it isn't immediately correct, subject to timeout + pending = -1 while pending < check_count and (time.time() - started_at < timeout): - pending = consumer.pending(partitions) + try: + pending = consumer.pending(partitions) + except FailedPayloadsError: + pass time.sleep(0.5) consumer.stop() -- cgit v1.2.1