remote debug arquillian testNG tests

java/execution/impl/src/com/intellij/execution/JavaTestFrameworkRunnableState.java

@@ -68,12 +68,32 @@ import java.util.List;
 import java.util.Locale;
 import java.util.Map;
 
-public abstract class JavaTestFrameworkRunnableState<T extends ModuleBasedConfiguration<JavaRunConfigurationModule> & CommonJavaRunConfigurationParameters & SMRunnerConsolePropertiesProvider> extends JavaCommandLineState {
+public abstract class JavaTestFrameworkRunnableState<T extends
+  ModuleBasedConfiguration<JavaRunConfigurationModule>
+  & CommonJavaRunConfigurationParameters
+  & SMRunnerConsolePropertiesProvider> extends JavaCommandLineState implements RemoteConnectionCreator {
   private static final Logger LOG = Logger.getInstance("#" + JavaTestFrameworkRunnableState.class.getName());
   protected ServerSocket myServerSocket;
   protected File myTempFile;
   protected File myWorkingDirsFile = null;
 
+  private RemoteConnectionCreator remoteConnectionCreator;
+
+  public void setRemoteConnectionCreator(RemoteConnectionCreator remoteConnectionCreator) {
+    this.remoteConnectionCreator = remoteConnectionCreator;
+  }
+
+  @Nullable
+  @Override
+  public RemoteConnection createRemoteConnection(ExecutionEnvironment environment) {
+    return remoteConnectionCreator == null ? null : remoteConnectionCreator.createRemoteConnection(environment);
+  }
+
+  @Override
+  public boolean isPollConnection() {
+    return remoteConnectionCreator != null && remoteConnectionCreator.isPollConnection();
+  }
+
   public JavaTestFrameworkRunnableState(ExecutionEnvironment environment) {
     super(environment);
   }

java/execution/openapi/src/com/intellij/execution/configurations/RemoteConnectionCreator.java

@@ -16,8 +16,10 @@
 package com.intellij.execution.configurations;
 
 import com.intellij.execution.runners.ExecutionEnvironment;
+import org.jetbrains.annotations.Nullable;
 
 public interface RemoteConnectionCreator {
+  @Nullable
   RemoteConnection createRemoteConnection(ExecutionEnvironment environment);
 
   boolean isPollConnection();