Retry on FailedPayloadsError from consumer.pending() in failover tests

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()