Implement FileFinder::prune

1 files changed, 21 insertions, 0 deletions

diff --git a/node-admin/src/test/java/com/yahoo/vespa/hosted/node/admin/task/util/file/FileFinderTest.java b/node-admin/src/test/java/com/yahoo/vespa/hosted/node/admin/task/util/file/FileFinderTest.java
index 8fff2141252..26897839b32 100644
--- a/node-admin/src/test/java/com/yahoo/vespa/hosted/node/admin/task/util/file/FileFinderTest.java
+++ b/node-admin/src/test/java/com/yahoo/vespa/hosted/node/admin/task/util/file/FileFinderTest.java
@@ -61,6 +61,27 @@ public class FileFinderTest {
         }
 
         @Test
+        public void all_files_recursive_with_prune_relative() {
+            assertFileHelper(FileFinder.files(testRoot()).prune(fileSystem.getPath("test")),
+
+                    of("file-1.json", "test.json", "test.txt"),
+                    of("test", "test/file.txt", "test/data.json", "test/subdir-1", "test/subdir-1/test", "test/subdir-2"));
+        }
+
+        @Test
+        public void all_files_recursive_with_prune_absolute() {
+            assertFileHelper(FileFinder.files(testRoot()).prune(testRoot().resolve("test/subdir-1")),
+
+                    of("file-1.json", "test.json", "test.txt", "test/file.txt", "test/data.json"),
+                    of("test", "test/subdir-1", "test/subdir-1/test", "test/subdir-2"));
+        }
+
+        @Test(expected = IllegalArgumentException.class)
+        public void throws_if_prune_path_not_under_base_path() {
+            FileFinder.files(Paths.get("/some/path")).prune(Paths.get("/other/path"));
+        }
+
+        @Test
         public void with_file_filter_recursive() {
             assertFileHelper(FileFinder.files(testRoot())
                             .match(FileFinder.nameEndsWith(".json")),