1 files changed, 6 insertions, 1 deletions
diff --git a/plugins/testng/src/com/theoryinpractice/testng/ui/TestNGResults.java b/plugins/testng/src/com/theoryinpractice/testng/ui/TestNGResults.java
index 442e03440ac5..78f980aea83d 100644
--- a/plugins/testng/src/com/theoryinpractice/testng/ui/TestNGResults.java
+++ b/plugins/testng/src/com/theoryinpractice/testng/ui/TestNGResults.java
@@ -378,7 +378,7 @@ public class TestNGResults extends TestResultsPanel implements TestFrameworkRunn
     rootNode.setStarted(true);
   }
 
-  public void finish() {
+  public void finish(final boolean started) {
     if (start > 0) {
       end = System.currentTimeMillis();
     }
@@ -409,6 +409,7 @@ public class TestNGResults extends TestResultsPanel implements TestFrameworkRunn
           }
         }
         tree.repaint();
+        TestsUIUtil.notifyByBalloon(project, started, rootNode, getProperties(), "in " + getTime());
       }
     });
   }
@@ -466,6 +467,10 @@ public class TestNGResults extends TestResultsPanel implements TestFrameworkRunn
     this.failedToStart = failedToStart;
   }
 
+  public boolean hasFinishedTests() {
+    return count > 0;
+  }
+
   private class OpenSourceSelectionListener implements TreeSelectionListener {
 
     public void valueChanged(TreeSelectionEvent e) {