diff options
Diffstat (limited to 'test/test_failover_integration.py')
-rw-r--r-- | test/test_failover_integration.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/test/test_failover_integration.py b/test/test_failover_integration.py index 5ffaa04..b54ace0 100644 --- a/test/test_failover_integration.py +++ b/test/test_failover_integration.py @@ -2,7 +2,7 @@ import logging import os import time -from kafka import KafkaClient, SimpleConsumer, KeyedProducer +from kafka import SimpleClient, SimpleConsumer, KeyedProducer from kafka.common import ( TopicPartition, FailedPayloadsError, ConnectionError, RequestTimedOutError ) @@ -34,7 +34,7 @@ class TestFailover(KafkaIntegrationTestCase): self.brokers = [KafkaFixture.instance(i, *kk_args) for i in range(replicas)] hosts = ['%s:%d' % (b.host, b.port) for b in self.brokers] - self.client = KafkaClient(hosts, timeout=2) + self.client = SimpleClient(hosts, timeout=2) super(TestFailover, self).setUp() def tearDown(self): @@ -214,7 +214,7 @@ class TestFailover(KafkaIntegrationTestCase): hosts = ','.join(['%s:%d' % (broker.host, broker.port) for broker in self.brokers]) - client = KafkaClient(hosts) + client = SimpleClient(hosts) consumer = SimpleConsumer(client, None, topic, partitions=partitions, auto_commit=False, |