From e3cd6a55af697c8f7ced11a7ade116565dbfeb50 Mon Sep 17 00:00:00 2001
From: Kirill Kalishev <kirill.kalishev@jetbrains.com>
Date: Thu, 20 May 2010 14:52:57 +0400
Subject: [PATCH] junit test runner supports app icon

---
 .../execution/junit2/ui/TestProgress.java         | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/plugins/junit/src/com/intellij/execution/junit2/ui/TestProgress.java b/plugins/junit/src/com/intellij/execution/junit2/ui/TestProgress.java
index 601cffecb9fe..c8d6a308704b 100644
--- a/plugins/junit/src/com/intellij/execution/junit2/ui/TestProgress.java
+++ b/plugins/junit/src/com/intellij/execution/junit2/ui/TestProgress.java
@@ -130,12 +130,17 @@ public class TestProgress extends DefaultBoundedRangeModel implements Disposable
     super.setValue(n);
     AppIcon icon = AppIcon.getInstance();
     if (n < getMaximum()) {
-      icon.setProgress(null, TESTS, AppIconScheme.Progress.TESTS, (double)n / (double)getMaximum(), myProblemsCounter == 0);
+      if (icon.setProgress(null, TESTS, AppIconScheme.Progress.TESTS, (double)n / (double)getMaximum(), myProblemsCounter == 0)) {
+        if (myProblemsCounter > 0) {
+          icon.setBadge(String.valueOf(myProblemsCounter));
+        }
+      }
     } else {
-      icon.hideProgress(null, TESTS);
-      if (myProblemsCounter > 0) {
-        icon.setBadge(String.valueOf(myProblemsCounter));
-        icon.requestAttention(false);
+      if (icon.hideProgress(null, TESTS)) {
+        if (myProblemsCounter > 0) {
+          icon.setBadge(String.valueOf(myProblemsCounter));
+          icon.requestAttention(true);
+        }
       }
     }
   }