diff options
Diffstat (limited to 'test/test_failover_integration.py')
| -rw-r--r-- | test/test_failover_integration.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/test/test_failover_integration.py b/test/test_failover_integration.py index 5e737b0..b572b4e 100644 --- a/test/test_failover_integration.py +++ b/test/test_failover_integration.py @@ -1,7 +1,8 @@ import logging import os import time -import unittest2 + +from . import unittest from kafka import KafkaClient, SimpleConsumer from kafka.common import TopicAndPartition, FailedPayloadsError, ConnectionError @@ -65,7 +66,7 @@ class TestFailover(KafkaIntegrationTestCase): while not recovered and (time.time() - started) < timeout: try: logging.debug("attempting to send 'success' message after leader killed") - producer.send_messages(topic, partition, 'success') + producer.send_messages(topic, partition, b'success') logging.debug("success!") recovered = True except (FailedPayloadsError, ConnectionError): @@ -84,7 +85,7 @@ class TestFailover(KafkaIntegrationTestCase): #@kafka_versions("all") - @unittest2.skip("async producer does not support reliable failover yet") + @unittest.skip("async producer does not support reliable failover yet") def test_switch_leader_async(self): topic = self.topic partition = 0 |