1 files changed, 114 insertions, 0 deletions
diff --git a/src/test/java/com/google/devtools/build/lib/skyframe/FileArtifactValueTest.java b/src/test/java/com/google/devtools/build/lib/skyframe/FileArtifactValueTest.java
index 641c28e283..ccf282243f 100644
--- a/src/test/java/com/google/devtools/build/lib/skyframe/FileArtifactValueTest.java
+++ b/src/test/java/com/google/devtools/build/lib/skyframe/FileArtifactValueTest.java
@@ -13,14 +13,39 @@
 // limitations under the License.
 package com.google.devtools.build.lib.skyframe;
 
+import static com.google.common.truth.Truth.assertThat;
+import static com.google.devtools.build.lib.skyframe.FileArtifactValue.create;
+import static org.junit.Assert.fail;
+
 import com.google.common.io.BaseEncoding;
 import com.google.common.testing.EqualsTester;
+import com.google.devtools.build.lib.vfs.FileSystem;
+import com.google.devtools.build.lib.vfs.FileSystemUtils;
+import com.google.devtools.build.lib.vfs.Path;
+import com.google.devtools.build.lib.vfs.inmemoryfs.InMemoryFileSystem;
+import java.io.IOException;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;
 
 @RunWith(JUnit4.class)
 public class FileArtifactValueTest {
+  private final FileSystem fs = new InMemoryFileSystem();
+
+  private Path scratchFile(String name, long mtime, String content) throws IOException {
+    Path path = fs.getPath(name);
+    FileSystemUtils.createDirectoryAndParents(path.getParentDirectory());
+    FileSystemUtils.writeContentAsLatin1(path, content);
+    path.setLastModifiedTime(mtime);
+    return path;
+  }
+
+  private Path scratchDir(String name, long mtime) throws IOException {
+    Path path = fs.getPath(name);
+    FileSystemUtils.createDirectoryAndParents(path);
+    path.setLastModifiedTime(mtime);
+    return path;
+  }
 
   private static byte[] toBytes(String hex) {
     return BaseEncoding.base16().upperCase().decode(hex);
@@ -48,4 +73,93 @@ public class FileArtifactValueTest {
         .addEqualityGroup("a string")
         .testEquals();
   }
+
+  @Test
+  public void testEquality() throws Exception {
+    Path path1 = scratchFile("/dir/artifact1", 0L, "content");
+    Path path2 = scratchFile("/dir/artifact2", 0L, "content");
+    Path digestPath = scratchFile("/dir/diffDigest", 0L, "1234567");
+    Path mtimePath = scratchFile("/dir/diffMtime", 1L, "content");
+
+    Path empty1 = scratchFile("/dir/empty1", 0L, "");
+    Path empty2 = scratchFile("/dir/empty2", 1L, "");
+    Path empty3 = scratchFile("/dir/empty3", 1L, "");
+
+    Path dir1 = scratchDir("/dir1", 0L);
+    Path dir2 = scratchDir("/dir2", 1L);
+    Path dir3 = scratchDir("/dir3", 1L);
+
+    new EqualsTester()
+        .addEqualityGroup(create(path1), create(path2), create(mtimePath))
+        .addEqualityGroup(create(digestPath))
+        .addEqualityGroup(create(empty1), create(empty2), create(empty3))
+        .addEqualityGroup(create(dir1))
+        .addEqualityGroup(create(dir2), create(dir3))
+        .testEquals();
+  }
+
+  @Test
+  public void testNoMtimeIfNonemptyFile() throws Exception {
+    Path path = scratchFile("/root/non-empty", 1L, "abc");
+    FileArtifactValue value = create(path);
+    assertThat(value.getDigest()).isEqualTo(path.getDigest());
+    assertThat(value.getSize()).isEqualTo(3L);
+    try {
+      value.getModifiedTime();
+      fail("mtime for non-empty file should not be stored.");
+    } catch (UnsupportedOperationException e) {
+      // Expected.
+    }
+  }
+
+  @Test
+  public void testDirectory() throws Exception {
+    Path path = fs.getPath("/dir");
+    FileSystemUtils.createDirectoryAndParents(path);
+    path.setLastModifiedTime(1L);
+    FileArtifactValue value = create(path);
+    assertThat(value.getDigest()).isNull();
+    assertThat(value.getModifiedTime()).isEqualTo(1L);
+  }
+
+  // Empty files are the same as normal files -- mtime is not stored.
+  @Test
+  public void testEmptyFile() throws Exception {
+    Path path = scratchFile("/root/empty", 1L, "");
+    path.setLastModifiedTime(1L);
+    FileArtifactValue value = create(path);
+    assertThat(value.getDigest()).isEqualTo(path.getDigest());
+    assertThat(value.getSize()).isEqualTo(0L);
+    try {
+      value.getModifiedTime();
+      fail("mtime for non-empty file should not be stored.");
+    } catch (UnsupportedOperationException e) {
+      // Expected.
+    }
+  }
+
+  @Test
+  public void testIOException() throws Exception {
+    final IOException exception = new IOException("beep");
+    FileSystem fs = new InMemoryFileSystem() {
+      @Override
+      public byte[] getDigest(Path path, HashFunction hf) throws IOException {
+        throw exception;
+      }
+
+      @Override
+      protected byte[] getFastDigest(Path path, HashFunction hashFunction) throws IOException {
+        throw exception;
+      }
+    };
+    Path path = fs.getPath("/some/path");
+    FileSystemUtils.createDirectoryAndParents(path.getParentDirectory());
+    FileSystemUtils.writeContentAsLatin1(path, "content");
+    try {
+      create(path);
+      fail();
+    } catch (IOException e) {
+      assertThat(e).isSameAs(exception);
+    }
+  }
 }