Allow test clusters to specify additional jvm arguments (elastic#97584)

Somet special tests like die-with-dignity need to pass additional jvm
arguments (exit on OOM). This commit adds a new jvmArg to the cluster
spec for the new test clusters infra.

Author: rjernst

test/test-clusters/src/main/java/org/elasticsearch/test/cluster/local/AbstractLocalClusterSpecBuilder.java

@@ -161,6 +161,7 @@ private LocalNodeSpec build(LocalClusterSpec cluster) {
                 getKeystorePassword(),
                 getExtraConfigFiles(),
                 getSystemProperties(),
+                getJvmArgs(),
                 getSecrets()
             );
         }

test/test-clusters/src/main/java/org/elasticsearch/test/cluster/local/AbstractLocalSpecBuilder.java

@@ -38,6 +38,7 @@ public abstract class AbstractLocalSpecBuilder<T extends LocalSpecBuilder<?>> im
     private final Map<String, Resource> keystoreFiles = new HashMap<>();
     private final Map<String, Resource> extraConfigFiles = new HashMap<>();
     private final Map<String, String> systemProperties = new HashMap<>();
+    private final List<String> jvmArgs = new ArrayList<>();
     private final Map<String, String> secrets = new HashMap<>();
     private DistributionType distributionType;
     private Version version;
@@ -218,6 +219,16 @@ public Map<String, String> getSystemProperties() {
         return inherit(() -> parent.getSystemProperties(), systemProperties);
     }
 
+    @Override
+    public T jvmArg(String arg) {
+        this.jvmArgs.add(arg);
+        return cast(this);
+    }
+
+    public List<String> getJvmArgs() {
+        return inherit(() -> parent.getJvmArgs(), jvmArgs);
+    }
+
     @Override
     public T keystorePassword(String password) {
         this.keystorePassword = password;

test/test-clusters/src/main/java/org/elasticsearch/test/cluster/local/LocalClusterFactory.java

@@ -635,6 +635,7 @@ private Map<String, String> getEnvironmentVariables() {
                     .map(p -> p.replace("${ES_PATH_CONF}", configDir.toString()))
                     .collect(Collectors.joining(" "));
             }
+            String jvmArgs = String.join(" ", spec.getJvmArgs());
 
             String debugArgs = "";
             if (Boolean.getBoolean(TESTS_CLUSTER_DEBUG_ENABLED_SYSPROP)) {
@@ -649,6 +650,7 @@ private Map<String, String> getEnvironmentVariables() {
                 + " "
                 + featureFlagProperties
                 + systemProperties
+                + jvmArgs
                 + debugArgs);
 
             return environment;

test/test-clusters/src/main/java/org/elasticsearch/test/cluster/local/LocalClusterSpec.java

@@ -83,6 +83,7 @@ public static class LocalNodeSpec {
         private final String keystorePassword;
         private final Map<String, Resource> extraConfigFiles;
         private final Map<String, String> systemProperties;
+        private final List<String> jvmArgs;
         private final Map<String, String> secrets;
         private Version version;
 
@@ -104,6 +105,7 @@ public LocalNodeSpec(
             String keystorePassword,
             Map<String, Resource> extraConfigFiles,
             Map<String, String> systemProperties,
+            List<String> jvmArgs,
             Map<String, String> secrets
         ) {
             this.cluster = cluster;
@@ -123,6 +125,7 @@ public LocalNodeSpec(
             this.keystorePassword = keystorePassword;
             this.extraConfigFiles = extraConfigFiles;
             this.systemProperties = systemProperties;
+            this.jvmArgs = jvmArgs;
             this.secrets = secrets;
         }
 
@@ -182,6 +185,10 @@ public Map<String, String> getSystemProperties() {
             return systemProperties;
         }
 
+        public List<String> getJvmArgs() {
+            return jvmArgs;
+        }
+
         public Map<String, String> getSecrets() {
             return secrets;
         }
@@ -299,6 +306,7 @@ private LocalNodeSpec getFilteredSpec(SettingsProvider filteredProvider, Setting
                         n.keystorePassword,
                         n.extraConfigFiles,
                         n.systemProperties,
+                        n.jvmArgs,
                         n.secrets
                     )
                 )

test/test-clusters/src/main/java/org/elasticsearch/test/cluster/local/LocalSpecBuilder.java

@@ -126,4 +126,9 @@ interface LocalSpecBuilder<T extends LocalSpecBuilder<?>> {
      * Adds a system property to node JVM arguments.
      */
     T systemProperty(String property, String value);
+
+    /**
+     * Adds an additional command line argument to node JVM arguments.
+     */
+    T jvmArg(String arg);
 }