diff options
Diffstat (limited to 'test/test_failover_integration.py')
| -rw-r--r-- | test/test_failover_integration.py | 6 |
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 |