From 0d1ccc7485acf09b282d8cc7a06cfd94e690eddb Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Mon, 15 Feb 2016 20:20:56 -0800 Subject: Support setting kafka instance port explicitly in fixture --- test/test_failover_integration.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) (limited to 'test/test_failover_integration.py') diff --git a/test/test_failover_integration.py b/test/test_failover_integration.py index afa4ebc..9409241 100644 --- a/test/test_failover_integration.py +++ b/test/test_failover_integration.py @@ -28,8 +28,11 @@ class TestFailover(KafkaIntegrationTestCase): # mini zookeeper, 3 kafka brokers self.zk = ZookeeperFixture.instance() - kk_args = [self.zk.host, self.zk.port, zk_chroot, replicas, partitions] - self.brokers = [KafkaFixture.instance(i, *kk_args) for i in range(replicas)] + kk_args = [self.zk.host, self.zk.port] + kk_kwargs = {'zk_chroot': zk_chroot, 'replicas': replicas, + 'partitions': partitions} + self.brokers = [KafkaFixture.instance(i, *kk_args, **kk_kwargs) + for i in range(replicas)] hosts = ['%s:%d' % (b.host, b.port) for b in self.brokers] self.client = SimpleClient(hosts, timeout=2) -- cgit v1.2.1