diff options
Diffstat (limited to 'platform/testRunner/src/com/intellij/execution/testframework/TestTreeView.java')
-rw-r--r-- | platform/testRunner/src/com/intellij/execution/testframework/TestTreeView.java | 9 |
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 |