From 75b1558d091d72eed42c006be72fc165eac654a8 Mon Sep 17 00:00:00 2001
From: Anna Kozlova <anna.kozlova@jetbrains.com>
Date: Mon, 13 Apr 2015 14:22:07 +0200
Subject: [PATCH] forked mode for sm runner

---
 .../rt/execution/junit/JUnitForkedStarter.java      | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/plugins/junit_rt/src/com/intellij/rt/execution/junit/JUnitForkedStarter.java b/plugins/junit_rt/src/com/intellij/rt/execution/junit/JUnitForkedStarter.java
index 2ab8cd01a73c..27116276ee97 100644
--- a/plugins/junit_rt/src/com/intellij/rt/execution/junit/JUnitForkedStarter.java
+++ b/plugins/junit_rt/src/com/intellij/rt/execution/junit/JUnitForkedStarter.java
@@ -16,6 +16,7 @@
 package com.intellij.rt.execution.junit;
 
 import com.intellij.rt.execution.CommandLineWrapper;
+import com.intellij.rt.execution.junit.segments.OutputObjectRegistry;
 import com.intellij.rt.execution.junit.segments.SegmentedOutputStream;
 
 import java.io.*;
@@ -93,7 +94,7 @@ public class JUnitForkedStarter {
     final Object description = testRunner.getTestToStart(args, params);
     if (description == null) return -1;
 
-    TreeSender.sendTree(testRunner, description, true);
+    TreeSender.sendTree(testRunner, description, !JUnitStarter.SM_RUNNER);
 
     long time = System.currentTimeMillis();
 
@@ -129,9 +130,10 @@ public class JUnitForkedStarter {
               File tempFile = File.createTempFile("idea_junit", ".tmp");
               tempFile.deleteOnExit();
               JUnitStarter.printClassesList(classNames, packageName + ", working directory: \'" + workingDir + "\'", "", tempFile);
+              final OutputObjectRegistry registry = testRunner.getRegistry();
+              final String startIndex = String.valueOf(registry != null ? registry.getKnownObject(rootDescriptor) - 1 : -1);
               childResult =
-                runChild(isJUnit4, listeners, out, err, parameters, "@" + tempFile.getAbsolutePath(), dir,
-                         String.valueOf(testRunner.getRegistry().getKnownObject(rootDescriptor) - 1), classpath, dynamicClasspath);
+                runChild(isJUnit4, listeners, out, err, parameters, "@" + tempFile.getAbsolutePath(), dir, startIndex, classpath, dynamicClasspath);
             } else {
               final List children = new ArrayList(testRunner.getChildTests(description));
               for (Iterator iterator = children.iterator(); iterator.hasNext(); ) {
@@ -155,7 +157,7 @@ public class JUnitForkedStarter {
     }
 
     time = System.currentTimeMillis() - time;
-    new TimeSender(testRunner.getRegistry()).printHeader(time);
+    if (!JUnitStarter.SM_RUNNER) new TimeSender(testRunner.getRegistry()).printHeader(time);
     return result;
   }
 
@@ -188,7 +190,8 @@ public class JUnitForkedStarter {
       final List childTests = testRunner.getChildTests(child);
       final int childResult;
       if (childTests.isEmpty() || !forkTillMethod) {
-        final int startIndex = testRunner.getRegistry().getKnownObject(child);
+        final OutputObjectRegistry registry = testRunner.getRegistry();
+        final int startIndex = registry != null ? registry.getKnownObject(child) : -1;
         childResult =
           runChild(isJUnit4, listeners, out, err, parameters, testRunner.getStartDescription(child), workingDir, String.valueOf(startIndex), classpath, dynamicClasspath);
       }