Author: jpeterka Date: 2012-03-19 03:22:03 -0400 (Mon, 19 Mar 2012) New Revision: 39601 Modified: trunk/hibernatetools/tests/org.jboss.tools.hibernate.ui.bot.test/src/org/jboss/tools/hb/ui/bot/common/Tree.java Log: getSubnodes routine added into Tree ui bot helper Modified: trunk/hibernatetools/tests/org.jboss.tools.hibernate.ui.bot.test/src/org/jboss/tools/hb/ui/bot/common/Tree.java =================================================================== --- trunk/hibernatetools/tests/org.jboss.tools.hibernate.ui.bot.test/src/org/jboss/tools/hb/ui/bot/common/Tree.java 2012-03-17 17:56:54 UTC (rev 39600) +++ trunk/hibernatetools/tests/org.jboss.tools.hibernate.ui.bot.test/src/org/jboss/tools/hb/ui/bot/common/Tree.java 2012-03-19 07:22:03 UTC (rev 39601) @@ -2,6 +2,9 @@ import static org.junit.Assert.fail; +import java.util.ArrayList; +import java.util.List; + import org.apache.log4j.Logger; import org.eclipse.swtbot.swt.finder.SWTBot; import org.eclipse.swtbot.swt.finder.widgets.SWTBotTree; @@ -81,10 +84,12 @@ private static void expandNode(SWTBotTreeItem item) { - item.expand(); if (!item.isExpanded()) { - fail("Unable to expand " + item.getText()); + item.expand(); } + if (!item.isExpanded()) { + log.error("Unable to expand:" + item.getText()); + } } @@ -104,12 +109,51 @@ } + private static boolean hasSubnodes(SWTBotTreeItem item) { + return (item.getNodes().size() > 0); + } + private static void collapseNode(SWTBotTreeItem item) { if (item.isExpanded()) { item.collapse(); } if (item.isExpanded()) { - fail("Unable to collapse" + item.getText()); + log.error("Unable to collapse:" + item.getText()); } } + + public static List<String> getSubNodes(SWTBot bot, SWTBotTreeItem item) { + + List<String> ret = new ArrayList<String>(); + expandNode(item); + + final int limit = 5; // 5 cycles max + final int sleep = 1000; // 1s + int counter = 0; + // 1st round + while (counter < limit) { + if (!hasSubnodes(item)) { + log.info("no subnodes"); + bot.sleep(sleep); + } + else { + ret = item.getNodes(); + break; + } + } + // 2nd round + collapseNode(item); + expandNode(item); + while (counter < limit) { + if (!hasSubnodes(item)) { + log.info("no subnodes"); + bot.sleep(sleep); + } + else { + ret = item.getNodes(); + break; + } + } + return ret; + } }