Fixups to benchmark scripts for py3 / new KafkaFixture interface

dpkp · dpkp · commit de6e9d3cc31d · 2019-03-30T08:25:40.000-07:00
diff --git a/benchmarks/consumer_performance.py b/benchmarks/consumer_performance.py
@@ -29,7 +29,7 @@ def start_brokers(n):
     replicas = min(n, 3)
     print('-> {0} Brokers [{1} partitions / {2} replicas]'.format(n, partitions, replicas))
     brokers = [
-        KafkaFixture.instance(i, zk.host, zk.port, zk_chroot='',
+        KafkaFixture.instance(i, zk, zk_chroot='',
                               partitions=partitions, replicas=replicas)
         for i in range(n)
     ]
@@ -148,7 +148,7 @@ def get_args_parser():
         help='Topic for consumer test',
         default='kafka-python-benchmark-test')
     parser.add_argument(
-        '--num-records', type=long,
+        '--num-records', type=int,
         help='number of messages to consume',
         default=1000000)
     parser.add_argument(
diff --git a/benchmarks/producer_performance.py b/benchmarks/producer_performance.py
@@ -26,7 +26,7 @@ def start_brokers(n):
     replicas = min(n, 3)
     print('-> {0} Brokers [{1} partitions / {2} replicas]'.format(n, partitions, replicas))
     brokers = [
-        KafkaFixture.instance(i, zk.host, zk.port, zk_chroot='',
+        KafkaFixture.instance(i, zk, zk_chroot='',
                               partitions=partitions, replicas=replicas)
         for i in range(n)
     ]
@@ -130,7 +130,7 @@ def get_args_parser():
         help='Topic name for test',
         default='kafka-python-benchmark-test')
     parser.add_argument(
-        '--num-records', type=long,
+        '--num-records', type=int,
         help='number of messages to produce',
         default=1000000)
     parser.add_argument(
