Tests for tree smart selection

2026-09-27 10:03:11 +07:00 · 2017-08-18 23:17:48 +02:00
parent 454ea28d1e
commit dde38152a1
1 changed files with 78 additions and 0 deletions
@@ -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");
+  }
+}