org.pantsbuild.tools.junit.impl.CustomAnnotationBuilder Maven / Gradle / Ivy

Go to download

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

A command line tool for running junit tests that provides functionality above and beyond that provided by org.junit.runner.JUnitCore.

The 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.impl;

import java.io.PrintStream;

import org.junit.internal.builders.AllDefaultPossibilitiesBuilder;
import org.junit.internal.builders.AnnotatedBuilder;
import org.junit.internal.builders.JUnit4Builder;
import org.junit.runner.Runner;
import org.junit.runners.model.RunnerBuilder;
import org.pantsbuild.tools.junit.withretry.JUnit4BuilderWithRetry;

/**
 * Needed to support retrying flaky tests as well as add support for running scala tests.
 * Using method overriding, gives us access to code in JUnit4 that cannot be customized
 * in a simpler way.
 */
public class CustomAnnotationBuilder extends AllDefaultPossibilitiesBuilder {

  private final int numRetries;
  private final PrintStream err;

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

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

  // override annotated builder to "fake" the scala test junit runner for scala tests
  @Override
  protected AnnotatedBuilder annotatedBuilder() {
    return new ScalaTestAnnotatedBuilder(this);
  }

  private static class ScalaTestAnnotatedBuilder extends AnnotatedBuilder {
    ScalaTestAnnotatedBuilder(RunnerBuilder suiteBuilder) {
      super(suiteBuilder);
    }

    @Override
    public Runner runnerForClass(Class testClass) throws Exception {
      Runner runner = super.runnerForClass(testClass);
      if (runner == null) {
        if (ScalaTestUtil.isScalaTestTest(testClass)) {
          return ScalaTestUtil.getJUnitRunner(testClass);
        }
      }
      return runner;
    }
  }
}