From 30ba5284f1b0ea03e706c6f3ac44b0542379ce80 Mon Sep 17 00:00:00 2001
From: Anna Kozlova <anna.kozlova@jetbrains.com>
Date: Thu, 20 Aug 2015 14:11:12 +0200
Subject: [PATCH] testng: allow to pass runner parameters (IDEA-144044)

---
 .../execution/JavaTestFrameworkRunnableState.java     |  7 ++++++-
 .../testng/configuration/TestNGRunnableState.java     | 11 +++++++++++
 2 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/java/execution/impl/src/com/intellij/execution/JavaTestFrameworkRunnableState.java b/java/execution/impl/src/com/intellij/execution/JavaTestFrameworkRunnableState.java
index 3382d0bf1933..4bb082b1d29c 100644
--- a/java/execution/impl/src/com/intellij/execution/JavaTestFrameworkRunnableState.java
+++ b/java/execution/impl/src/com/intellij/execution/JavaTestFrameworkRunnableState.java
@@ -62,6 +62,7 @@ import org.jetbrains.jps.model.serialization.PathMacroUtil;
 import java.io.*;
 import java.net.InetAddress;
 import java.net.ServerSocket;
+import java.util.Collections;
 import java.util.List;
 import java.util.Locale;
 import java.util.Map;
@@ -195,12 +196,16 @@ public abstract class JavaTestFrameworkRunnableState<T extends ModuleBasedConfig
     }
 
     if (!StringUtil.isEmptyOrSpaces(parameters)) {
-      javaParameters.getProgramParametersList().add("@name" + parameters);
+      javaParameters.getProgramParametersList().addAll(getNamedParams(parameters));
     }
 
     return javaParameters;
   }
 
+  protected List<String> getNamedParams(String parameters) {
+    return Collections.singletonList("@name" + parameters);
+  }
+
   private ServerSocket myForkSocket = null;
 
   @Nullable
diff --git a/plugins/testng/src/com/theoryinpractice/testng/configuration/TestNGRunnableState.java b/plugins/testng/src/com/theoryinpractice/testng/configuration/TestNGRunnableState.java
index 409084ce294c..0bc127bf2344 100644
--- a/plugins/testng/src/com/theoryinpractice/testng/configuration/TestNGRunnableState.java
+++ b/plugins/testng/src/com/theoryinpractice/testng/configuration/TestNGRunnableState.java
@@ -217,6 +217,17 @@ public class TestNGRunnableState extends JavaTestFrameworkRunnableState<TestNGCo
     return javaParameters;
   }
 
+  @Override
+  protected List<String> getNamedParams(String parameters) {
+    try {
+      Integer.parseInt(parameters);
+      return super.getNamedParams(parameters);
+    }
+    catch (NumberFormatException e) {
+      return Arrays.asList(parameters.split(" "));
+    }
+  }
+
   @NotNull
   @Override
   protected String getForkMode() {