TreeDemoTest.java revision 13978:1993af50385d
1/* 2 * Copyright (c) 2010, 2016, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. 8 * 9 * This code is distributed in the hope that it will be useful, but WITHOUT 10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 12 * version 2 for more details (a copy is included in the LICENSE file that 13 * accompanied this code). 14 * 15 * You should have received a copy of the GNU General Public License version 16 * 2 along with this work; if not, write to the Free Software Foundation, 17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 18 * 19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 20 * or visit www.oracle.com if you need additional information or have any 21 * questions. 22 */ 23 24import com.sun.swingset3.demos.tree.TreeDemo; 25import static com.sun.swingset3.demos.tree.TreeDemo.DEMO_TITLE; 26import javax.swing.tree.TreePath; 27import static org.testng.AssertJUnit.*; 28import org.testng.annotations.Test; 29import org.netbeans.jemmy.ClassReference; 30import org.netbeans.jemmy.operators.JFrameOperator; 31import org.netbeans.jemmy.operators.JTreeOperator; 32import static org.jemmy2ext.JemmyExt.captureDebugInfoOnFail; 33 34/* 35 * @test 36 * @key headful 37 * @summary Verifies SwingSet3 TreeDemo by expanding all collapsed nodes in the 38 * tree and then collapsing all the expanded nodes in the tree. It 39 * verifies the number of nodes expanded, number of nodes collapsed and 40 * number of rows in the tree in the begininng, after expanding and 41 * after collapsing the nodes. It also checks that the tree grows 42 * vertically (as ScrollPane allows it). 43 * 44 * @library /sanity/client/lib/jemmy/src 45 * @library /sanity/client/lib/Jemmy2Ext/src 46 * @library /sanity/client/lib/SwingSet3/src 47 * @build org.jemmy2ext.JemmyExt 48 * @build com.sun.swingset3.demos.tree.TreeDemo 49 * @run testng TreeDemoTest 50 */ 51public class TreeDemoTest { 52 53 @Test 54 public void test() throws Exception { 55 captureDebugInfoOnFail(() -> { 56 new ClassReference(TreeDemo.class.getCanonicalName()).startApplication(); 57 58 JFrameOperator frame = new JFrameOperator(DEMO_TITLE); 59 60 JTreeOperator tree = new JTreeOperator(frame); 61 62 assertEquals("Initial number of rows in the tree", 4, tree.getRowCount()); 63 64 int initialTreeHeight = tree.getHeight(); 65 66 // expand all nodes 67 int expandsCount = 0; 68 for (int i = 0; i < tree.getRowCount(); i++) { 69 TreePath tp = tree.getPathForRow(i); 70 if (tree.getChildCount(tp) > 0 && !tree.isExpanded(tp)) { 71 tree.expandRow(i); 72 expandsCount++; 73 } 74 } 75 76 assertEquals("Number of rows expanded", 75, expandsCount); 77 assertEquals("Number of rows in the tree after expanding all of them", 78 616, tree.getRowCount()); 79 80 int expandedTreeHeight = tree.getHeight(); 81 assertTrue("Expanded tree height has increased, current " 82 + expandedTreeHeight + " > initial " + initialTreeHeight, 83 expandedTreeHeight > initialTreeHeight); 84 85 // collapse all nodes 86 int collapsesCount = 0; 87 for (int i = tree.getRowCount() - 1; i >= 0; i--) { 88 TreePath tp = tree.getPathForRow(i); 89 if (tree.getChildCount(tp) > 0 && tree.isExpanded(tp)) { 90 tree.collapseRow(i); 91 collapsesCount++; 92 } 93 } 94 95 assertEquals("Number of rows collapsed", 76, collapsesCount); 96 assertEquals("Number of rows in the tree after collapsing all of them", 97 1, tree.getRowCount()); 98 99 int collapsedTreeHeight = tree.getHeight(); 100 assertTrue("Collpased tree height is not longer than initial, " 101 + "current " + collapsedTreeHeight + " <= initial " 102 + initialTreeHeight, 103 collapsedTreeHeight <= initialTreeHeight); 104 105 }); 106 } 107 108} 109