diff --git a/tests/opentelemetry/kafka/test_kafka.py b/tests/opentelemetry/kafka/test_kafka.py index 0967069ade..cc38a66281 100644 --- a/tests/opentelemetry/kafka/test_kafka.py +++ b/tests/opentelemetry/kafka/test_kafka.py @@ -261,17 +261,19 @@ async def handler(m, baggage: CurrentBaggage): @pytest.mark.kafka class TestPublishWithTelemetry(TestPublish): - def get_broker(self, apply_types: bool = False): + def get_broker(self, apply_types: bool = False, **kwargs): return KafkaBroker( middlewares=(KafkaTelemetryMiddleware(),), apply_types=apply_types, + **kwargs, ) @pytest.mark.kafka class TestConsumeWithTelemetry(TestConsume): - def get_broker(self, apply_types: bool = False): + def get_broker(self, apply_types: bool = False, **kwargs): return KafkaBroker( middlewares=(KafkaTelemetryMiddleware(),), apply_types=apply_types, + **kwargs, )