org.pantsbuild.tools.junit.withretry.AllDefaultPossibilitiesBuilderWithRetry Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of junit-runner-withretry Show documentation

Provides an org.junit.runner.Runner that supports retries of failing tests.

There is a newer version: 0.0.19

Show newest version

// Copyright 2015 Pants project contributors (see CONTRIBUTORS.md).
// Licensed under the Apache License, Version 2.0 (see LICENSE).

package org.pantsbuild.tools.junit.withretry;

import java.io.PrintStream;

import org.junit.internal.builders.AllDefaultPossibilitiesBuilder;
import org.junit.internal.builders.JUnit4Builder;

/**
 * Needed to support retrying flaky tests. Using method overriding, gives us access to code
 * in JUnit4 that cannot be customized in a simpler way.
 */
public class AllDefaultPossibilitiesBuilderWithRetry extends AllDefaultPossibilitiesBuilder {

  private final int numRetries;
  private final PrintStream err;

  public AllDefaultPossibilitiesBuilderWithRetry(int numRetries, PrintStream err) {
    super(true);
    this.numRetries = numRetries;
    this.err = err;
  }

  @Override
  public JUnit4Builder junit4Builder() {
    return new JUnit4BuilderWithRetry(numRetries, err);
  }
}