openstack/deb-python-kafka
Code Issues Proposed changes

Catch and retry on RequestTimedOutError in producer failover test

Browse Source
This commit is contained in:
Dana Powers
2015-12-29 17:49:04 -08:00
parent c549a2b403
commit 3748bade28
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
test/test_failover_integration.py
View File

@@ -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