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 failure) throws 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;
}
}
}