1 files changed, 75 insertions, 0 deletions
diff --git a/java/org/brotli/wrapper/dec/EagerStreamTest.java b/java/org/brotli/wrapper/dec/EagerStreamTest.java
new file mode 100755
index 0000000..9166092
--- /dev/null
+++ b/java/org/brotli/wrapper/dec/EagerStreamTest.java
@@ -0,0 +1,75 @@
+/* Copyright 2017 Google Inc. All Rights Reserved.
+
+   Distributed under MIT license.
+   See file LICENSE for detail or copy at https://opensource.org/licenses/MIT
+*/
+
+package org.brotli.wrapper.dec;
+
+import static org.junit.Assert.assertEquals;
+
+import org.brotli.integration.BrotliJniTestBase;
+import java.io.IOException;
+import java.io.InputStream;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.JUnit4;
+
+/** Tests for {@link org.brotli.wrapper.dec.BrotliInputStream}. */
+@RunWith(JUnit4.class)
+public class EagerStreamTest extends BrotliJniTestBase {
+
+  @Test
+  public void testEagerReading() throws IOException {
+    final StringBuilder log = new StringBuilder();
+    final byte[] data = {0, 0, 16, 42, 3};
+    InputStream source = new InputStream() {
+      int index;
+
+      @Override
+      public int read() {
+        if (index < data.length) {
+          log.append("<").append(index);
+          return data[index++];
+        } else {
+          log.append("<#");
+          return -1;
+        }
+      }
+
+      @Override
+      public int read(byte[] b) throws IOException {
+        return read(b, 0, b.length);
+      }
+
+      @Override
+      public int read(byte[] b, int off, int len) throws IOException {
+        if (len < 1) {
+          return 0;
+        }
+        int d = read();
+        if (d == -1) {
+          return 0;
+        }
+        b[off] = (byte) d;
+        return 1;
+      }
+    };
+    BrotliInputStream reader = new BrotliInputStream(source);
+    reader.setEager(true);
+    int count = 0;
+    while (true) {
+      log.append("^").append(count);
+      int b = reader.read();
+      if (b == -1) {
+        log.append(">#");
+        break;
+      } else {
+        log.append(">").append(count++);
+      }
+    }
+    // Lazy log:  ^0<0<1<2<3<4>0^1>#
+    assertEquals("^0<0<1<2<3>0^1<4>#", log.toString());
+  }
+
+}