Transactional Unit Testing

TransactionalAnnotationTest.java

package com.wongnai.linter;

import java.util.Arrays;

import org.junit.Test;
import org.springframework.transaction.annotation.Transactional;

import com.tngtech.archunit.base.DescribedPredicate;
import com.tngtech.archunit.core.domain.JavaClass;
import com.tngtech.archunit.core.domain.JavaClasses;
import com.tngtech.archunit.core.domain.JavaMethod;
import com.tngtech.archunit.core.importer.ClassFileImporter;


public class TransactionalAnnotationTest {
	@Test
	public void testTransactionalWithCheckedExceptionMustSetRollbackFor() {
		JavaClasses importedClasses = new ClassFileImporter().importPackages("com.wongnai");

		importedClasses.stream()
			.flatMap(importedClass -> importedClass.getMethods().stream())
			.filter(method -> method.isAnnotatedWith(Transactional.class))
			.filter(this::isThrowCheckedException)
			.forEach(this::assertHasRollBackFor);
	}
	
	private void assertHasRollBackFor(JavaMethod checkedExceptionMethod) {
		Transactional tx = checkedExceptionMethod.getAnnotationOfType(Transactional.class);
		if (!Arrays.asList(tx.rollbackFor()).contains(Exception.class)) {
			String name = String.format("%s.%s", checkedExceptionMethod.getOwner().getName(), checkedExceptionMethod.getName());
			throw new RuntimeException("Method must contains rollback for: " + name);
		}
	}

	private boolean isThrowCheckedException(JavaMethod method) {
		return method.getThrowsClause().containsType(DescribedPredicate
				.describe("checked exception", JavaClass.Predicates.assignableTo(Exception.class)));
	}
}