Easier local Kafka integration tests (#8107)

Co-authored-by: Sergey Kulik <[email protected]> GitOrigin-RevId: 1affd42c1428e6c62917fe455a56003937285b1d
pathwaycom · Jan 29, 2025 · 9eb8756 · 9eb8756
1 parent 34ae6b9
commit 9eb8756
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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",
     )