1 files changed, 178 insertions, 0 deletions
diff --git a/jimfs/src/test/java/com/google/common/jimfs/FileSystemStateTest.java b/jimfs/src/test/java/com/google/common/jimfs/FileSystemStateTest.java
new file mode 100644
index 0000000..f8143b6
--- /dev/null
+++ b/jimfs/src/test/java/com/google/common/jimfs/FileSystemStateTest.java
@@ -0,0 +1,178 @@
+/*
+ * 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 org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+import static org.junit.Assert.fail;
+
+import com.google.common.collect.ImmutableList;
+import com.google.common.collect.ImmutableSet;
+import java.io.Closeable;
+import java.io.IOException;
+import java.nio.file.ClosedFileSystemException;
+import java.util.List;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.JUnit4;
+
+/**
+ * Tests for {@link FileSystemState}.
+ *
+ * @author Colin Decker
+ */
+@RunWith(JUnit4.class)
+public class FileSystemStateTest {
+
+  private final TestRunnable onClose = new TestRunnable();
+  private final FileSystemState state = new FileSystemState(onClose);
+
+  @Test
+  public void testIsOpen() throws IOException {
+    assertTrue(state.isOpen());
+    state.close();
+    assertFalse(state.isOpen());
+  }
+
+  @Test
+  public void testCheckOpen() throws IOException {
+    state.checkOpen(); // does not throw
+    state.close();
+    try {
+      state.checkOpen();
+      fail();
+    } catch (ClosedFileSystemException expected) {
+    }
+  }
+
+  @Test
+  public void testClose_callsOnCloseRunnable() throws IOException {
+    assertEquals(0, onClose.runCount);
+    state.close();
+    assertEquals(1, onClose.runCount);
+  }
+
+  @Test
+  public void testClose_multipleTimesDoNothing() throws IOException {
+    state.close();
+    assertEquals(1, onClose.runCount);
+    state.close();
+    state.close();
+    assertEquals(1, onClose.runCount);
+  }
+
+  @Test
+  public void testClose_registeredResourceIsClosed() throws IOException {
+    TestCloseable resource = new TestCloseable();
+    state.register(resource);
+    assertFalse(resource.closed);
+    state.close();
+    assertTrue(resource.closed);
+  }
+
+  @Test
+  public void testClose_unregisteredResourceIsNotClosed() throws IOException {
+    TestCloseable resource = new TestCloseable();
+    state.register(resource);
+    assertFalse(resource.closed);
+    state.unregister(resource);
+    state.close();
+    assertFalse(resource.closed);
+  }
+
+  @Test
+  public void testClose_multipleRegisteredResourcesAreClosed() throws IOException {
+    List<TestCloseable> resources =
+        ImmutableList.of(new TestCloseable(), new TestCloseable(), new TestCloseable());
+    for (TestCloseable resource : resources) {
+      state.register(resource);
+      assertFalse(resource.closed);
+    }
+    state.close();
+    for (TestCloseable resource : resources) {
+      assertTrue(resource.closed);
+    }
+  }
+
+  @Test
+  public void testClose_resourcesThatThrowOnClose() {
+    List<TestCloseable> resources =
+        ImmutableList.of(
+            new TestCloseable(),
+            new ThrowsOnClose("a"),
+            new TestCloseable(),
+            new ThrowsOnClose("b"),
+            new ThrowsOnClose("c"),
+            new TestCloseable(),
+            new TestCloseable());
+    for (TestCloseable resource : resources) {
+      state.register(resource);
+      assertFalse(resource.closed);
+    }
+
+    try {
+      state.close();
+      fail();
+    } catch (IOException expected) {
+      Throwable[] suppressed = expected.getSuppressed();
+      assertEquals(2, suppressed.length);
+      ImmutableSet<String> messages =
+          ImmutableSet.of(
+              expected.getMessage(), suppressed[0].getMessage(), suppressed[1].getMessage());
+      assertEquals(ImmutableSet.of("a", "b", "c"), messages);
+    }
+
+    for (TestCloseable resource : resources) {
+      assertTrue(resource.closed);
+    }
+  }
+
+  private static class TestCloseable implements Closeable {
+
+    boolean closed = false;
+
+    @Override
+    public void close() throws IOException {
+      closed = true;
+    }
+  }
+
+  private static final class TestRunnable implements Runnable {
+    int runCount = 0;
+
+    @Override
+    public void run() {
+      runCount++;
+    }
+  }
+
+  private static class ThrowsOnClose extends TestCloseable {
+
+    private final String string;
+
+    private ThrowsOnClose(String string) {
+      this.string = string;
+    }
+
+    @Override
+    public void close() throws IOException {
+      super.close();
+      throw new IOException(string);
+    }
+  }
+}