From 60aafa307f2d2bfc84a49abb1755449ac5a748d9 Mon Sep 17 00:00:00 2001
From: Kirill Kalishev <kirill.kalishev@jetbrains.com>
Date: Tue, 13 Sep 2011 08:49:25 +0400
Subject: [PATCH] TreeUiTests fixed

---
 .../core-impl/src/com/intellij/mock/MockApplication.java  | 8 ++++----
 .../com/intellij/ide/util/treeView/AbstractTreeUi.java    | 4 ++++
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/platform/core-impl/src/com/intellij/mock/MockApplication.java b/platform/core-impl/src/com/intellij/mock/MockApplication.java
index 0f47da2e6dc0..813d514eea6d 100644
--- a/platform/core-impl/src/com/intellij/mock/MockApplication.java
+++ b/platform/core-impl/src/com/intellij/mock/MockApplication.java
@@ -210,24 +210,24 @@ public class MockApplication extends MockComponentManager implements Application
   @NotNull
   @Override
   public ModalityState getCurrentModalityState() {
-    throw new UnsupportedOperationException();
+    return getNoneModalityState();
   }
 
   @Override
   public ModalityState getAnyModalityState() {
-    throw new UnsupportedOperationException();
+    return getNoneModalityState();
   }
 
   @NotNull
   @Override
   public ModalityState getModalityStateForComponent(@NotNull Component c) {
-    throw new UnsupportedOperationException();
+    return getNoneModalityState();
   }
 
   @NotNull
   @Override
   public ModalityState getDefaultModalityState() {
-    throw new UnsupportedOperationException();
+    return getNoneModalityState();
   }
 
   @Override
diff --git a/platform/platform-api/src/com/intellij/ide/util/treeView/AbstractTreeUi.java b/platform/platform-api/src/com/intellij/ide/util/treeView/AbstractTreeUi.java
index 6ad45058a876..6453af754439 100644
--- a/platform/platform-api/src/com/intellij/ide/util/treeView/AbstractTreeUi.java
+++ b/platform/platform-api/src/com/intellij/ide/util/treeView/AbstractTreeUi.java
@@ -1330,6 +1330,10 @@ public class AbstractTreeUi {
                   }
                 }
 
+                if (!canInitiateNewActivity()) {
+                  throw new ProcessCanceledException();
+                }
+
                 final Object element = getElementFor(node);
                 addNodeAction(element, new NodeAction() {
                   public void onReady(final DefaultMutableTreeNode node) {