1 files changed, 21 insertions, 0 deletions
diff --git a/src/test/java/com/google/devtools/build/lib/actions/ArtifactTest.java b/src/test/java/com/google/devtools/build/lib/actions/ArtifactTest.java
index 81fabbb16a..cd978217d0 100644
--- a/src/test/java/com/google/devtools/build/lib/actions/ArtifactTest.java
+++ b/src/test/java/com/google/devtools/build/lib/actions/ArtifactTest.java
@@ -337,6 +337,27 @@ public class ArtifactTest {
     assertThat(constructed).isEqualTo("aaa/bbb/ccc/ddd");
   }
 
+  @Test
+  public void testIsSourceArtifact() throws Exception {
+    assertThat(
+        new Artifact(scratch.file("/src/foo.cc"), Root.asSourceRoot(scratch.dir("/")),
+            new PathFragment("src/foo.cc"))
+            .isSourceArtifact())
+        .isTrue();
+    assertThat(
+        new Artifact(scratch.file("/genfiles/aaa/bar.out"),
+            Root.asDerivedRoot(scratch.dir("/genfiles"), scratch.dir("/genfiles/aaa")))
+            .isSourceArtifact())
+        .isFalse();
+
+  }
+
+  @Test
+  public void testGetRoot() throws Exception {
+    Root root = Root.asDerivedRoot(scratch.dir("/newRoot"));
+    assertThat(new Artifact(scratch.file("/newRoot/foo"), root).getRoot()).isEqualTo(root);
+  }
+
   private Artifact createDirNameArtifact() throws Exception {
     return new Artifact(scratch.file("/aaa/bbb/ccc/ddd"), Root.asDerivedRoot(scratch.dir("/")));
   }