testing framework information added to ebuild where applicable

1 files changed, 42 insertions, 0 deletions

diff --git a/src/main/java/org/gentoo/java/ebuilder/maven/MavenEbuilder.java b/src/main/java/org/gentoo/java/ebuilder/maven/MavenEbuilder.java
index 4160e39..76a27a8 100644
--- a/src/main/java/org/gentoo/java/ebuilder/maven/MavenEbuilder.java
+++ b/src/main/java/org/gentoo/java/ebuilder/maven/MavenEbuilder.java
@@ -94,6 +94,39 @@ public class MavenEbuilder {
     }
 
     /**
+     * Determines the testing framework.
+     *
+     * @param testDepenencies    list of test dependencies
+     * @param commonDependencies list of common dependencies
+     *
+     * @return testing framework name or null
+     */
+    private String determineTestingFramework(
+            final List<ResolvedDependency> testDepenencies,
+            final List<ResolvedDependency> commonDependencies) {
+        final List<ResolvedDependency> dependencies = new ArrayList<>(
+                testDepenencies.size() + commonDependencies.size());
+        dependencies.addAll(testDepenencies);
+        dependencies.addAll(commonDependencies);
+
+        for (final ResolvedDependency dependency : dependencies) {
+            final MavenProject.Dependency mavenDependency
+                    = dependency.getMavenDependency();
+
+            if (mavenDependency.getGroupId() == null) {
+                continue;
+            }
+
+            if ("junit".equals(mavenDependency.getGroupId())
+                    && "junit".equals(mavenDependency.getArtifactId())) {
+                return "junit";
+            }
+        }
+
+        return null;
+    }
+
+    /**
      * Attempts to resolve dependencies using the specified cache.
      *
      * @param dependencies list of maven dependencies
@@ -490,6 +523,15 @@ public class MavenEbuilder {
             writer.println('"');
         }
 
+        final String testingFramework = determineTestingFramework(
+                testDependencies, commonDependencies);
+
+        if (testingFramework != null) {
+            writer.print("JAVA_TESTING_FRAMEWORK=\"");
+            writer.print(testingFramework);
+            writer.println('"');
+        }
+
         writer.print("JAVA_SRC_DIR=\"");
         writer.print(config.getWorkdir().relativize(
                 mavenProject.getSourceDirectory()));