diff options
Diffstat (limited to 'test/test_failover_integration.py')
| -rw-r--r-- | test/test_failover_integration.py | 9 |
1 files changed, 5 insertions, 4 deletions
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() |