Ability to add test listener

Author: artyomcool

src/main/kotlin/com/jaredsburrows/spoon/SpoonExtension.kt

@@ -1,5 +1,7 @@
 package com.jaredsburrows.spoon
 
+import com.android.ddmlib.testrunner.ITestRunListener
+
 /**
  * Variables based on the following documentation:
  * - https://developer.android.com/reference/android/support/test/runner/AndroidJUnitRunner.html
@@ -78,7 +80,8 @@ open class SpoonExtension { // Extensions cannot be final
   /** Toggle sharding. (false by default) */
   var shard: Boolean = false
 
-  // TODO testRunListener
+  /** Add listener for tests execution */
+  var testRunListener: ITestRunListener? = null
 
   // TODO terminateAdb
 

src/main/kotlin/com/jaredsburrows/spoon/SpoonTask.kt

@@ -1,6 +1,8 @@
 package com.jaredsburrows.spoon
 
 import com.android.ddmlib.testrunner.IRemoteAndroidTestRunner.TestSize
+import com.android.ddmlib.testrunner.ITestRunListener
+import com.android.ddmlib.testrunner.TestIdentifier
 import com.squareup.spoon.SpoonRunner
 import org.gradle.api.DefaultTask
 import org.gradle.api.GradleException
@@ -55,6 +57,10 @@ open class SpoonTask : DefaultTask() {
       .setSingleInstrumentationCall(extension.singleInstrumentationCall)
       .setClearAppDataBeforeEachTest(extension.clearAppDataBeforeEachTest)
 
+    if (extension.testRunListener != null) {
+      builder.addTestRunListener(extension.testRunListener)
+    }
+
     // APKs
     if (!testing) {
       builder.setTestApk(instrumentationApk)