如何通过反射等方式分析测试代码 排查单元测试违规启动SpringContext问题

有两种成熟的实现思路，你可以根据项目需求选择：

方案1：静态代码扫描（首选，性价比最高） #

用专门的架构校验工具ArchUnit直接写测试代码扫描所有测试类的字节码，检查标注@UnitTest的测试是否携带了Spring测试相关注解，不需要运行原有测试就能完成校验，完全符合你要的「用测试校验测试代码」的需求。

实现步骤： #

1. 引入ArchUnit依赖（以Maven为例）：

<dependency>
    <groupId>com.tngtech.archunit</groupId>
    <artifactId>archunit-junit5</artifactId>
    <version>1.2.1</version>
    <scope>test</scope>
</dependency>

2. 编写校验测试代码，示例如下：

import com.tngtech.archunit.core.domain.JavaClasses;
import com.tngtech.archunit.core.importer.ClassFileImporter;
import com.tngtech.archunit.lang.ArchRule;
import static com.tngtech.archunit.lang.syntax.ArchRuleDefinition.noClasses;
import static com.tngtech.archunit.lang.syntax.ArchRuleDefinition.noMethods;

@Test
void checkIfUnitTestsAreCorrect() {
    // 替换为你项目测试代码的根包路径
    JavaClasses testClasses = new ClassFileImporter().importPackages("com.your.project.test");
    
    // 校验规则：标注@UnitTest的类不能加任何Spring测试启动注解
    ArchRule classLevelRule = noClasses()
            .that().areAnnotatedWith("com.your.project.annotation.UnitTest") // 替换为@UnitTest的全限定名
            .should().beAnnotatedWith("org.springframework.boot.test.context.SpringBootTest")
            .orShould().beAnnotatedWith("org.springframework.test.context.ContextConfiguration")
            .orShould().beAnnotatedWith("org.springframework.boot.test.autoconfigure.web.servlet.WebMvcTest")
            // 其他你项目中会触发SpringContext启动的注解都可以依次添加
            .because("标注@UnitTest的单元测试不允许启动Spring上下文");
    
    // 校验规则：标注@UnitTest的方法所属的类不能加Spring测试启动注解
    ArchRule methodLevelRule = noMethods()
            .that().areAnnotatedWith("com.your.project.annotation.UnitTest")
            .should().beDeclaredInClassesThat().areAnnotatedWith("org.springframework.boot.test.context.SpringBootTest")
            .orShould().beDeclaredInClassesThat().areAnnotatedWith("org.springframework.test.context.ContextConfiguration")
            .because("标注@UnitTest的单元测试不允许启动Spring上下文");

    // 执行校验，违规时直接抛出异常终止测试，输出所有违规类/方法
    classLevelRule.check(testClasses);
    methodLevelRule.check(testClasses);
}

这个方案可以直接集成到CI流水线，每次代码提交自动运行，有违规直接阻断合并，扫描速度快，误判率低。

方案2：运行时兜底校验（适合需要拦截手动启动SpringContext的场景） #

如果存在开发通过手动编码new AnnotationConfigApplicationContext()的方式启动SpringContext的情况，静态扫描可能漏判，可以加一层运行时监听器做兜底：
实现一个Spring上下文启动监听器，只要SpringContext启动就检查当前运行的测试是否标注了@UnitTest，是则直接报错：

import org.springframework.context.ApplicationListener;
import org.springframework.context.event.ContextRefreshedEvent;
import java.lang.reflect.Method;

public class UnitTestSpringStartupCheckListener implements ApplicationListener<ContextRefreshedEvent> {
    @Override
    public void onApplicationEvent(ContextRefreshedEvent event) {
        StackTraceElement[] stackTrace = Thread.currentThread().getStackTrace();
        for (StackTraceElement stackFrame : stackTrace) {
            try {
                Class<?> testClazz = Class.forName(stackFrame.getClassName());
                Method testMethod = testClazz.getMethod(stackFrame.getMethodName());
                // 检查当前运行的测试是否标注@UnitTest
                if (testMethod.isAnnotationPresent(UnitTest.class) || testClazz.isAnnotationPresent(UnitTest.class)) {
                    throw new IllegalStateException(
                        String.format("违规检测：@UnitTest测试 %s.%s 不允许启动SpringContext", 
                        testClazz.getName(), testMethod.getName())
                    );
                }
            } catch (NoSuchMethodException | ClassNotFoundException e) {
                // 非测试方法的栈帧直接跳过
            }
        }
    }
}

然后将这个监听器注册到Spring的全局监听器列表，所有SpringContext启动都会触发校验。

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