Added reRunsCount parameter to re-run failed tests

src/main/groovy/com/stanfy/spoon/gradle/SpoonExtension.groovy

@@ -56,4 +56,7 @@ class SpoonExtension {
 
   /** Grants all permisions for Android M >= devices */
   boolean grantAllPermissions
+
+  /** Tests re-run count to re-run tests in case of failure in order to eliminate flakiness */
+  int reRunsCount = 1;
 }

src/main/groovy/com/stanfy/spoon/gradle/SpoonPlugin.groovy

@@ -126,6 +126,7 @@ class SpoonPlugin implements Plugin<Project> {
         codeCoverage = config.codeCoverage
         shard = config.shard
         grantAllPermissions = config.grantAllPermissions
+        reRunsCount = config.reRunsCount
         if (config.adbTimeout != -1) {
           // Timeout is defined in seconds in the config.
           adbTimeout = config.adbTimeout * 1000

src/main/groovy/com/stanfy/spoon/gradle/SpoonRunTask.groovy

@@ -101,6 +101,9 @@ class SpoonRunTask extends DefaultTask implements VerificationTask {
   /** Grants all permisions for Android M >= devices */
   boolean grantAllPermissions
 
+  /** Tests re-run count to re-run tests in case of failure in order to eliminate flakiness */
+  int reRunsCount;
+
   @TaskAction
   void runSpoon() {
     LOG.info("Run instrumentation tests $instrumentationApk for app $applicationApk")
@@ -189,6 +192,12 @@ class SpoonRunTask extends DefaultTask implements VerificationTask {
     }
 
     boolean success = runBuilder.build().run()
+    int runsCount = 1;
+    while(!success && reRunsCount > runsCount) {
+        LOG.info("Tests failed. Trying to re-run now")
+        success = runBuilder.build().run()
+        runsCount++
+    }
 
     if (!success && !ignoreFailures) {
       throw new GradleException("Tests failed! See ${output}/index.html")