[FLINK-8961][tests] Add MiniClusterResource#getClientConfiguration

deepfield · Apr 16, 2018 · b1f3ca3 · b1f3ca3
1 parent 8eb4604
commit b1f3ca3
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/...parent/flink-test-utils/src/main/java/org/apache/flink/test/util/MiniClusterResource.java b/...parent/flink-test-utils/src/main/java/org/apache/flink/test/util/MiniClusterResource.java
@@ -25,8 +25,10 @@
 import org.apache.flink.configuration.ConfigConstants;
 import org.apache.flink.configuration.Configuration;
 import org.apache.flink.configuration.CoreOptions;
+import org.apache.flink.configuration.JobManagerOptions;
 import org.apache.flink.configuration.RestOptions;
 import org.apache.flink.configuration.TaskManagerOptions;
+import org.apache.flink.configuration.UnmodifiableConfiguration;
 import org.apache.flink.runtime.akka.AkkaUtils;
 import org.apache.flink.runtime.minicluster.JobExecutorService;
 import org.apache.flink.runtime.minicluster.LocalFlinkMiniCluster;
@@ -67,6 +69,8 @@ public class MiniClusterResource extends ExternalResource {
 
  private ClusterClient<?> clusterClient;
 
+ private Configuration restClusterClientConfig;
+
  private int numberSlots = -1;
 
  private TestEnvironment executionEnvironment;
@@ -117,6 +121,10 @@ public ClusterClient<?> getClusterClient() {
  return clusterClient;
  }
 
+ public Configuration getClientConfiguration() {
+ return restClusterClientConfig;
+ }
+
  public TestEnvironment getTestEnvironment() {
  return executionEnvironment;
  }
@@ -194,6 +202,9 @@ private void startLegacyMiniCluster() throws Exception {
  if (enableClusterClient) {
  clusterClient = new StandaloneClusterClient(configuration, flinkMiniCluster.highAvailabilityServices(), true);
  }
+ Configuration restClientConfig = new Configuration();
+ restClientConfig.setInteger(JobManagerOptions.PORT, flinkMiniCluster.getLeaderRPCPort());
+ this.restClusterClientConfig = new UnmodifiableConfiguration(restClientConfig);
  }
 
  private void startMiniCluster() throws Exception {
@@ -229,6 +240,10 @@ private void startMiniCluster() throws Exception {
  if (enableClusterClient) {
  clusterClient = new MiniClusterClient(configuration, miniCluster);
  }
+ Configuration restClientConfig = new Configuration();
+ restClientConfig.setString(JobManagerOptions.ADDRESS, miniCluster.getRestAddress().getHost());
+ restClientConfig.setInteger(RestOptions.REST_PORT, miniCluster.getRestAddress().getPort());
+ this.restClusterClientConfig = new UnmodifiableConfiguration(restClientConfig);
  }
 
  /**