Make testXjc attach test sources

src/main/java/org/codehaus/mojo/jaxb2/AbstractJaxbMojo.java

@@ -361,10 +361,17 @@ public final void execute() throws MojoExecutionException, MojoFailureException
             }
 
             // Add the output Directory.
-            getProject().addCompileSourceRoot(canonicalPathToOutputDirectory);
+            addGeneratedSourcesToProjectSourceRoot(canonicalPathToOutputDirectory);
         }
     }
 
+    /**
+     * Adds any directories containing the generated XJC classes to the appropriate Project compilation sources;
+     * either {@code TestCompileSourceRoot} or {@code CompileSourceRoot} depending on the exact Mojo implementation
+     * of this AbstractJavaGeneratorMojo.
+     */
+    protected abstract void addGeneratedSourcesToProjectSourceRoot(String canonicalPathToOutputDirectory);
+
     /**
      * Implement this method to check if this AbstractJaxbMojo should skip executing altogether.
      *

src/main/java/org/codehaus/mojo/jaxb2/javageneration/AbstractJavaGeneratorMojo.java

@@ -503,7 +503,8 @@ protected boolean performExecution() throws MojoExecutionException, MojoFailureE
             }
 
             // Add the generated source root to the project, enabling tooling and other plugins to see them.
-            addGeneratedSourcesToProjectSourceRoot();
+            String canonicalPathToOutputDirectory = FileSystemUtilities.getCanonicalPath(getOutputDirectory());
+            addGeneratedSourcesToProjectSourceRoot(canonicalPathToOutputDirectory);
 
             // Copy all source XSDs to the resulting artifact?
             if (xsdPathWithinArtifact != null) {
@@ -579,13 +580,6 @@ protected boolean performExecution() throws MojoExecutionException, MojoFailureE
      */
     protected abstract List<File> getSourceXJBs();
 
-    /**
-     * Adds any directories containing the generated XJC classes to the appropriate Project compilation sources;
-     * either {@code TestCompileSourceRoot} or {@code CompileSourceRoot} depending on the exact Mojo implementation
-     * of this AbstractJavaGeneratorMojo.
-     */
-    protected abstract void addGeneratedSourcesToProjectSourceRoot();
-
     //
     // Private helpers
     //

src/main/java/org/codehaus/mojo/jaxb2/javageneration/TestXjcMojo.java

@@ -328,7 +328,7 @@ protected List<String> getClasspath() throws MojoExecutionException {
      * {@inheritDoc}
      */
     @Override
-    protected void addGeneratedSourcesToProjectSourceRoot() {
+    protected void addGeneratedSourcesToProjectSourceRoot(String canonicalPathToOutputDirectory) {
         getProject().addTestCompileSourceRoot(getOutputDirectory().getAbsolutePath());
     }
 

src/main/java/org/codehaus/mojo/jaxb2/javageneration/XjcMojo.java

@@ -339,8 +339,8 @@ protected List<String> getClasspath() throws MojoExecutionException {
      * {@inheritDoc}
      */
     @Override
-    protected void addGeneratedSourcesToProjectSourceRoot() {
-        getProject().addCompileSourceRoot(FileSystemUtilities.getCanonicalPath(getOutputDirectory()));
+    protected void addGeneratedSourcesToProjectSourceRoot(String canonicalPathToOutputDirectory) {
+        getProject().addCompileSourceRoot(canonicalPathToOutputDirectory);
     }
 
     /**

src/main/java/org/codehaus/mojo/jaxb2/schemageneration/SchemaGenerationMojo.java

@@ -246,6 +246,14 @@ protected List<String> getClasspath() throws MojoExecutionException {
         return toReturn;
     }
 
+    /**
+     * {@inheritDoc}
+     */
+    @Override
+    protected void addGeneratedSourcesToProjectSourceRoot(String canonicalPathToOutputDirectory) {
+        getProject().addCompileSourceRoot(canonicalPathToOutputDirectory);
+    }
+
     /**
      * {@inheritDoc}
      */

src/main/java/org/codehaus/mojo/jaxb2/schemageneration/TestSchemaGenerationMojo.java

@@ -215,6 +215,14 @@ protected List<String> getClasspath() throws MojoExecutionException {
         return toReturn;
     }
 
+    /**
+     * {@inheritDoc}
+     */
+    @Override
+    protected void addGeneratedSourcesToProjectSourceRoot(String canonicalPathToOutputDirectory) {
+        getProject().addTestCompileSourceRoot(canonicalPathToOutputDirectory);
+    }
+
     /**
      * {@inheritDoc}
      */