1 files changed, 64 insertions, 0 deletions
diff --git a/node-admin/src/test/java/com/yahoo/vespa/hosted/node/admin/task/util/file/FileSnapshotTest.java b/node-admin/src/test/java/com/yahoo/vespa/hosted/node/admin/task/util/file/FileSnapshotTest.java
new file mode 100644
index 00000000000..8c73d522f1d
--- /dev/null
+++ b/node-admin/src/test/java/com/yahoo/vespa/hosted/node/admin/task/util/file/FileSnapshotTest.java
@@ -0,0 +1,64 @@
+// Copyright 2019 Oath Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.vespa.hosted.node.admin.task.util.file;
+
+import com.yahoo.vespa.test.file.TestFileSystem;
+import org.junit.Test;
+
+import java.nio.file.FileSystem;
+import java.nio.file.Path;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertSame;
+import static org.junit.Assert.assertTrue;
+
+/**
+ * @author hakonhall
+ */
+public class FileSnapshotTest {
+    private final FileSystem fileSystem = TestFileSystem.create();
+    private final UnixPath path = new UnixPath(fileSystem.getPath("/var/lib/file.txt"));
+
+    private FileSnapshot fileSnapshot = FileSnapshot.forPath(path.toPath());
+
+    @Test
+    public void fileDoesNotExist() {
+        assertFalse(fileSnapshot.exists());
+        assertFalse(fileSnapshot.attributes().isPresent());
+        assertFalse(fileSnapshot.content().isPresent());
+        assertEquals(path.toPath(), fileSnapshot.path());
+    }
+
+    @Test
+    public void directory() {
+        path.createParents().createDirectory();
+        fileSnapshot = fileSnapshot.snapshot();
+        assertTrue(fileSnapshot.exists());
+        assertTrue(fileSnapshot.attributes().isPresent());
+        assertTrue(fileSnapshot.attributes().get().isDirectory());
+    }
+
+    @Test
+    public void regularFile() {
+        path.createParents().writeUtf8File("file content");
+        fileSnapshot = fileSnapshot.snapshot();
+        assertTrue(fileSnapshot.exists());
+        assertTrue(fileSnapshot.attributes().isPresent());
+        assertTrue(fileSnapshot.attributes().get().isRegularFile());
+        assertTrue(fileSnapshot.utf8Content().isPresent());
+        assertEquals("file content", fileSnapshot.utf8Content().get());
+
+        FileSnapshot newFileSnapshot = fileSnapshot.snapshot();
+        assertSame(fileSnapshot, newFileSnapshot);
+    }
+
+    @Test
+    public void fileRemoval() {
+        path.createParents().writeUtf8File("file content");
+        fileSnapshot = fileSnapshot.snapshot();
+        assertTrue(fileSnapshot.exists());
+        path.deleteIfExists();
+        fileSnapshot = fileSnapshot.snapshot();
+        assertFalse(fileSnapshot.exists());
+    }
+}
+\ No newline at end of file