[FLINK-32625][tests] MiniClusterTestEnvironment supports customized M…

…iniClusterResourceConfiguration Signed-off-by: tison <wander4096@gmail.com>
apache · Jul 18, 2023 · e2939f5 · e2939f5
1 parent 663a68f
commit e2939f5
Showing 1 changed file with 17 additions and 11 deletions.
diff --git a/...ain/java/org/apache/flink/connector/testframe/environment/MiniClusterTestEnvironment.java b/...ain/java/org/apache/flink/connector/testframe/environment/MiniClusterTestEnvironment.java
@@ -60,24 +60,30 @@ public class MiniClusterTestEnvironment implements TestEnvironment, ClusterContr
     private boolean isStarted = false;
 
     public MiniClusterTestEnvironment() {
-        Configuration conf = new Configuration();
-        conf.set(METRIC_FETCHER_UPDATE_INTERVAL, METRIC_FETCHER_UPDATE_INTERVAL_MS);
-        this.miniCluster =
-                new MiniClusterWithClientResource(
-                        new MiniClusterResourceConfiguration.Builder()
-                                .setConfiguration(conf)
-                                .setNumberTaskManagers(1)
-                                .setNumberSlotsPerTaskManager(6)
-                                .setRpcServiceSharing(RpcServiceSharing.DEDICATED)
-                                .withHaLeadershipControl()
-                                .build());
+        this(defaultMiniClusterResourceConfiguration());
+    }
+
+    public MiniClusterTestEnvironment(MiniClusterResourceConfiguration conf) {
+        this.miniCluster = new MiniClusterWithClientResource(conf);
         try {
             this.checkpointPath = Files.createTempDirectory("minicluster-environment-checkpoint-");
         } catch (IOException e) {
             throw new RuntimeException("Failed to create temporary checkpoint directory", e);
         }
     }
 
+    private static MiniClusterResourceConfiguration defaultMiniClusterResourceConfiguration() {
+        Configuration conf = new Configuration();
+        conf.set(METRIC_FETCHER_UPDATE_INTERVAL, METRIC_FETCHER_UPDATE_INTERVAL_MS);
+        return new MiniClusterResourceConfiguration.Builder()
+                .setConfiguration(conf)
+                .setNumberTaskManagers(1)
+                .setNumberSlotsPerTaskManager(6)
+                .setRpcServiceSharing(RpcServiceSharing.DEDICATED)
+                .withHaLeadershipControl()
+                .build();
+    }
+
     @Override
     public StreamExecutionEnvironment createExecutionEnvironment(
             TestEnvironmentSettings envOptions) {