From 19d1d5fb53817e4737206e3a08565a1639271aef Mon Sep 17 00:00:00 2001
From: Anna Kozlova <anna.kozlova@jetbrains.com>
Date: Fri, 10 Apr 2015 15:25:58 +0200
Subject: [PATCH] java sm runner: ensure temp files are deleted after exist

---
 .../JavaTestFrameworkRunnableState.java       |  7 +++++
 .../intellij/execution/junit/TestObject.java  | 27 ++++++++++++-------
 2 files changed, 25 insertions(+), 9 deletions(-)

diff --git a/java/execution/impl/src/com/intellij/execution/JavaTestFrameworkRunnableState.java b/java/execution/impl/src/com/intellij/execution/JavaTestFrameworkRunnableState.java
index a7641e2a4c4f..64d6caca0408 100644
--- a/java/execution/impl/src/com/intellij/execution/JavaTestFrameworkRunnableState.java
+++ b/java/execution/impl/src/com/intellij/execution/JavaTestFrameworkRunnableState.java
@@ -111,6 +111,8 @@ public abstract class JavaTestFrameworkRunnableState<T extends ModuleBasedConfig
                 !ResetConfigurationModuleAdapter.tryWithAnotherModule(getConfiguration(), testConsoleProperties.isDebug())) {
               TestsUIUtil.notifyByBalloon(testConsoleProperties.getProject(), viewer.hasTestSuites(), viewer.getRoot(), testConsoleProperties, null);
             }
+
+            deleteTempFiles();
           }
         };
         SwingUtilities.invokeLater(runnable);
@@ -214,5 +216,10 @@ public abstract class JavaTestFrameworkRunnableState<T extends ModuleBasedConfig
     }
   }
 
+  protected void deleteTempFiles() {
+    if (myTempFile != null) {
+      FileUtil.delete(myTempFile);
+    }
+  }
 
 }
diff --git a/plugins/junit/src/com/intellij/execution/junit/TestObject.java b/plugins/junit/src/com/intellij/execution/junit/TestObject.java
index d41883cbf444..6fd9f56b952a 100644
--- a/plugins/junit/src/com/intellij/execution/junit/TestObject.java
+++ b/plugins/junit/src/com/intellij/execution/junit/TestObject.java
@@ -17,7 +17,10 @@
 package com.intellij.execution.junit;
 
 import com.intellij.execution.*;
-import com.intellij.execution.configurations.*;
+import com.intellij.execution.configurations.JavaParameters;
+import com.intellij.execution.configurations.ParametersList;
+import com.intellij.execution.configurations.RunnerSettings;
+import com.intellij.execution.configurations.RuntimeConfigurationException;
 import com.intellij.execution.junit2.TestProxy;
 import com.intellij.execution.junit2.segments.DeferredActionsQueue;
 import com.intellij.execution.junit2.segments.DeferredActionsQueueImpl;
@@ -55,7 +58,6 @@ import com.intellij.openapi.util.Disposer;
 import com.intellij.openapi.util.Getter;
 import com.intellij.openapi.util.Key;
 import com.intellij.openapi.util.io.FileUtil;
-import com.intellij.openapi.util.registry.Registry;
 import com.intellij.openapi.util.text.StringUtil;
 import com.intellij.openapi.util.text.StringUtilRt;
 import com.intellij.openapi.vfs.CharsetToolkit;
@@ -87,7 +89,7 @@ public abstract class TestObject extends JavaTestFrameworkRunnableState<JUnitCon
 
   private final JUnitConfiguration myConfiguration;
   protected File myWorkingDirsFile = null;
-  public File myListenersFile;
+  protected File myListenersFile;
 
   public static TestObject fromString(final String id,
                                       final JUnitConfiguration configuration,
@@ -235,12 +237,7 @@ public abstract class TestObject extends JavaTestFrameworkRunnableState<JUnitCon
       @Override
       public void processTerminated(ProcessEvent event) {
         handler.removeProcessListener(this);
-        if (myTempFile != null) {
-          FileUtil.delete(myTempFile);
-        }
-        if (myListenersFile != null) {
-          FileUtil.delete(myListenersFile);
-        }
+        deleteTempFiles();
         final Runnable runnable = new Runnable() {
           @Override
           public void run() {
@@ -506,6 +503,18 @@ public abstract class TestObject extends JavaTestFrameworkRunnableState<JUnitCon
     }
   }
 
+  @Override
+  protected void deleteTempFiles() {
+    super.deleteTempFiles();
+    if (myListenersFile != null) {
+      FileUtil.delete(myListenersFile);
+    }
+    
+    if (myWorkingDirsFile != null) {
+      FileUtil.delete(myWorkingDirsFile);
+    }
+  }
+
   public void clear() {
   }