130 lines
4.6 KiB
Java
130 lines
4.6 KiB
Java
/*
|
|
* Copyright (c) 2010, 2016, Oracle and/or its affiliates. All rights reserved.
|
|
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
|
|
*
|
|
* This code is free software; you can redistribute it and/or modify it
|
|
* under the terms of the GNU General Public License version 2 only, as
|
|
* published by the Free Software Foundation.
|
|
*
|
|
* This code is distributed in the hope that it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
|
* version 2 for more details (a copy is included in the LICENSE file that
|
|
* accompanied this code).
|
|
*
|
|
* You should have received a copy of the GNU General Public License version
|
|
* 2 along with this work; if not, write to the Free Software Foundation,
|
|
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*
|
|
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
|
|
* or visit www.oracle.com if you need additional information or have any
|
|
* questions.
|
|
*/
|
|
|
|
import com.sun.swingset3.demos.tree.TreeDemo;
|
|
import static com.sun.swingset3.demos.tree.TreeDemo.DEMO_TITLE;
|
|
|
|
import java.awt.Component;
|
|
import javax.swing.tree.TreePath;
|
|
|
|
import org.jtregext.GuiTestListener;
|
|
|
|
import org.netbeans.jemmy.ClassReference;
|
|
import org.netbeans.jemmy.ComponentChooser;
|
|
import org.netbeans.jemmy.operators.JFrameOperator;
|
|
import org.netbeans.jemmy.operators.JTreeOperator;
|
|
|
|
import org.testng.annotations.Listeners;
|
|
import org.testng.annotations.Test;
|
|
import static org.testng.AssertJUnit.*;
|
|
|
|
/*
|
|
* @test
|
|
* @key headful
|
|
* @summary Verifies SwingSet3 TreeDemo by expanding all collapsed nodes in the
|
|
* tree and then collapsing all the expanded nodes in the tree. It
|
|
* verifies the number of nodes expanded, number of nodes collapsed and
|
|
* number of rows in the tree in the begininng, after expanding and
|
|
* after collapsing the nodes. It also checks that the tree grows
|
|
* vertically (as ScrollPane allows it).
|
|
*
|
|
* @library /sanity/client/lib/jemmy/src
|
|
* @library /sanity/client/lib/Extensions/src
|
|
* @library /sanity/client/lib/SwingSet3/src
|
|
* @modules java.desktop
|
|
* java.logging
|
|
* @build org.jemmy2ext.JemmyExt
|
|
* @build com.sun.swingset3.demos.tree.TreeDemo
|
|
* @run testng TreeDemoTest
|
|
*/
|
|
@Listeners(GuiTestListener.class)
|
|
public class TreeDemoTest {
|
|
|
|
private static final int NODES_TO_EXPAND = 75;
|
|
private static final int NODES_TOTAL = 616;
|
|
|
|
private void waitRowCount(JTreeOperator tree, int count) {
|
|
tree.waitState(new ComponentChooser() {
|
|
public boolean checkComponent(Component comp) {
|
|
return tree.getRowCount() == count;
|
|
}
|
|
public String getDescription() {
|
|
return "A tree to have " + count + " rows";
|
|
}
|
|
});
|
|
}
|
|
|
|
@Test
|
|
public void test() throws Exception {
|
|
|
|
new ClassReference(TreeDemo.class.getCanonicalName()).startApplication();
|
|
|
|
JFrameOperator frame = new JFrameOperator(DEMO_TITLE);
|
|
|
|
JTreeOperator tree = new JTreeOperator(frame);
|
|
|
|
assertEquals("Initial number of rows in the tree", 4, tree.getRowCount());
|
|
|
|
int initialTreeHeight = tree.getHeight();
|
|
|
|
// expand all nodes
|
|
int expandsCount = 0;
|
|
for (int i = 0; i < tree.getRowCount(); i++) {
|
|
TreePath tp = tree.getPathForRow(i);
|
|
if (tree.getChildCount(tp) > 0 && !tree.isExpanded(tp)) {
|
|
tree.expandRow(i);
|
|
expandsCount++;
|
|
}
|
|
}
|
|
|
|
assertEquals("Number of rows expanded", NODES_TO_EXPAND, expandsCount);
|
|
waitRowCount(tree, NODES_TOTAL);
|
|
|
|
int expandedTreeHeight = tree.getHeight();
|
|
assertTrue("Expanded tree height has increased, current "
|
|
+ expandedTreeHeight + " > initial " + initialTreeHeight,
|
|
expandedTreeHeight > initialTreeHeight);
|
|
|
|
// collapse all nodes
|
|
int collapsesCount = 0;
|
|
for (int i = tree.getRowCount() - 1; i >= 0; i--) {
|
|
TreePath tp = tree.getPathForRow(i);
|
|
if (tree.getChildCount(tp) > 0 && tree.isExpanded(tp)) {
|
|
tree.collapseRow(i);
|
|
collapsesCount++;
|
|
}
|
|
}
|
|
|
|
assertEquals("Number of rows collapsed", NODES_TO_EXPAND + 1, collapsesCount);
|
|
waitRowCount(tree, 1);
|
|
|
|
int collapsedTreeHeight = tree.getHeight();
|
|
assertTrue("Collpased tree height is not longer than initial, "
|
|
+ "current " + collapsedTreeHeight + " <= initial "
|
|
+ initialTreeHeight,
|
|
collapsedTreeHeight <= initialTreeHeight);
|
|
|
|
}
|
|
|
|
}
|