summaryrefslogtreecommitdiff
path: root/test/test_sender.py
diff options
context:
space:
mode:
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"), [