From 9a741933dc692bb560bdfff421531d8fd4798493 Mon Sep 17 00:00:00 2001 From: wheelly Date: Sun, 18 Feb 2024 11:26:55 +0200 Subject: [PATCH] #138 - kafka.read - reformat - missed file --- integration-tests/common/kafka_utils.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/integration-tests/common/kafka_utils.py b/integration-tests/common/kafka_utils.py index a1b2f0d5..aca2ad05 100644 --- a/integration-tests/common/kafka_utils.py +++ b/integration-tests/common/kafka_utils.py @@ -1,8 +1,11 @@ from kafka import KafkaProducer from testcontainers.kafka import KafkaContainer + + def get_kafka_container() -> KafkaContainer: return (KafkaContainer(image="confluentinc/cp-kafka:latest") .with_bind_ports(KafkaContainer.KAFKA_PORT, KafkaContainer.KAFKA_PORT)) + def get_kafka_producer(bootstrap_servers: str) -> KafkaProducer: return KafkaProducer(bootstrap_servers=bootstrap_servers)