如何在编译或运行时阻止项目测试类使用org.junit.After注解？

这个问题我之前帮团队解决过类似的，Spring AOP确实搞不定测试类——毕竟它们大多不会被Spring容器管理，切面根本切不到。给你几个靠谱的方案，分运行时拦截和编译时拦截两种，看你团队的偏好来选：

既然是JUnit的注解，直接用JUnit自己的扩展机制来拦截最直接，不管你的测试类是不是Spring Bean都能生效。

针对JUnit 4的实现 #

写一个自定义TestRule，在测试执行前扫描所有方法，一旦发现@After注解就抛出异常：

import org.junit.rules.TestRule;
import org.junit.runner.Description;
import org.junit.runners.model.Statement;
import java.lang.reflect.Method;

public class ForbidAfterAnnotationRule implements TestRule {
    @Override
    public Statement apply(Statement base, Description description) {
        return new Statement() {
            @Override
            public void evaluate() throws Throwable {
                // 扫描当前测试类的所有方法
                for (Method method : description.getTestClass().getDeclaredMethods()) {
                    if (method.isAnnotationPresent(org.junit.After.class)) {
                        throw new IllegalStateException(
                            "禁止在测试类中使用@After注解，会破坏测试逻辑！方法名：" + method.getName()
                        );
                    }
                }
                base.evaluate();
            }
        };
    }
}

然后在所有测试类（或者写一个基类让所有测试继承）里加上这个Rule：

@Rule
public ForbidAfterAnnotationRule forbidAfterRule = new ForbidAfterAnnotationRule();

针对JUnit 5的实现 #

用JUnit 5的Extension机制，实现BeforeAllCallback来提前检查：

import org.junit.jupiter.api.extension.BeforeAllCallback;
import org.junit.jupiter.api.extension.ExtensionContext;
import java.lang.reflect.Method;

public class ForbidAfterAnnotationExtension implements BeforeAllCallback {
    @Override
    public void beforeAll(ExtensionContext context) throws Exception {
        Class<?> testClass = context.getRequiredTestClass();
        for (Method method : testClass.getDeclaredMethods()) {
            if (method.isAnnotationPresent(org.junit.jupiter.api.AfterEach.class) 
                // 兼容JUnit 4的@After注解（如果项目里混着用的话）
                || method.isAnnotationPresent(org.junit.After.class)) {
                throw new IllegalStateException(
                    "禁止在测试类中使用@After/@AfterEach注解，会破坏测试逻辑！方法名：" + method.getName()
                );
            }
        }
    }
}

可以通过@ExtendWith注解把扩展加到单个测试类，或者在junit-platform.properties里全局配置，让所有测试类自动生效：

# 开启扩展自动检测，或者直接指定扩展类全限定名
junit.jupiter.extensions.classes=com.yourpackage.ForbidAfterAnnotationExtension

如果想在代码提交、编译阶段就把问题拦下来，避免等到运行测试才发现，推荐用ArchUnit——这是一个专门做架构约束检查的工具，能帮你在单元测试阶段扫描所有代码，强制执行规则。

先在构建文件里引入ArchUnit依赖，然后写一个架构测试：

import com.tngtech.archunit.core.domain.JavaClasses;
import com.tngtech.archunit.core.importer.ClassFileImporter;
import com.tngtech.archunit.lang.ArchRule;
import org.junit.jupiter.api.Test;
import static com.tngtech.archunit.lang.syntax.ArchRuleDefinition.noMethods;

public class TestAnnotationConstraintsTest {
    @Test
    public void testClassesShouldNotUseAfterAnnotation() {
        // 扫描所有测试类（替换成你项目的测试类包路径）
        JavaClasses importedClasses = new ClassFileImporter()
            .importPackages("com.yourpackage.test");

        // 定义规则：测试类中的方法不能标注@org.junit.After
        ArchRule rule = noMethods()
            .that().areDeclaredInClassesThat().haveSimpleNameEndingWith("Test")
            .should().beAnnotatedWith(org.junit.After.class);

        // 执行检查，违反规则会直接导致测试失败
        rule.check(importedClasses);
    }
}

这个测试会在你执行mvn test或gradle test时自动运行，一旦发现有测试类用了@After，构建就会失败，从根源上阻止违规代码合并。

如果想让整个构建流程强制检查，还可以用Maven Enforcer插件配合自定义规则，或者在Gradle里集成ArchUnit的检查任务。不过相对来说，ArchUnit的方案已经足够灵活，而且代码可读性更高，一般优先选它。

内容的提问来源于stack exchange，提问作者BartekN