git.gag.com Git - debian/openrocket/blob - core/test/net/sf/openrocket/file/iterator/TestFileIterator.java

   1 package net.sf.openrocket.file.iterator;
   2
   3 import static org.junit.Assert.*;
   4
   5 import java.io.ByteArrayInputStream;
   6 import java.io.InputStream;
   7
   8 import net.sf.openrocket.util.Pair;
   9
  10 import org.junit.Test;
  11
  12 public class TestFileIterator {
  13
  14         @Test
  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 }));
  18
  19                 FileIterator iterator = new FileIterator() {
  20                         private int count = 0;
  21
  22                         @Override
  23                         protected Pair<String, InputStream> findNext() {
  24                                 count++;
  25                                 switch (count) {
  26                                 case 1:
  27                                         return one;
  28                                 case 2:
  29                                         return two;
  30                                 default:
  31                                         return null;
  32                                 }
  33                         }
  34                 };
  35
  36                 assertTrue(iterator.hasNext());
  37                 assertEquals(one, iterator.next());
  38                 assertEquals(two, iterator.next());
  39                 assertFalse(iterator.hasNext());
  40         }
  41 }