1 files changed, 7 insertions, 2 deletions
diff --git a/platform/testRunner/src/com/intellij/execution/testframework/TestTreeView.java b/platform/testRunner/src/com/intellij/execution/testframework/TestTreeView.java
index 839568b4c2a9..5d1d2b346a85 100644
--- a/platform/testRunner/src/com/intellij/execution/testframework/TestTreeView.java
+++ b/platform/testRunner/src/com/intellij/execution/testframework/TestTreeView.java
@@ -46,6 +46,8 @@ import java.util.ArrayList;
 import java.util.List;
 
 public abstract class TestTreeView extends Tree implements DataProvider, CopyProvider {
+  public static final DataKey<TestFrameworkRunningModel> MODEL_DATA_KEY = DataKey.create("testFrameworkModel.dataId");
+
   private TestFrameworkRunningModel myModel;
 
   protected abstract TreeCellRenderer getRenderer(TestConsoleProperties properties);
@@ -127,7 +129,11 @@ public abstract class TestTreeView extends Tree implements DataProvider, CopyPro
         return locations.isEmpty() ? null : locations.toArray(new Location[locations.size()]);
       }
     }
-    
+
+    if (MODEL_DATA_KEY.is(dataId)) {
+      return myModel;
+    }
+
     final TreePath selectionPath = getSelectionPath();
     if (selectionPath == null) return null;
     final AbstractTestProxy testProxy = getSelectedTest(selectionPath);
@@ -162,7 +168,6 @@ public abstract class TestTreeView extends Tree implements DataProvider, CopyPro
     });
     TreeUtil.installActions(this);
     PopupHandler.installPopupHandler(this, IdeActions.GROUP_TESTTREE_POPUP, ActionPlaces.TESTTREE_VIEW_POPUP);
-    ViewAssertEqualsDiffAction.registerShortcut(this);
   }
 
   @JdkConstants.TreeSelectionMode