diff options
Diffstat (limited to 'test/test_sender.py')
-rw-r--r-- | test/test_sender.py | 16 |
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"), [ |