Skip to content

Instantly share code, notes, and snippets.

@fappel
Last active June 5, 2023 16:58
Show Gist options
  • Save fappel/8bcb2aea4b39ff9cfb6e to your computer and use it in GitHub Desktop.
Save fappel/8bcb2aea4b39ff9cfb6e to your computer and use it in GitHub Desktop.
JUnit 4 TestRule to run a test repeatedly for a specified amount of repititions
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
@Retention( RetentionPolicy.RUNTIME )
@Target( {
java.lang.annotation.ElementType.METHOD
} )
public @interface Repeat {
public abstract int times();
}
import org.junit.rules.TestRule;
import org.junit.runner.Description;
import org.junit.runners.model.Statement;
public class RepeatRule implements TestRule {
private static class RepeatStatement extends Statement {
private final int times;
private final Statement statement;
private RepeatStatement( int times, Statement statement ) {
this.times = times;
this.statement = statement;
}
@Override
public void evaluate() throws Throwable {
for( int i = 0; i < times; i++ ) {
statement.evaluate();
}
}
}
@Override
public Statement apply( Statement statement, Description description ) {
Statement result = statement;
Repeat repeat = description.getAnnotation( Repeat.class );
if( repeat != null ) {
int times = repeat.times();
result = new RepeatStatement( times, statement );
}
return result;
}
}
@cbismuth
Copy link

cbismuth commented Sep 6, 2016

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment