Added configuration closure if 'test' task is found

Author: ysb33r

src/main/groovy/com/github/jrubygradle/JRubyJarPlugin.groovy

@@ -2,7 +2,10 @@ package com.github.jrubygradle
 
 import org.gradle.api.Plugin
 import org.gradle.api.Project
+import org.gradle.api.Task
+import org.gradle.api.UnknownTaskException
 import org.gradle.api.tasks.bundling.Jar
+import org.gradle.api.tasks.testing.Test
 
 /**
  * @author Schalk W. Cronjé
@@ -16,20 +19,31 @@ class JRubyJarPlugin implements Plugin<Project> {
         project.configurations.maybeCreate('testGems')
         project.configurations.maybeCreate('runtimeGems')
 
+        // TODO: This is currently fixed. Really need to find a way to allow user to configure the version
         project.dependencies {
             jrubyEmbeds group: 'com.lookout', name: 'warbler-bootstrap', version: '1.0.0'
         }
 
-// TODO: This will depend on which plugin we pull in
-//            // In order to update the testing cycle we need to tell unit tests where to
-//            // find GEMs. We are assuming that if someone includes this plugin, that they
-//            // will be writing tests that includes jruby and that they might need some
-//            // GEMs as part of the tests.
-//            project.tasks.test {
-//                environment GEM_HOME : project.extensions.getByName('jruby').gemInstallDir
-//                dependsOn 'jrubyPrepareGems'
-//            }
-
+        // In order to update the testing cycle we need to tell unit tests where to
+        // find GEMs. We are assuming that if someone includes this plugin, that they
+        // will be writing tests that includes jruby and that they might need some
+        // GEMs as part of the tests.
+        def testConfiguration = { Task t ->
+            environment GEM_HOME : project.extensions.getByName('jruby').gemInstallDir
+            dependsOn 'jrubyPrepareGems'
+        }
+        try {
+            Task t = project.tasks.getByName('test')
+            if( t instanceof Test) {
+                project.configure(t,testConfiguration)
+            }
+        } catch(UnknownTaskException) {
+            project.tasks.whenTaskAdded { Task t ->
+                if(t.name == 'test' && t instanceof Test) {
+                    project.configure(t,testConfiguration)
+                }
+            }
+        }
 
        if(!Jar.metaClass.respondsTo(Jar.class,'jruby',Closure)) {
            Jar.metaClass.jruby = { Closure extraConfig ->

src/test/groovy/com/github/jrubygradle/JRubyJarPluginSpec.groovy

@@ -10,6 +10,7 @@ import static org.gradle.api.logging.LogLevel.LIFECYCLE
 
 /**
  * @author R. Tyler Croy
+ * @author Schalk W. Cronjé
  *
  */
 class JRubyJarPluginSpec extends Specification {