Catch and retry on RequestTimedOutError in producer failover test

1 files changed, 4 insertions, 2 deletions

diff --git a/test/test_failover_integration.py b/test/test_failover_integration.py
index 04c9e2b..339a08b 100644
--- a/test/test_failover_integration.py
+++ b/test/test_failover_integration.py
@@ -3,7 +3,9 @@ import os
 import time
 
 from kafka import KafkaClient, SimpleConsumer, KeyedProducer
-from kafka.common import TopicPartition, FailedPayloadsError, ConnectionError
+from kafka.common import (
+    TopicPartition, FailedPayloadsError, ConnectionError, RequestTimedOutError
+)
 from kafka.producer.base import Producer
 
 from test.fixtures import ZookeeperFixture, KafkaFixture
@@ -77,7 +79,7 @@ class TestFailover(KafkaIntegrationTestCase):
                 producer.send_messages(topic, partition, b'success')
                 log.debug("success!")
                 recovered = True
-            except (FailedPayloadsError, ConnectionError):
+            except (FailedPayloadsError, ConnectionError, RequestTimedOutError):
                 log.debug("caught exception sending message -- will retry")
                 continue