diff options
| author | Dana Powers <dana.powers@rd.io> | 2016-01-07 17:57:24 -0800 |
|---|---|---|
| committer | Dana Powers <dana.powers@rd.io> | 2016-01-07 17:57:24 -0800 |
| commit | 9a8af1499ca425366d934487469d9977fae7fe5f (patch) | |
| tree | 01ad6026ca2ecf62e554f4fabacc045b46c01737 /test/test_failover_integration.py | |
| parent | d4e85ecd1d8acac1a0f74d164b67faefd99987e4 (diff) | |
| download | kafka-python-0.9.tar.gz | |
Fix KafkaClient->SimpleClient references0.9
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, |