com.github.database.rider.cdi.DBUnitInterceptorImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of rider-cdi Show documentation
There is a newer version: 1.44.0
package com.github.database.rider.cdi;

import java.io.Serializable;
import java.lang.reflect.Method;

import javax.inject.Inject;
import javax.interceptor.AroundInvoke;
import javax.interceptor.Interceptor;
import javax.interceptor.InvocationContext;
import javax.persistence.EntityManager;

import org.junit.After;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.AfterClass;
import com.github.database.rider.cdi.api.DBUnitInterceptor;
import com.github.database.rider.core.api.dataset.DataSet;
import com.github.database.rider.core.api.dataset.ExpectedDataSet;
import com.github.database.rider.core.api.leak.LeakHunter;
import com.github.database.rider.core.configuration.DBUnitConfig;
import com.github.database.rider.core.configuration.DataSetConfig;
import com.github.database.rider.core.leak.LeakHunterFactory;
import com.github.database.rider.core.util.AnnotationUtils;

/**
 * Created by pestano on 26/07/15.
 */
@Interceptor
@DBUnitInterceptor
public class DBUnitInterceptorImpl implements Serializable {

	@Inject
	DataSetProcessor dataSetProcessor;

	@Inject
	private EntityManager em;

	@AroundInvoke
	public Object intercept(InvocationContext invocationContext) throws Exception {

		Method method = invocationContext.getMethod();
		if(method.isAnnotationPresent(Before.class) || method.isAnnotationPresent(BeforeClass.class) || method.isAnnotationPresent(After.class) || method.isAnnotationPresent(AfterClass.class)) {
			return invocationContext.proceed();
		}
		Object proceed = null;
		DBUnitConfig dbUnitConfig = DBUnitConfig.from(invocationContext.getMethod());

		DataSet usingDataSet = resolveDataSet(invocationContext, dbUnitConfig);

		if (usingDataSet != null) {
			DataSetConfig dataSetConfig = new DataSetConfig(usingDataSet.value()).cleanAfter(usingDataSet.cleanAfter())
					.cleanBefore(usingDataSet.cleanBefore()).disableConstraints(usingDataSet.disableConstraints())
					.executeScripsBefore(usingDataSet.executeScriptsBefore())
					.executeScriptsAfter(usingDataSet.executeScriptsAfter())
					.executeStatementsAfter(usingDataSet.executeStatementsAfter())
					.executeStatementsBefore(usingDataSet.executeStatementsBefore()).strategy(usingDataSet.strategy())
					.transactional(usingDataSet.transactional()).tableOrdering(usingDataSet.tableOrdering())
					.useSequenceFiltering(usingDataSet.useSequenceFiltering());
			dataSetProcessor.process(dataSetConfig, dbUnitConfig);
			boolean isTransactionalTest = dataSetConfig.isTransactional();
			if (isTransactionalTest) {
				em.getTransaction().begin();
			}
			LeakHunter leakHunter = null;
			try {
				if (dbUnitConfig.isLeakHunter()) {
					leakHunter = LeakHunterFactory.from(dataSetProcessor.getDataSetExecutor().getRiderDataSource(),
							invocationContext.getMethod().getName());
					leakHunter.measureConnectionsBeforeExecution();
				}
				proceed = invocationContext.proceed();

				if (isTransactionalTest) {
					em.getTransaction().commit();
				}
				ExpectedDataSet expectedDataSet = invocationContext.getMethod().getAnnotation(ExpectedDataSet.class);
				if (expectedDataSet != null) {
					dataSetProcessor.compareCurrentDataSetWith(
							new DataSetConfig(expectedDataSet.value()).disableConstraints(true),
							expectedDataSet.ignoreCols());
				}
			} finally {
				if (isTransactionalTest && em.getTransaction().isActive()) {
					em.getTransaction().rollback();
				}

				if (leakHunter != null) {
					leakHunter.checkConnectionsAfterExecution();
				}

				dataSetProcessor.exportDataSet(invocationContext.getMethod());

				if (usingDataSet.executeStatementsAfter().length > 0) {
					dataSetProcessor.executeStatements(dataSetConfig.getExecuteStatementsAfter());
				}

				if (usingDataSet.executeScriptsAfter().length > 0) {
					for (int i = 0; i < usingDataSet.executeScriptsAfter().length; i++) {
						dataSetProcessor.executeScript(usingDataSet.executeScriptsAfter()[i]);
					}
				}

				if (usingDataSet.cleanAfter()) {
					dataSetProcessor.clearDatabase(dataSetConfig);
				}

				dataSetProcessor.enableConstraints();
				em.clear();
			} // end finally

		} else {// no dataset provided, just proceed and check expectedDataSet
			try {
				proceed = invocationContext.proceed();
				ExpectedDataSet expectedDataSet = invocationContext.getMethod().getAnnotation(ExpectedDataSet.class);
				if (expectedDataSet != null) {
					dataSetProcessor.compareCurrentDataSetWith(
							new DataSetConfig(expectedDataSet.value()).disableConstraints(true),
							expectedDataSet.ignoreCols());
				}
			} finally {
				dataSetProcessor.exportDataSet(invocationContext.getMethod());
			}

		}

		return proceed;
	}

	private DataSet resolveDataSet(InvocationContext invocationContext, DBUnitConfig config) {
		DataSet methodAnnotation = AnnotationUtils.findAnnotation(invocationContext.getMethod(), DataSet.class);
		DataSet classAnnotation = AnnotationUtils.findAnnotation(invocationContext.getMethod().getDeclaringClass(),
				DataSet.class);

		if (config.isMergeDataSets() && (classAnnotation != null && methodAnnotation != null)) {
			return AnnotationUtils.mergeDataSetAnnotations(classAnnotation, methodAnnotation);
		}

		if (methodAnnotation != null) {
			return methodAnnotation;
		}

		return classAnnotation;

	}

}