List list = Scans.me().scanPackage("org.nutz");
List reqs = new ArrayList();
Map reqMap = new HashMap();
for (Class clazz : list) {
Method[] methods = clazz.getMethods();
for (Method method : methods) {
if (method.getAnnotation(Test.class) != null) {
//将单个TestCase(即一个Test Method),封装为Junit的Test Request
Request req = Request.method(clazz, method.getName());
reqs.add(req);
reqMap.put(req , method);//在最终打印测试结果时,方便查找具体出错的Method
}
}
}
// 因为reqs 是一个List,我们可以按需调整TestCase的顺序
// 正序 //nothing change.
// 反序Collections.reverse(reqs)
// 乱序Collections.shuffle(reqs)
//把执行顺序保存下来,方便重现执行顺序
try {
FileWriter fw = new FileWriter("./test_order.txt");
for (Request request : reqs) {
fw.write(reqMap.get(request).toString());
fw.write("\n");
}
fw.flush();
fw.close();
}
catch (IOException e) {}
//到这里, List已经按我们预期的方式排好,可以执行测试了
final TestResult result = new TestResult();
RunNotifier notifier = new RunNotifier();
notifier.addListener(new RunListener() { //需要设置一个RunListener,以便收集测试结果
public void testFailure(Failure failure) throws Exception {
result.addError(asTest(failure.getDescription()), failure.getException());
}
public void testFinished(Description description) throws Exception {
result.endTest(asTest(description));
}
public void testStarted(Description description) throws Exception {
result.startTest(asTest(description));
}
public junit.framework.Test asTest(Description description) {
return new junit.framework.Test() {
public void run(TestResult result) {
throw Lang.noImplement();
}
public int countTestCases() {
return 1;
}
};
}
});
//来吧,执行之!!
for (Request request : reqs) {
request.getRunner().run(notifier);
}
//接下来,就是打印结果了.
System.out.printf("Run %d , Fail %d , Error %d \n", result.runCount(), result.failureCount(), result.errorCount());
if (result.failureCount() > 0) { //断言失败的TestCase
Enumeration enu = result.failures();
while (enu.hasMoreElements()) {
TestFailure testFailure = (TestFailure) enu.nextElement();
System.out.println("--Fail------------------------------------------------");
System.out.println(testFailure.trace());
testFailure.thrownException().printStackTrace(System.out);
}
}
if (result.errorCount() > 0) { //抛异常的TestCase
Enumeration enu = result.errors();
while (enu.hasMoreElements()) {
TestFailure testFailure = (TestFailure) enu.nextElement();
System.out.println("--ERROR------------------------------------------------");
System.out.println(testFailure.trace());
testFailure.thrownException().printStackTrace(System.out);
}
}