summaryrefslogtreecommitdiff
path: root/test/test_client.py
diff options
context:
space:
mode:
authorMark Roberts <wizzat@gmail.com>2014-04-23 13:10:43 -0700
committerMark Roberts <wizzat@gmail.com>2014-04-23 13:10:43 -0700
commit583d3ae22bf6fc373c03e50c7e4eb26eaa50db4b (patch)
treee5d649029cd4960a4f00ccd16f5837de45f4d699 /test/test_client.py
parent764f2053ad4dd73dc391416ddd4cfa345271efcb (diff)
downloadkafka-python-583d3ae22bf6fc373c03e50c7e4eb26eaa50db4b.tar.gz
Fix Python 2.6 support
Diffstat (limited to 'test/test_client.py')
-rw-r--r--test/test_client.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/test/test_client.py b/test/test_client.py
index 9520d48..fe9beff 100644
--- a/test/test_client.py
+++ b/test/test_client.py
@@ -1,7 +1,7 @@
import os
import random
import struct
-import unittest
+import unittest2
from mock import MagicMock, patch
@@ -15,7 +15,7 @@ from kafka.protocol import (
create_message, KafkaProtocol
)
-class TestKafkaClient(unittest.TestCase):
+class TestKafkaClient(unittest2.TestCase):
def test_init_with_list(self):
with patch.object(KafkaClient, 'load_metadata_for_topics'):
client = KafkaClient(hosts=['kafka01:9092', 'kafka02:9092', 'kafka03:9092'])