1 files changed, 205 insertions, 0 deletions
diff --git a/jimfs/src/test/java/com/google/common/jimfs/JimfsOutputStreamTest.java b/jimfs/src/test/java/com/google/common/jimfs/JimfsOutputStreamTest.java
new file mode 100644
index 0000000..3c230a7
--- /dev/null
+++ b/jimfs/src/test/java/com/google/common/jimfs/JimfsOutputStreamTest.java
@@ -0,0 +1,205 @@
+/*
+ * Copyright 2013 Google Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.google.common.jimfs;
+
+import static com.google.common.jimfs.TestUtils.bytes;
+import static com.google.common.jimfs.TestUtils.regularFile;
+import static java.nio.charset.StandardCharsets.UTF_8;
+import static org.junit.Assert.assertArrayEquals;
+import static org.junit.Assert.fail;
+
+import com.google.common.util.concurrent.Runnables;
+import java.io.BufferedOutputStream;
+import java.io.IOException;
+import java.io.OutputStreamWriter;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.JUnit4;
+
+/**
+ * Tests for {@link JimfsOutputStream}.
+ *
+ * @author Colin Decker
+ */
+@RunWith(JUnit4.class)
+public class JimfsOutputStreamTest {
+
+  @Test
+  public void testWrite_singleByte() throws IOException {
+    JimfsOutputStream out = newOutputStream(false);
+    out.write(1);
+    out.write(2);
+    out.write(3);
+    assertStoreContains(out, 1, 2, 3);
+  }
+
+  @Test
+  public void testWrite_wholeArray() throws IOException {
+    JimfsOutputStream out = newOutputStream(false);
+    out.write(new byte[] {1, 2, 3, 4});
+    assertStoreContains(out, 1, 2, 3, 4);
+  }
+
+  @Test
+  public void testWrite_partialArray() throws IOException {
+    JimfsOutputStream out = newOutputStream(false);
+    out.write(new byte[] {1, 2, 3, 4, 5, 6}, 1, 3);
+    assertStoreContains(out, 2, 3, 4);
+  }
+
+  @Test
+  public void testWrite_partialArray_invalidInput() throws IOException {
+    JimfsOutputStream out = newOutputStream(false);
+
+    try {
+      out.write(new byte[3], -1, 1);
+      fail();
+    } catch (IndexOutOfBoundsException expected) {
+    }
+
+    try {
+      out.write(new byte[3], 0, 4);
+      fail();
+    } catch (IndexOutOfBoundsException expected) {
+    }
+
+    try {
+      out.write(new byte[3], 1, 3);
+      fail();
+    } catch (IndexOutOfBoundsException expected) {
+    }
+  }
+
+  @Test
+  public void testWrite_singleByte_appendMode() throws IOException {
+    JimfsOutputStream out = newOutputStream(true);
+    addBytesToStore(out, 9, 8, 7);
+    out.write(1);
+    out.write(2);
+    out.write(3);
+    assertStoreContains(out, 9, 8, 7, 1, 2, 3);
+  }
+
+  @Test
+  public void testWrite_wholeArray_appendMode() throws IOException {
+    JimfsOutputStream out = newOutputStream(true);
+    addBytesToStore(out, 9, 8, 7);
+    out.write(new byte[] {1, 2, 3, 4});
+    assertStoreContains(out, 9, 8, 7, 1, 2, 3, 4);
+  }
+
+  @Test
+  public void testWrite_partialArray_appendMode() throws IOException {
+    JimfsOutputStream out = newOutputStream(true);
+    addBytesToStore(out, 9, 8, 7);
+    out.write(new byte[] {1, 2, 3, 4, 5, 6}, 1, 3);
+    assertStoreContains(out, 9, 8, 7, 2, 3, 4);
+  }
+
+  @Test
+  public void testWrite_singleByte_overwriting() throws IOException {
+    JimfsOutputStream out = newOutputStream(false);
+    addBytesToStore(out, 9, 8, 7, 6, 5, 4, 3);
+    out.write(1);
+    out.write(2);
+    out.write(3);
+    assertStoreContains(out, 1, 2, 3, 6, 5, 4, 3);
+  }
+
+  @Test
+  public void testWrite_wholeArray_overwriting() throws IOException {
+    JimfsOutputStream out = newOutputStream(false);
+    addBytesToStore(out, 9, 8, 7, 6, 5, 4, 3);
+    out.write(new byte[] {1, 2, 3, 4});
+    assertStoreContains(out, 1, 2, 3, 4, 5, 4, 3);
+  }
+
+  @Test
+  public void testWrite_partialArray_overwriting() throws IOException {
+    JimfsOutputStream out = newOutputStream(false);
+    addBytesToStore(out, 9, 8, 7, 6, 5, 4, 3);
+    out.write(new byte[] {1, 2, 3, 4, 5, 6}, 1, 3);
+    assertStoreContains(out, 2, 3, 4, 6, 5, 4, 3);
+  }
+
+  @Test
+  public void testClosedOutputStream_throwsException() throws IOException {
+    JimfsOutputStream out = newOutputStream(false);
+    out.close();
+
+    try {
+      out.write(1);
+      fail();
+    } catch (IOException expected) {
+    }
+
+    try {
+      out.write(new byte[3]);
+      fail();
+    } catch (IOException expected) {
+    }
+
+    try {
+      out.write(new byte[10], 1, 3);
+      fail();
+    } catch (IOException expected) {
+    }
+
+    out.close(); // does nothing
+  }
+
+  @Test
+  public void testClosedOutputStream_doesNotThrowOnFlush() throws IOException {
+    JimfsOutputStream out = newOutputStream(false);
+    out.close();
+    out.flush(); // does nothing
+
+    try (JimfsOutputStream out2 = newOutputStream(false);
+        BufferedOutputStream bout = new BufferedOutputStream(out2);
+        OutputStreamWriter writer = new OutputStreamWriter(bout, UTF_8)) {
+      /*
+       * This specific scenario is why flush() shouldn't throw when the stream is already closed.
+       * Nesting try-with-resources like this will cause close() to be called on the
+       * BufferedOutputStream multiple times. Each time, BufferedOutputStream will first call
+       * out2.flush(), then call out2.close(). If out2.flush() throws when the stream is already
+       * closed, the second flush() will throw an exception. Prior to JDK8, this exception would be
+       * swallowed and ignored completely; in JDK8, the exception is thrown from close().
+       */
+    }
+  }
+
+  private static JimfsOutputStream newOutputStream(boolean append) {
+    RegularFile file = regularFile(0);
+    return new JimfsOutputStream(file, append, new FileSystemState(Runnables.doNothing()));
+  }
+
+  @SuppressWarnings("GuardedByChecker")
+  private static void addBytesToStore(JimfsOutputStream out, int... bytes) throws IOException {
+    RegularFile file = out.file;
+    long pos = file.sizeWithoutLocking();
+    for (int b : bytes) {
+      file.write(pos++, (byte) b);
+    }
+  }
+
+  @SuppressWarnings("GuardedByChecker")
+  private static void assertStoreContains(JimfsOutputStream out, int... bytes) {
+    byte[] actualBytes = new byte[bytes.length];
+    out.file.read(0, actualBytes, 0, actualBytes.length);
+    assertArrayEquals(bytes(bytes), actualBytes);
+  }
+}