filebot/test/net/sourceforge/tuned/TreeIteratorTest.java

77 lines
1.7 KiB
Java
Raw Normal View History

package net.sourceforge.tuned;
2009-05-17 17:22:44 +00:00
import static org.junit.Assert.*;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import org.junit.Before;
import org.junit.Test;
public class TreeIteratorTest {
private List<Object> tree;
@Before
public void setUp() throws Exception {
tree = new ArrayList<Object>();
tree.add("r1");
List<Object> branch = new ArrayList<Object>();
branch.add("b1");
branch.add("b2");
tree.add(branch);
tree.add("r2");
List<Object> treetop = new ArrayList<Object>();
treetop.add("t1");
treetop.add("t2");
treetop.add("t3");
List<Object> trunk = new ArrayList<Object>();
trunk.add(treetop);
tree.add(trunk);
}
@Test
public void iterate() {
TreeIterator<Object> treeIterator = new TreeIterator<Object>(tree) {
@SuppressWarnings("unchecked")
@Override
protected Iterator<Object> children(Object node) {
if (node instanceof Iterable)
return ((Iterable) node).iterator();
return null;
}
};
// check leafs (String) and nodes (Iterable)
assertTrue(treeIterator.next() instanceof Iterable); // root
assertEquals("r1", treeIterator.next());
assertTrue(treeIterator.next() instanceof Iterable); // branch
assertEquals("b1", treeIterator.next());
assertEquals("b2", treeIterator.next());
assertEquals("r2", treeIterator.next());
assertTrue(treeIterator.next() instanceof Iterable); // trunk
assertTrue(treeIterator.next() instanceof Iterable); // treetop
assertEquals("t1", treeIterator.next());
assertEquals("t2", treeIterator.next());
assertEquals("t3", treeIterator.next());
assertFalse(treeIterator.hasNext());
}
}