From dde38152a1f16845c0d3eb84ecbd022e65951657 Mon Sep 17 00:00:00 2001
From: Konstantin Bulenkov <kb@jetbrains.com>
Date: Fri, 18 Aug 2017 23:16:50 +0200
Subject: [PATCH] Tests for tree smart selection

---
 .../util/treeView/TreeSmartSelectTest.java    | 78 +++++++++++++++++++
 1 file changed, 78 insertions(+)
 create mode 100644 platform/platform-tests/testSrc/com/intellij/ide/util/treeView/TreeSmartSelectTest.java

diff --git a/platform/platform-tests/testSrc/com/intellij/ide/util/treeView/TreeSmartSelectTest.java b/platform/platform-tests/testSrc/com/intellij/ide/util/treeView/TreeSmartSelectTest.java
new file mode 100644
index 000000000000..b63bb3c902fe
--- /dev/null
+++ b/platform/platform-tests/testSrc/com/intellij/ide/util/treeView/TreeSmartSelectTest.java
@@ -0,0 +1,78 @@
+/*
+ * Copyright 2000-2017 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.intellij.ide.util.treeView;
+
+import com.intellij.ui.tree.TreeSmartSelectProvider;
+import com.intellij.util.ui.tree.TreeUtil;
+
+/**
+ * @author Konstantin Bulenkov
+ */
+public class TreeSmartSelectTest extends AbstractTreeBuilderTest {
+  TreeSmartSelectProvider myProvider = new TreeSmartSelectProvider();
+  public TreeSmartSelectTest() {
+    super(true);
+  }
+
+  public void testSelectionIncrease() throws Exception {
+    doAndWaitForBuilder(() -> {
+                          Node intellij = myRoot.addChild("com").addChild("intellij");
+                          intellij.addChild("a");
+                          intellij.addChild("b");
+                          Node c = intellij.addChild("c");
+                          TreeUtil.expandAll(myTree);
+                          getBuilder().select(c.myElement);
+                        });
+    assertTree("-/\n" +
+               " -com\n" +
+               "  -intellij\n" +
+               "   a\n" +
+               "   b\n" +
+               "   [c]\n");
+
+    myProvider.increaseSelection(myTree);
+    assertTree("-/\n" +
+               " -com\n" +
+               "  -intellij\n" +
+               "   [a]\n" +
+               "   [b]\n" +
+               "   [c]\n");
+
+    myProvider.increaseSelection(myTree);
+    assertTree("-/\n" +
+               " -com\n" +
+               "  -[intellij]\n" +
+               "   [a]\n" +
+               "   [b]\n" +
+               "   [c]\n");
+
+    myProvider.increaseSelection(myTree);
+    assertTree("-/\n" +
+               " -[com]\n" +
+               "  -[intellij]\n" +
+               "   [a]\n" +
+               "   [b]\n" +
+               "   [c]\n");
+
+    myProvider.increaseSelection(myTree);
+    assertTree("-/\n" +
+               " -[com]\n" +
+               "  -[intellij]\n" +
+               "   [a]\n" +
+               "   [b]\n" +
+               "   [c]\n");
+  }
+}