com.github.bookong.zest.ZestSpringJUnit4ClassRunner Maven / Gradle / Ivy

Go to download
package com.github.bookong.zest;

import java.lang.reflect.Field;
import java.util.List;

import javax.sql.DataSource;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.junit.Ignore;
import org.junit.Test;
import org.junit.runner.Description;
import org.junit.runner.notification.RunNotifier;
import org.junit.runners.model.FrameworkMethod;
import org.junit.runners.model.InitializationError;
import org.junit.runners.model.Statement;
import org.springframework.jdbc.datasource.DataSourceUtils;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

import com.github.bookong.zest.core.ZestLauncher;
import com.github.bookong.zest.core.ZestStatement;
import com.github.bookong.zest.core.annotations.ZestTest;
import com.github.bookong.zest.core.annotations.ZestDataSource;
import com.github.bookong.zest.util.ReflectHelper;

/**
 * @author jiangxu
 *
 */
public class ZestSpringJUnit4ClassRunner extends SpringJUnit4ClassRunner {
	protected static final Log logger = LogFactory.getLog(ZestSpringJUnit4ClassRunner.class);
	protected ZestLauncher zestLauncher = new ZestLauncher();

	public ZestSpringJUnit4ClassRunner(Class clazz) throws InitializationError {
		super(clazz);
		zestLauncher.init(clazz);
	}

	/**
	 * 忽略没有 Test 注解的方法这种错误，因为我们的测试方法是用 MTest 标识的
	 */
	@Override
	protected void collectInitializationErrors(List errors) {
		super.collectInitializationErrors(errors);
		ZestLauncher.ignoreNoRunnableMethodsError(errors);
	}

	/**
	 * 让 JUnit 接受 MTest 注解的方法
	 */
	@Override
	protected List getChildren() {
		List list = computeTestMethods();
		list.addAll(getTestClass().getAnnotatedMethods(ZestTest.class));
		return list;
	}

	/**
	 * 如果是用 MTest 注解的方法，用 minonLauncher 来执行
	 */
	@Override
	protected void runChild(FrameworkMethod frameworkMethod, RunNotifier notifier) {
		ZestTest ztest = frameworkMethod.getAnnotation(ZestTest.class);
		if (ztest == null) {
			super.runChild(frameworkMethod, notifier);
		} else {
			Description description = describeChild(frameworkMethod);
			if (frameworkMethod.getAnnotation(Ignore.class) != null) {
				notifier.fireTestIgnored(description);
			} else {
				zestLauncher.run(frameworkMethod, methodBlock(frameworkMethod), description, notifier);
			}
		}
	}

	/**
	 * Returns a {@link Statement} that invokes {@code method} on {@code test}
	 */
	@Override
	protected Statement methodInvoker(FrameworkMethod method, Object test) {
		ZestTest ztest = method.getAnnotation(ZestTest.class);
		if (ztest == null) {
			return super.methodInvoker(method, test);
		} else {
			for (Field f : test.getClass().getDeclaredFields()) {
				ZestDataSource zestDataSource = f.getAnnotation(ZestDataSource.class);
				if (zestDataSource != null) {
					Object obj = ReflectHelper.getValueByFieldName(test, f.getName());
					if (obj instanceof DataSource) {
						zestLauncher.setConnection(zestDataSource.value(),
								DataSourceUtils.getConnection((DataSource) obj));
						try {
							zestLauncher.setExecuter(zestDataSource.value(), zestDataSource.executerClazz()
									.newInstance());
						} catch (Exception e) {
							throw new RuntimeException("Fail to set executer. Executer class:"
									+ zestDataSource.executerClazz().getName(), e);
						}
					}
				}
			}

			return new ZestStatement(getTestClass(), method, test, zestLauncher);
		}
	}
}