Allow passing coordinator properties for Delta query runner

trinodb · Mar 30, 2022 · dec5ca6 · dec5ca6
1 parent 9476347
commit dec5ca6
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/plugin/trino-delta-lake/src/test/java/io/trino/plugin/deltalake/DeltaLakeQueryRunner.java b/plugin/trino-delta-lake/src/test/java/io/trino/plugin/deltalake/DeltaLakeQueryRunner.java
@@ -88,13 +88,14 @@ public static DistributedQueryRunner createDeltaLakeQueryRunner(Map<String, Stri
     public static DistributedQueryRunner createS3DeltaLakeQueryRunner(String catalogName, String schemaName, Map<String, String> connectorProperties, String minioAddress, TestingHadoop testingHadoop)
             throws Exception
     {
-        return createS3DeltaLakeQueryRunner(catalogName, schemaName, ImmutableMap.of(), connectorProperties, minioAddress, testingHadoop, queryRunner -> {});
+        return createS3DeltaLakeQueryRunner(catalogName, schemaName, ImmutableMap.of(), ImmutableMap.of(), connectorProperties, minioAddress, testingHadoop, queryRunner -> {});
     }
 
     public static DistributedQueryRunner createS3DeltaLakeQueryRunner(
             String catalogName,
             String schemaName,
             Map<String, String> extraProperties,
+            Map<String, String> coordinatorProperties,
             Map<String, String> connectorProperties,
             String minioAddress,
             TestingHadoop testingHadoop,
@@ -104,6 +105,7 @@ public static DistributedQueryRunner createS3DeltaLakeQueryRunner(
         return createDockerizedDeltaLakeQueryRunner(
                 catalogName,
                 schemaName,
+                coordinatorProperties,
                 extraProperties,
                 ImmutableMap.<String, String>builder()
                         .put("hive.s3.aws-access-key", MINIO_ACCESS_KEY)
@@ -166,6 +168,7 @@ public static DistributedQueryRunner createDockerizedDeltaLakeQueryRunner(
         return createDockerizedDeltaLakeQueryRunner(
                 catalogName,
                 schemaName,
+                ImmutableMap.of(),
                 extraProperties,
                 connectorProperties,
                 testingHadoop,
@@ -175,6 +178,7 @@ public static DistributedQueryRunner createDockerizedDeltaLakeQueryRunner(
     public static DistributedQueryRunner createDockerizedDeltaLakeQueryRunner(
             String catalogName,
             String schemaName,
+            Map<String, String> coordinatorProperties,
             Map<String, String> extraProperties,
             Map<String, String> connectorProperties,
             TestingHadoop testingHadoop,
@@ -188,6 +192,7 @@ public static DistributedQueryRunner createDockerizedDeltaLakeQueryRunner(
 
         DistributedQueryRunner.Builder<?> builder = DistributedQueryRunner.builder(session);
         extraProperties.forEach(builder::addExtraProperty);
+        coordinatorProperties.forEach(builder::setSingleCoordinatorProperty);
         builder.setAdditionalSetup(additionalSetup);
         DistributedQueryRunner queryRunner = builder.build();
 
@@ -255,7 +260,7 @@ public static void main(String[] args)
                 queryRunner = DeltaLakeQueryRunner.createS3DeltaLakeQueryRunner(
                         DELTA_CATALOG,
                         schema,
-                        ImmutableMap.of("http-server.http.port", "8080"),
+                        ImmutableMap.of("http-server.http.port", "8080"), ImmutableMap.of(),
                         ImmutableMap.of(),
                         dockerizedMinioDataLake.getMinioAddress(),
                         dockerizedMinioDataLake.getTestingHadoop(),