From 2a6b5be74b9bd86cad34897dd090a2b57ae6e558 Mon Sep 17 00:00:00 2001
From: Sergey Ignatov <sergey.ignatov@jetbrains.com>
Date: Sun, 10 Jun 2018 06:47:43 +0300
Subject: [PATCH] test discovery: collapse the whole tree

---
 .../execution/testDiscovery/actions/DiscoveredTestsTree.java | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/java/execution/impl/src/com/intellij/execution/testDiscovery/actions/DiscoveredTestsTree.java b/java/execution/impl/src/com/intellij/execution/testDiscovery/actions/DiscoveredTestsTree.java
index 8cbffef6b2cc..3fc8b63f9730 100644
--- a/java/execution/impl/src/com/intellij/execution/testDiscovery/actions/DiscoveredTestsTree.java
+++ b/java/execution/impl/src/com/intellij/execution/testDiscovery/actions/DiscoveredTestsTree.java
@@ -89,7 +89,10 @@ class DiscoveredTestsTree extends Tree implements DataProvider {
       protected void process(TreeModelEvent event, EventType type) {
         if (!myAlreadyDone && myModel.getTestCount() != 0) {
           myAlreadyDone = true;
-          EdtInvocationManager.getInstance().invokeLater(() -> TreeUtil.selectFirstNode(DiscoveredTestsTree.this));
+          EdtInvocationManager.getInstance().invokeLater(() -> {
+            TreeUtil.collapseAll(DiscoveredTestsTree.this, 0);
+            TreeUtil.selectFirstNode(DiscoveredTestsTree.this);
+          });
         }
       }
     });