Support setting kafka instance port explicitly in fixture

1 files changed, 5 insertions, 2 deletions

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)