Run Kafka and Pinot docker in Pinot query runner

trinodb · Aug 10, 2021 · 2f3889e · 2f3889e
1 parent 750000c
commit 2f3889e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/plugin/trino-pinot/src/test/java/io/trino/plugin/pinot/PinotQueryRunner.java b/plugin/trino-pinot/src/test/java/io/trino/plugin/pinot/PinotQueryRunner.java
@@ -20,6 +20,7 @@
 import io.trino.Session;
 import io.trino.metadata.SessionPropertyManager;
 import io.trino.testing.DistributedQueryRunner;
+import io.trino.testing.kafka.TestingKafka;
 
 import java.util.Map;
 import java.util.Optional;
@@ -59,9 +60,13 @@ public static void main(String[] args)
             throws Exception
     {
         Logging.initialize();
+        TestingKafka kafka = TestingKafka.createWithSchemaRegistry();
+        kafka.start();
+        TestingPinotCluster pinot = new TestingPinotCluster(kafka.getNetwork());
+        pinot.start();
         Map<String, String> properties = ImmutableMap.of("http-server.http.port", "8080");
         Map<String, String> pinotProperties = ImmutableMap.<String, String>builder()
-                .put("pinot.controller-urls", "localhost:9000")
+                .put("pinot.controller-urls", pinot.getControllerConnectString())
                 .put("pinot.segments-per-split", "10")
                 .put("pinot.request-timeout", "3m")
                 .build();