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

package net.sourceforge.tuned;


import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;

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());
	}
}