From 9eb875679bae6c04027344347f5a98b66d84711b Mon Sep 17 00:00:00 2001 From: Jakub Kowalski Date: Wed, 29 Jan 2025 11:56:53 +0100 Subject: [PATCH] Easier local Kafka integration tests (#8107) Co-authored-by: Sergey Kulik <104143901+zxqfd555-pw@users.noreply.github.com> GitOrigin-RevId: 1affd42c1428e6c62917fe455a56003937285b1d --- integration_tests/kafka/test_simple.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/integration_tests/kafka/test_simple.py b/integration_tests/kafka/test_simple.py index 7e214da5..0c4f84b8 100644 --- a/integration_tests/kafka/test_simple.py +++ b/integration_tests/kafka/test_simple.py @@ -169,7 +169,7 @@ def test_kafka_simple_wrapper_bytes_io( kafka_context.fill(["foo", "bar"]) table = pw.io.kafka.simple_read( - "kafka:9092", + kafka_context.default_rdkafka_settings()["bootstrap.servers"], kafka_context.input_topic, ) pw.io.jsonlines.write(table, tmp_path / "output.jsonl") @@ -178,7 +178,7 @@ def test_kafka_simple_wrapper_bytes_io( # check that reread will have all these messages again G.clear() table = pw.io.kafka.simple_read( - "kafka:9092", + kafka_context.default_rdkafka_settings()["bootstrap.servers"], kafka_context.input_topic, ) pw.io.jsonlines.write(table, tmp_path / "output.jsonl") @@ -201,7 +201,7 @@ def test_kafka_simple_wrapper_plaintext_io( kafka_context.fill(["foo", "bar"]) table = pw.io.kafka.simple_read( - "kafka:9092", + kafka_context.default_rdkafka_settings()["bootstrap.servers"], kafka_context.input_topic, format="plaintext", ) @@ -211,7 +211,7 @@ def test_kafka_simple_wrapper_plaintext_io( # check that reread will have all these messages again G.clear() table = pw.io.kafka.simple_read( - "kafka:9092", + kafka_context.default_rdkafka_settings()["bootstrap.servers"], kafka_context.input_topic, format="plaintext", )