summaryrefslogtreecommitdiff
path: root/test/test_failover_integration.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/test_failover_integration.py')
-rw-r--r--test/test_failover_integration.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/test/test_failover_integration.py b/test/test_failover_integration.py
index 6c0e662..40269ee 100644
--- a/test/test_failover_integration.py
+++ b/test/test_failover_integration.py
@@ -1,7 +1,7 @@
import logging
import os
import time
-import unittest2
+from . import unittest
from kafka import * # noqa
from kafka.common import * # noqa
@@ -82,7 +82,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):
key, topic, partition = random_string(5), self.topic, 0