Open Source Repository

Home /junit/junit-4.8.2 | Repository Home


junit/framework/JUnit4TestAdapterCache.java
/**
 
 */
package junit.framework;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;

import org.junit.runner.Description;
import org.junit.runner.notification.Failure;
import org.junit.runner.notification.RunListener;
import org.junit.runner.notification.RunNotifier;

public class JUnit4TestAdapterCache extends HashMap<Description, Test> {
  private static final long serialVersionUID = 1L;
  private static final JUnit4TestAdapterCache fInstance = new JUnit4TestAdapterCache();

  public static JUnit4TestAdapterCache getDefault() {
    return fInstance;
  }
  
  public Test asTest(Description description) {
    if (description.isSuite())
      return createTest(description);
    else {
      if (!containsKey(description))
        put(description, createTest(description));
      return get(description);
    }
  }

  Test createTest(Description description) {
    if (description.isTest())
      return new JUnit4TestCaseFacade(description);
    else {
      TestSuite suite = new TestSuite(description.getDisplayName());
      for (Description child : description.getChildren())
        suite.addTest(asTest(child));
      return suite;
    }
  }

  public RunNotifier getNotifier(final TestResult result,
      final JUnit4TestAdapter adapter) {
    RunNotifier notifier = new RunNotifier();
    notifier.addListener(new RunListener() {
      @Override
      public void testFailure(Failure failurethrows Exception {
        result.addError(asTest(failure.getDescription()), failure.getException());
      }

      @Override
      public void testFinished(Description description)
          throws Exception {
        result.endTest(asTest(description));
      }

      @Override
      public void testStarted(Description description)
          throws Exception {
        result.startTest(asTest(description));
      }
    });
    return notifier;
  }

  public List<Test> asTestList(Description description) {
    if (description.isTest())
      return Arrays.asList(asTest(description));
    else {
      List<Test> returnThis = new ArrayList<Test>();
      for (Description child : description.getChildren()) {
        returnThis.add(asTest(child));
      }
      return returnThis;
    }
  }

}