Add test target as well

1 files changed, 15 insertions, 11 deletions

diff --git a/src/main/groovy/FlycheckAndroidJavaTask.groovy b/src/main/groovy/FlycheckAndroidJavaTask.groovy
index 28ca6bb..4984d69 100644
--- a/src/main/groovy/FlycheckAndroidJavaTask.groovy
+++ b/src/main/groovy/FlycheckAndroidJavaTask.groovy
@@ -14,17 +14,21 @@ class FlycheckAndroidJavaTask extends DefaultTask {
 
     @TaskAction
     def action() {
-        def compile = project.tasks.findByName('compileDebugJavaWithJavac')
-        if (compile instanceof JavaCompile) {
-            println '***'
-            println 'args=' + compile.options.compilerArgs
-            println 'encoding=' + compile.options.encoding
-            println 'bootcp=' + compile.options.bootClasspath
-            println 'cp=' + compile.classpath.asPath
-            println 'source=' + compile.sourceCompatibility
-            println 'target=' + compile.targetCompatibility
-            println 'files=' + compile.inputs.files.asPath
-            println 'output=' + compile.destinationDir
+        def tasks = ['compileDebugJavaWithJavac',
+                     'compileDebugUnitTestJavaWithJavac']
+        for (String task : tasks) {
+            def compile = project.tasks.findByName(task)
+            if (compile instanceof JavaCompile) {
+                println '***'
+                println 'args=' + compile.options.compilerArgs
+                println 'encoding=' + compile.options.encoding
+                println 'bootcp=' + compile.options.bootClasspath
+                println 'cp=' + compile.classpath.asPath
+                println 'source=' + compile.sourceCompatibility
+                println 'target=' + compile.targetCompatibility
+                println 'files=' + compile.inputs.files.asPath
+                println 'output=' + compile.destinationDir
+            }
         }
     }
 }