Merge pull request #754 from dpkp/benchmarks

Producer metrics + consumer/producer benchmark scripts

1 files changed, 11 insertions, 5 deletions

diff --git a/test/test_sender.py b/test/test_sender.py
index 44105e2..cf911e1 100644
--- a/test/test_sender.py
+++ b/test/test_sender.py
@@ -7,12 +7,13 @@ import pytest
 
 from kafka.client_async import KafkaClient
 from kafka.cluster import ClusterMetadata
-from kafka.producer.buffer import MessageSetBuffer
-from kafka.producer.sender import Sender
-from kafka.producer.record_accumulator import RecordAccumulator, RecordBatch
 import kafka.errors as Errors
 from kafka.future import Future
+from kafka.metrics import Metrics
+from kafka.producer.buffer import MessageSetBuffer
 from kafka.protocol.produce import ProduceRequest
+from kafka.producer.record_accumulator import RecordAccumulator, RecordBatch
+from kafka.producer.sender import Sender
 from kafka.structs import TopicPartition, OffsetAndMetadata
 
 
@@ -29,8 +30,13 @@ def accumulator():
 
 
 @pytest.fixture
-def sender(client, accumulator):
-    return Sender(client, client.cluster, accumulator)
+def metrics():
+    return Metrics()
+
+
+@pytest.fixture
+def sender(client, accumulator, metrics):
+    return Sender(client, client.cluster, accumulator, metrics)
 
 
 @pytest.mark.parametrize(("api_version", "produce_version"), [