STAMP-project#19 Fix selection of tests

nrainer · Sep 3, 2018 · e77cd45 · e77cd45
1 parent 888fb35
commit e77cd45
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 2 deletions.
diff --git a/src/main/java/eu/stamp_project/PmpContext.java b/src/main/java/eu/stamp_project/PmpContext.java
@@ -113,12 +113,21 @@ public static Boolean oneFileExists(List<String> fileNameList)
 
     // **********************************************************************
     public static ArrayList<String> getClasses(MavenProject theProject)
+    {
+        return getClassesInternal(theProject.getBuild().getOutputDirectory());
+    }
+
+    public static ArrayList<String> getTestClasses(MavenProject theProject)
+    {
+        return getClassesInternal(theProject.getBuild().getTestOutputDirectory());
+    }
+
+    private static ArrayList<String> getClassesInternal(String buildOutputDirectory)
     {
         ArrayList<String> classList = new ArrayList<String>();
         ArrayList<String> classFilterList = new ArrayList<String>();
-        String outputDirName = theProject.getBuild().getOutputDirectory();
         String aFilter;
-        File outputDir = new File(outputDirName);
+        File outputDir = new File(buildOutputDirectory);
 
         if (outputDir.exists())
         {

diff --git a/src/main/java/org/pitest/maven/PmpMojo.java b/src/main/java/org/pitest/maven/PmpMojo.java
@@ -183,6 +183,13 @@ public void updateTargetClasses()
             }
         }
     }
+
+    public void updateTargetTests() {
+    	if (targetTests == null || targetTests.isEmpty()) {
+    		targetTests = new ArrayList<>();
+    		targetTests.addAll(PmpContext.getTestClasses(getProject()));
+    	}
+    }
 
     // **********************************************************************
     // protected
@@ -219,6 +226,7 @@ protected RunDecision shouldRun()
         PmpContext.getInstance().setCurrentProject(myPmpProject);
 
         updateTargetClasses();
+        updateTargetTests();
 
         if (getProject().getPackaging().equals("pom") &&
              myPmpProject.hasTestCompileSourceRoots() &&