adding more unit tests

LibraryOfCongress · Mar 24, 2016 · 9267fc9 · 9267fc9
1 parent 626fb89
commit 9267fc9
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/src/test/java/gov/loc/repository/bagit/creator/AddPayloadToBagManifestVistorTest.java b/src/test/java/gov/loc/repository/bagit/creator/AddPayloadToBagManifestVistorTest.java
@@ -1,5 +1,7 @@
 package gov.loc.repository.bagit.creator;
 
+import java.io.IOException;
+import java.nio.file.FileVisitResult;
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.nio.file.Paths;
@@ -26,4 +28,32 @@ public void includeDotKeepFilesInManifest() throws Exception{
     assertEquals(1, manifest.getFileToChecksumMap().size());
     assertTrue(manifest.getFileToChecksumMap().containsKey(start.resolve("fooDir/.keep")));
   }
+
+  @Test
+  public void testSkipDotBagitDir() throws IOException{
+    AddPayloadToBagManifestVistor sut = new AddPayloadToBagManifestVistor(null, null, true);
+    FileVisitResult returned = sut.preVisitDirectory(Paths.get("/foo/.bagit"), null);
+    assertEquals(FileVisitResult.SKIP_SUBTREE, returned);
+  }
+
+  @Test
+  public void testSkipHiddenDirectory() throws IOException{
+    AddPayloadToBagManifestVistor sut = new AddPayloadToBagManifestVistor(null, null, false);
+    FileVisitResult returned = sut.preVisitDirectory(Paths.get("/foo/.someHiddenDir"), null);
+    assertEquals(FileVisitResult.SKIP_SUBTREE, returned);
+  }
+
+  @Test
+  public void testIncludeHiddenDirectory() throws IOException{
+    AddPayloadToBagManifestVistor sut = new AddPayloadToBagManifestVistor(null, null, true);
+    FileVisitResult returned = sut.preVisitDirectory(Paths.get("/foo/.someHiddenDir"), null);
+    assertEquals(FileVisitResult.CONTINUE, returned);
+  }
+
+  @Test
+  public void testSkipHiddenFile() throws IOException{
+    AddPayloadToBagManifestVistor sut = new AddPayloadToBagManifestVistor(null, null, false);
+    FileVisitResult returned = sut.visitFile(Paths.get("/foo/.someHiddenDir"), null);
+    assertEquals(FileVisitResult.CONTINUE, returned);
+  }
 }