Home | History | Annotate | Download | only in framework 
      1 /**
      2  *
      3  */
      4 package junit.framework;
      5 
      6 import java.util.ArrayList;
      7 import java.util.Arrays;
      8 import java.util.HashMap;
      9 import java.util.List;
     10 
     11 import org.junit.runner.Description;
     12 import org.junit.runner.notification.Failure;
     13 import org.junit.runner.notification.RunListener;
     14 import org.junit.runner.notification.RunNotifier;
     15 
     16 public class JUnit4TestAdapterCache extends HashMap<Description, Test> {
     17 	private static final long serialVersionUID = 1L;
     18 	private static final JUnit4TestAdapterCache fInstance = new JUnit4TestAdapterCache();
     19 
     20 	public static JUnit4TestAdapterCache getDefault() {
     21 		return fInstance;
     22 	}
     23 
     24 	public Test asTest(Description description) {
     25 		if (description.isSuite())
     26 			return createTest(description);
     27 		else {
     28 			if (!containsKey(description))
     29 				put(description, createTest(description));
     30 			return get(description);
     31 		}
     32 	}
     33 
     34 	Test createTest(Description description) {
     35 		if (description.isTest())
     36 			return new JUnit4TestCaseFacade(description);
     37 		else {
     38 			TestSuite suite = new TestSuite(description.getDisplayName());
     39 			for (Description child : description.getChildren())
     40 				suite.addTest(asTest(child));
     41 			return suite;
     42 		}
     43 	}
     44 
     45 	public RunNotifier getNotifier(final TestResult result,
     46 			final JUnit4TestAdapter adapter) {
     47 		RunNotifier notifier = new RunNotifier();
     48 		notifier.addListener(new RunListener() {
     49 			@Override
     50 			public void testFailure(Failure failure) throws Exception {
     51 				result.addError(asTest(failure.getDescription()), failure.getException());
     52 			}
     53 
     54 			@Override
     55 			public void testFinished(Description description)
     56 					throws Exception {
     57 				result.endTest(asTest(description));
     58 			}
     59 
     60 			@Override
     61 			public void testStarted(Description description)
     62 					throws Exception {
     63 				result.startTest(asTest(description));
     64 			}
     65 		});
     66 		return notifier;
     67 	}
     68 
     69 	public List<Test> asTestList(Description description) {
     70 		if (description.isTest())
     71 			return Arrays.asList(asTest(description));
     72 		else {
     73 			List<Test> returnThis = new ArrayList<Test>();
     74 			for (Description child : description.getChildren()) {
     75 				returnThis.add(asTest(child));
     76 			}
     77 			return returnThis;
     78 		}
     79 	}
     80 
     81 }