From cdf4412adc77b2d6c5de7180f45f871453fd298e Mon Sep 17 00:00:00 2001
From: Anna Kozlova <anna.kozlova@jetbrains.com>
Date: Wed, 9 Apr 2014 11:57:35 +0200
Subject: [PATCH] provide project for TestStatusListener

---
 .../testframework/TestStatusListener.java     | 19 ++++++++++++++++---
 .../junit2/ui/TestsPacketsReceiver.java       |  2 +-
 .../testng/ui/TestNGResults.java              |  2 +-
 3 files changed, 18 insertions(+), 5 deletions(-)

diff --git a/platform/testRunner/src/com/intellij/execution/testframework/TestStatusListener.java b/platform/testRunner/src/com/intellij/execution/testframework/TestStatusListener.java
index 3f859643ba25..08130de9b495 100644
--- a/platform/testRunner/src/com/intellij/execution/testframework/TestStatusListener.java
+++ b/platform/testRunner/src/com/intellij/execution/testframework/TestStatusListener.java
@@ -17,15 +17,28 @@ package com.intellij.execution.testframework;
 
 import com.intellij.openapi.extensions.ExtensionPointName;
 import com.intellij.openapi.extensions.Extensions;
+import com.intellij.openapi.project.Project;
 
 public abstract class TestStatusListener {
   public static final ExtensionPointName<TestStatusListener> EP_NAME = ExtensionPointName.create("com.intellij.testStatusListener");
 
   public abstract void testSuiteFinished(AbstractTestProxy root);
 
+  public void testSuiteFinished(AbstractTestProxy root, Project project) {
+    testSuiteFinished(root);
+  }
+
+  @Deprecated
+  @SuppressWarnings("UnusedDeclaration")
   public static void notifySuiteFinished(AbstractTestProxy root) {
-      for (TestStatusListener statusListener : Extensions.getExtensions(EP_NAME)) {
-        statusListener.testSuiteFinished(root);
-      }
+    for (TestStatusListener statusListener : Extensions.getExtensions(EP_NAME)) {
+      statusListener.testSuiteFinished(root);
     }
+  }
+
+  public static void notifySuiteFinished(AbstractTestProxy root, Project project) {
+    for (TestStatusListener statusListener : Extensions.getExtensions(EP_NAME)) {
+      statusListener.testSuiteFinished(root, project);
+    }
+  }
 }
diff --git a/plugins/junit/src/com/intellij/execution/junit2/ui/TestsPacketsReceiver.java b/plugins/junit/src/com/intellij/execution/junit2/ui/TestsPacketsReceiver.java
index af330c93d747..95303c9a1bc2 100644
--- a/plugins/junit/src/com/intellij/execution/junit2/ui/TestsPacketsReceiver.java
+++ b/plugins/junit/src/com/intellij/execution/junit2/ui/TestsPacketsReceiver.java
@@ -202,7 +202,7 @@ public class TestsPacketsReceiver implements OutputPacketProcessor, Disposable {
     final JUnitRunningModel model = getModel();
     if (model != null) {
       model.getNotifier().fireRunnerStateChanged(new CompletionEvent(true, reader.readInt()));
-      TestStatusListener.notifySuiteFinished(model.getRoot());
+      TestStatusListener.notifySuiteFinished(model.getRoot(), model.getProject());
       terminateStillRunning(model);
     }
   }
diff --git a/plugins/testng/src/com/theoryinpractice/testng/ui/TestNGResults.java b/plugins/testng/src/com/theoryinpractice/testng/ui/TestNGResults.java
index 3eaf5e6271d5..442e03440ac5 100644
--- a/plugins/testng/src/com/theoryinpractice/testng/ui/TestNGResults.java
+++ b/plugins/testng/src/com/theoryinpractice/testng/ui/TestNGResults.java
@@ -398,7 +398,7 @@ public class TestNGResults extends TestResultsPanel implements TestFrameworkRunn
           myStatusLine.setStatusColor(ColorProgressBar.GREEN);
         }
         rootNode.setInProgress(false);
-        TestStatusListener.notifySuiteFinished(rootNode);
+        TestStatusListener.notifySuiteFinished(rootNode, project);
         if (TestNGConsoleProperties.SELECT_FIRST_DEFECT.value(myProperties)) {
           selectTest(rootNode.getFirstDefect());
         }