[Tests] Limit Pulsar Functions Java runtime to 128MB of RAM in integr…

…ation tests (apache#10567)
sarvex · May 13, 2021 · 61dc9e3 · 61dc9e3
1 parent 1926dee
commit 61dc9e3
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/...n/src/test/java/org/apache/pulsar/tests/integration/functions/utils/CommandGenerator.java b/...n/src/test/java/org/apache/pulsar/tests/integration/functions/utils/CommandGenerator.java
@@ -30,6 +30,9 @@
 @Setter
 @ToString
 public class CommandGenerator {
+    private static final long MB = 1048576L;
+    private static final long JAVA_RUNTIME_RAM_BYTES = 128 * MB;
+
     public enum Runtime {
         JAVA,
         PYTHON,
@@ -119,7 +122,6 @@ public String generateLocalRunCommand(String codeFile) {
                 }
                 break;
         }
-
         return commandBuilder.toString();
     }
 
@@ -192,6 +194,7 @@ public String generateCreateFunctionCommand(String codeFile) {
         switch (runtime){
             case JAVA:
                 commandBuilder.append(" --jar " + JAVAJAR);
+                commandBuilder.append(" --ram " + JAVA_RUNTIME_RAM_BYTES);
                 break;
             case PYTHON:
                 if (codeFile != null) {
@@ -281,6 +284,7 @@ public String generateUpdateFunctionCommand(String codeFile) {
             switch (runtime) {
                 case JAVA:
                     commandBuilder.append(" --jar " + JAVAJAR);
+                    commandBuilder.append(" --ram " + JAVA_RUNTIME_RAM_BYTES);
                     break;
                 case PYTHON:
                     if (codeFile != null) {