1 package net.sf.openrocket.file.iterator;
3 import static org.junit.Assert.*;
5 import java.io.ByteArrayInputStream;
6 import java.io.InputStream;
8 import net.sf.openrocket.util.Pair;
10 import org.junit.Test;
12 public class TestFileIterator {
15 public void testFileIterator() {
16 final Pair<String, InputStream> one = new Pair<String, InputStream>("one", new ByteArrayInputStream(new byte[] { 1 }));
17 final Pair<String, InputStream> two = new Pair<String, InputStream>("two", new ByteArrayInputStream(new byte[] { 2 }));
19 FileIterator iterator = new FileIterator() {
20 private int count = 0;
23 protected Pair<String, InputStream> findNext() {
36 assertTrue(iterator.hasNext());
37 assertEquals(one, iterator.next());
38 assertEquals(two, iterator.next());
39 assertFalse(iterator.hasNext());