summaryrefslogtreecommitdiff
path: root/test/test_failover_integration.py
diff options
context:
space:
mode:
authorBruno ReniƩ <brutasse@gmail.com>2014-08-28 14:40:01 +0200
committerMark Roberts <wizzat@fb.com>2014-09-03 09:55:44 -0700
commit83af5102e995e854a1980b90f1400afdd098da37 (patch)
tree403e6dabbe62ac4f5bb6b2a76a5eed0f183d86b3 /test/test_failover_integration.py
parenteddd1436c226545237aa057c35719950702466ed (diff)
downloadkafka-python-83af5102e995e854a1980b90f1400afdd098da37.tar.gz
Use base unittest or unittest2 depending on python version
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