Add ProjectDefinition#resetSourceDirs() and resetTestDirs()

2 files changed, 30 insertions, 0 deletions

diff --git a/sonar-plugin-api/src/main/java/org/sonar/api/batch/bootstrap/ProjectDefinition.java b/sonar-plugin-api/src/main/java/org/sonar/api/batch/bootstrap/ProjectDefinition.java
index 71c6b9acfd2..1b062bc8131 100644
--- a/sonar-plugin-api/src/main/java/org/sonar/api/batch/bootstrap/ProjectDefinition.java
+++ b/sonar-plugin-api/src/main/java/org/sonar/api/batch/bootstrap/ProjectDefinition.java
@@ -161,6 +161,11 @@ public final class ProjectDefinition implements BatchComponent {
     return this;
   }
 
+  public ProjectDefinition resetSourceDirs() {
+    properties.remove(SOURCE_DIRS_PROPERTY);
+    return this;
+  }
+
   public ProjectDefinition setSourceDir(File path) {
     setSourceDir(path.getAbsolutePath());
     return this;
@@ -216,6 +221,11 @@ public final class ProjectDefinition implements BatchComponent {
     return this;
   }
 
+  public ProjectDefinition resetTestDirs() {
+    properties.remove(TEST_DIRS_PROPERTY);
+    return this;
+  }
+
 
   /**
    * Adding source files is possible only if no source directories have been set.
diff --git a/sonar-plugin-api/src/test/java/org/sonar/api/batch/bootstrap/ProjectDefinitionTest.java b/sonar-plugin-api/src/test/java/org/sonar/api/batch/bootstrap/ProjectDefinitionTest.java
index 9b92b37e880..ed13c6a719c 100644
--- a/sonar-plugin-api/src/test/java/org/sonar/api/batch/bootstrap/ProjectDefinitionTest.java
+++ b/sonar-plugin-api/src/test/java/org/sonar/api/batch/bootstrap/ProjectDefinitionTest.java
@@ -112,6 +112,26 @@ public class ProjectDefinitionTest {
     assertThat(child.getParent(), is(root));
   }
 
+  @Test
+  public void shouldResetSourceDirs() {
+    ProjectDefinition root = new ProjectDefinition(new File("."), new File("."), new Properties());
+    root.addSourceDirs("src", "src2/main");
+    assertThat(root.getSourceDirs().size(), is(2));
+
+    root.resetSourceDirs();
+    assertThat(root.getSourceDirs().size(), is(0));
+  }
+
+  @Test
+  public void shouldResetTestDirs() {
+    ProjectDefinition root = new ProjectDefinition(new File("."), new File("."), new Properties());
+    root.addTestDirs("src", "src2/test");
+    assertThat(root.getTestDirs().size(), is(2));
+
+    root.resetTestDirs();
+    assertThat(root.getTestDirs().size(), is(0));
+  }
+
   private static void assertFiles(List<String> paths, String... values) {
     assertThat(paths.size(), is(values.length));
     for (int i = 0; i < values.length; i++) {