io.leopard.test.IntegrationRunner Maven / Gradle / Ivy

Go to download
package io.leopard.test;

import java.io.IOException;
import java.io.InputStream;
import java.util.List;
import java.util.Properties;

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.springframework.test.context.TestContextManager;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
import org.springframework.transaction.annotation.Transactional;

public class IntegrationRunner extends SpringJUnit4ClassRunner {

	private Class clazz;

	public IntegrationRunner(Class clazz) throws InitializationError {
		super(clazz);
		this.clazz = clazz;
		// System.err.println("IntegrationRunner clazz:" + clazz.getName());

		// super.getTestContextManager();
		// TODO
		// UseH2 anno = clazz.getAnnotation(UseH2.class);
		// if (anno != null) {
		// boolean useH2 = anno.value();
		// boolean rollback = anno.rollback();
		// DefaultH2DataSource.setUseH2(useH2, "integration", !rollback);
		// }
	}

	@Override
	protected TestContextManager createTestContextManager(Class clazz) {
		// System.err.println("IntegrationRunner createTestContextManager clazz:" + clazz.getName());
		try {
			TestContextManager manager = super.createTestContextManager(clazz);
			// System.err.println("IntegrationRunner createTestContextManager manager:" + manager);
			return manager;
		}
		catch (RuntimeException e) {
			e.printStackTrace();
			throw e;
		}
	}

	protected boolean isOnlyTransactional() throws IOException {
		int testCount = this.getDescription().testCount();
		if (testCount == 1) {
			return false;
		}

		InputStream input = clazz.getResourceAsStream("/autounit.properties");
		Properties props = new Properties();
		props.load(input);
		input.close();
		String transactional = (String) props.get("transactional");
		return "true".equals(transactional);
	}

	@Override
	protected void runChild(final FrameworkMethod method, RunNotifier notifier) {
		boolean isOnlyTransactional;
		try {
			isOnlyTransactional = isOnlyTransactional();
			// System.err.println("isOnlyTransactional:" + isOnlyTransactional);
		}
		catch (IOException e) {
			isOnlyTransactional = false;
		}

		if (!isOnlyTransactional) {
			super.runChild(method, notifier);
			return;
		}

		Description description = describeChild(method);
		if (method.getAnnotation(Transactional.class) == null) {
			notifier.fireTestIgnored(description);
		}
		else {
			super.runChild(method, notifier);
		}
	}

	@Override
	protected void collectInitializationErrors(List errors) {
		super.collectInitializationErrors(errors);

	}

	@Override
	protected void validateTestMethods(List errors) {
		// 忽略测试代码合法性检查
		// super.validateTestMethods(errors);
	}

}