Add test support for ccm Scylla Cloud clusters

This amounts to enabling support for using "--sni-proxy" and "--sni-port" flags
and probing for path to generated Scylla Cloud configuration file.

Author: Bouncheck

test-infra/src/main/java/com/datastax/oss/driver/api/testinfra/ccm/CcmBridge.java

@@ -23,6 +23,7 @@
 
 import com.datastax.oss.driver.api.core.Version;
 import com.datastax.oss.driver.shaded.guava.common.base.Joiner;
+import com.datastax.oss.driver.shaded.guava.common.base.Preconditions;
 import com.datastax.oss.driver.shaded.guava.common.collect.ImmutableMap;
 import com.datastax.oss.driver.shaded.guava.common.collect.Maps;
 import com.datastax.oss.driver.shaded.guava.common.io.Resources;
@@ -179,6 +180,8 @@ public class CcmBridge implements AutoCloseable {
   private final List<String> dseWorkloads;
   private final String jvmArgs;
 
+  private final int sniProxyPort;
+
   private CcmBridge(
       Path configDirectory,
       int[] nodes,
@@ -188,7 +191,8 @@ private CcmBridge(
       List<String> dseConfigurationRawYaml,
       List<String> createOptions,
       Collection<String> jvmArgs,
-      List<String> dseWorkloads) {
+      List<String> dseWorkloads,
+      int sniProxyPort) {
     this.configDirectory = configDirectory;
     if (nodes.length == 1) {
       // Hack to ensure that the default DC is always called 'dc1': pass a list ('-nX:0') even if
@@ -216,6 +220,7 @@ private CcmBridge(
     }
     this.jvmArgs = allJvmArgs.toString();
     this.dseWorkloads = dseWorkloads;
+    this.sniProxyPort = sniProxyPort;
   }
 
   // Copied from Netty's PlatformDependent to avoid the dependency on Netty
@@ -290,6 +295,14 @@ private String getCcmVersionString(Version version) {
     return version.toString();
   }
 
+  public String getScyllaCloudConfigPathString() {
+    return configDirectory.toFile().getAbsolutePath() + "/" + CLUSTER_NAME + "/config_data.yaml";
+  }
+
+  public String getIpPrefix() {
+    return ipPrefix;
+  }
+
   public void create() {
     if (created.compareAndSet(false, true)) {
       if (INSTALL_DIRECTORY != null) {
@@ -359,7 +372,12 @@ public void reloadCore(int node, String keyspace, String table, boolean reindex)
   public void start() {
     if (started.compareAndSet(false, true)) {
       try {
-        execute("start", jvmArgs, "--wait-for-binary-proto");
+        execute(
+            "start",
+            jvmArgs,
+            "--wait-for-binary-proto",
+            (sniProxyPort >= 0 ? "--sni-proxy" : ""),
+            (sniProxyPort > 0 ? "--sni-port=" + sniProxyPort : ""));
       } catch (RuntimeException re) {
         // if something went wrong starting CCM, see if we can also dump the error
         executeCheckLogError();
@@ -510,6 +528,8 @@ public static class Builder {
 
     private final Path configDirectory;
 
+    private int sniProxyPort = -1;
+
     private Builder() {
       try {
         this.configDirectory = Files.createTempDirectory("ccm");
@@ -614,6 +634,14 @@ public Builder withDseWorkloads(String... workloads) {
       return this;
     }
 
+    /** Enable SNI proxy and use given port number. Port 0 means any port. */
+    public Builder withSniProxy(int port) {
+      Preconditions.checkArgument(port >= 0);
+      Preconditions.checkArgument(port <= 65535);
+      this.sniProxyPort = port;
+      return this;
+    }
+
     public CcmBridge build() {
       return new CcmBridge(
           configDirectory,
@@ -624,7 +652,8 @@ public CcmBridge build() {
           dseRawYaml,
           createOptions,
           jvmArgs,
-          dseWorkloads);
+          dseWorkloads,
+          sniProxyPort);
     }
   }
 }

test-infra/src/main/java/com/datastax/oss/driver/api/testinfra/ccm/CustomCcmRule.java

@@ -112,6 +112,11 @@ public Builder withSslAuth() {
       return this;
     }
 
+    public Builder withSniProxy(int port) {
+      bridgeBuilder.withSniProxy(port);
+      return this;
+    }
+
     public CustomCcmRule build() {
       return new CustomCcmRule(bridgeBuilder.build());
     }