From 127a8db4007548db5fd29cde9c14332c9dfc32ef Mon Sep 17 00:00:00 2001
From: Anna Kozlova <anna.kozlova@jetbrains.com>
Date: Mon, 13 Apr 2015 16:03:53 +0200
Subject: [PATCH] ensure ServiceMessageTypes is in classpath if different
 classpath is used per run

---
 .../intellij/execution/JavaTestFrameworkRunnableState.java  | 6 +++++-
 .../junit/src/com/intellij/execution/junit/TestObject.java  | 4 ++++
 2 files changed, 9 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 2b909ea2def4..47341bbfa346 100644
--- a/java/execution/impl/src/com/intellij/execution/JavaTestFrameworkRunnableState.java
+++ b/java/execution/impl/src/com/intellij/execution/JavaTestFrameworkRunnableState.java
@@ -84,7 +84,7 @@ public abstract class JavaTestFrameworkRunnableState<T extends ModuleBasedConfig
   }
 
   protected ExecutionResult startSMRunner(Executor executor) throws ExecutionException {
-    if (!Registry.is(getFrameworkId() + "_sm_runner")) {
+    if (!isSmRunnerUsed()) {
       return null;
     }
     getJavaParameters().getVMParametersList().addProperty("idea." + getFrameworkId() + ".sm_runner");
@@ -137,6 +137,10 @@ public abstract class JavaTestFrameworkRunnableState<T extends ModuleBasedConfig
     return result;
   }
 
+  protected boolean isSmRunnerUsed() {
+    return Registry.is(getFrameworkId() + "_sm_runner");
+  }
+
   @Override
   protected JavaParameters createJavaParameters() throws ExecutionException {
     final JavaParameters javaParameters = new JavaParameters();
diff --git a/plugins/junit/src/com/intellij/execution/junit/TestObject.java b/plugins/junit/src/com/intellij/execution/junit/TestObject.java
index fe800b546ff9..e5fc301a0244 100644
--- a/plugins/junit/src/com/intellij/execution/junit/TestObject.java
+++ b/plugins/junit/src/com/intellij/execution/junit/TestObject.java
@@ -72,6 +72,7 @@ import com.intellij.rt.execution.junit.RepeatCount;
 import com.intellij.util.Function;
 import com.intellij.util.PathUtil;
 import com.intellij.util.ui.UIUtil;
+import jetbrains.buildServer.messages.serviceMessages.ServiceMessageTypes;
 import org.jetbrains.annotations.NonNls;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.jps.model.serialization.PathMacroUtil;
@@ -467,6 +468,9 @@ public abstract class TestObject extends JavaTestFrameworkRunnableState<JUnitCon
             if (classpath == null) {
               final JavaParameters parameters = new JavaParameters();
               configureAdditionalClasspath(parameters);
+              if (isSmRunnerUsed()) {
+                parameters.getClassPath().add(PathUtil.getJarPathForClass(ServiceMessageTypes.class));
+              }
               JavaParametersUtil.configureModule(module, parameters, JavaParameters.JDK_AND_CLASSES_AND_TESTS,
                                                  getConfiguration().isAlternativeJrePathEnabled() ? getConfiguration()
                                                    .getAlternativeJrePath() : null);