Implemented repeat attempts by default

library/src/main/java/com/schibsted/spain/barista/rule/flaky/FlakyTestRule.kt

@@ -29,10 +29,23 @@ class FlakyTestRule : TestRule {
     return this
   }
 
+  /**
+   * Utility method to use @[Repeat] by default in all test methods.
+   * <br></br>
+   * Use this method when constructing the Rule to apply a default behavior of @[Repeat] without having to add the annotation to
+   * each test. This can help you to find flaky tests.
+   * <br></br>
+   * The default behavior can be overridden with [Repeat] or [Repeat].
+   */
+  fun repeatAttemptsByDefault(defaultAttempts: Int): FlakyTestRule {
+    flakyStatementBuilder.setRepeatAttemptsByDefault(defaultAttempts)
+    return this
+  }
+
   override fun apply(base: Statement, description: Description): Statement {
     return flakyStatementBuilder
         .setBase(base)
         .setDescription(description)
         .build()
   }
-}
+}

library/src/main/java/com/schibsted/spain/barista/rule/flaky/internal/FlakyStatementBuilder.java

@@ -11,6 +11,8 @@ public class FlakyStatementBuilder {
   private Description description;
   private boolean useAllowFlakyByDefault = false;
   private int defaultAllowFlakyAttempts = 0;
+  private int defaultRepeatAttempts = 1;
+  private boolean useRepeatByDefault = false;
 
   public FlakyStatementBuilder setBase(Statement base) {
     this.base = base;
@@ -22,8 +24,16 @@ public FlakyStatementBuilder setDescription(Description description) {
     return this;
   }
 
+  public FlakyStatementBuilder setRepeatAttemptsByDefault(int attempts) {
+    useAllowFlakyByDefault = false;
+    useRepeatByDefault = true;
+    defaultRepeatAttempts = attempts;
+    return this;
+  }
+
   public FlakyStatementBuilder allowFlakyAttemptsByDefault(int attempts) {
     useAllowFlakyByDefault = true;
+    useRepeatByDefault = false;
     defaultAllowFlakyAttempts = attempts;
     return this;
   }
@@ -46,6 +56,8 @@ public Statement build() {
       return new AllowFlakyStatement(attempts, base);
     } else if (useAllowFlakyByDefault) {
       return new AllowFlakyStatement(defaultAllowFlakyAttempts, base);
+    } else if (useRepeatByDefault) {
+      return new RepeatStatement(defaultRepeatAttempts, base);
     } else {
       return base;
     }

library/src/test/java/com/schibsted/spain/barista/rule/flaky/FlakyStatementBuilderTest.java

@@ -44,6 +44,20 @@ public void repeatStatementReturnedWhenRepeatAnnotationFound() throws Exception
     assertTrue(resultStatement instanceof RepeatStatement);
   }
 
+  @Test
+  public void repeatStatementReturnedWhenRepeatSetRepeatAttemptsByDefault() throws Exception {
+    Statement baseStatement = new SomeStatement();
+    Description description = Description.EMPTY;
+
+    Statement resultStatement = new FlakyStatementBuilder()
+        .setBase(baseStatement)
+        .setDescription(description)
+        .setRepeatAttemptsByDefault(3)
+        .build();
+
+    assertTrue(resultStatement instanceof RepeatStatement);
+  }
+
   @Test
   public void allowFlakyStatementReturnedWhenAllowFlakyAnnotationFound() throws Exception {
     Statement baseStatement = new SomeStatement();
@@ -68,6 +82,36 @@ public void allowFlakyStatementReturnedWhenNoAnnotationsFoundButUsesDefault() th
     assertTrue(resultStatement instanceof AllowFlakyStatement);
   }
 
+  @Test
+  public void lastStatementReturnedWhenRepeatSetRepeatAttemptsByDefaultAndAllowFlakyStatementAtTheSameTime() throws Exception {
+    Statement baseStatement = new SomeStatement();
+    Description description = Description.EMPTY;
+
+    Statement resultStatement = new FlakyStatementBuilder()
+        .setBase(baseStatement)
+        .setDescription(description)
+        .allowFlakyAttemptsByDefault(5)
+        .setRepeatAttemptsByDefault(3)
+        .build();
+
+    assertTrue(resultStatement instanceof RepeatStatement);
+  }
+
+  @Test
+  public void allowFlakyStatementReturnedWhenAllowFlakyAnnotationFoundEvenRepeatStatement() throws Exception {
+    Statement baseStatement = new SomeStatement();
+    Description description = Description.EMPTY;
+
+    Statement resultStatement = new FlakyStatementBuilder()
+        .setBase(baseStatement)
+        .setDescription(description)
+        .allowFlakyAttemptsByDefault(5)
+        .setRepeatAttemptsByDefault(3)
+        .build();
+
+    assertTrue(resultStatement instanceof RepeatStatement);
+  }
+
   //region Shortcut methods
   private Statement createStatement(Statement baseStatement, Description description) {
     return new FlakyStatementBuilder()
@@ -115,4 +159,4 @@ public void evaluate() throws Throwable {
     }
   }
   //endregion
-}
+}