1 files changed, 81 insertions, 0 deletions
diff --git a/junit4/src/main/java/junit/framework/JUnit4TestAdapterCache.java b/junit4/src/main/java/junit/framework/JUnit4TestAdapterCache.java
new file mode 100644
index 0000000..26175c5
--- /dev/null
+++ b/junit4/src/main/java/junit/framework/JUnit4TestAdapterCache.java
@@ -0,0 +1,81 @@
+/**
+ * 
+ */
+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;
+		}
+	}
+
+}
+\ No newline at end of file