summaryrefslogtreecommitdiff
path: root/test/test_sender.py
diff options
context:
space:
mode:
authorDana Powers <dana.powers@gmail.com>2016-07-10 21:36:41 -0700
committerDana Powers <dana.powers@gmail.com>2016-07-16 14:05:49 -0700
commitc34d13879641d27cceb9403a4e6617152dfda0f3 (patch)
tree5e7208f13a59352097829fda76cf007e9aec2d53 /test/test_sender.py
parent20f4c95289c694f81a60228a9820601eb57402f4 (diff)
downloadkafka-python-c34d13879641d27cceb9403a4e6617152dfda0f3.tar.gz
Add initial producer-sender metrics
Diffstat (limited to 'test/test_sender.py')
-rw-r--r--test/test_sender.py16
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"), [