Spring Boot集成测试中如何重置ByteBuddy修改的类？

我们的Spring Boot应用通过ByteBuddy的annotateType(...)方法为类添加特定注解，示例代码如下：

private DynamicType.Loaded<? extends MyClassToRedefine> getLoadedTypeOfResource() {
    if (myClassToRedefineImplLoadedType == null) {
            try (var unloadedResource = new ByteBuddy()
                    .subclass(MyClassToRedefine.class)
                    .annotateType(newAnnotation1)
                    .annotateType(newAnnotation2)
                    .annotateType(newAnnotation3)
                    .make()) {
             myClassToRedefineImplLoadedType = unloadedResource.load(ClassLoader.getSystemClassLoader());
            }
    }
    return myClassToRedefineImplLoadedType;
}

测试使用的注解配置：

@SpringBootTest(properties = "spring.main.allow-bean-definition-overriding=true")
@EnableKubeAPIServer(updateKubeConfigFile = true)
@DirtiesContext

单独运行测试一切正常，但批量运行时，动态添加的注解在测试间保持不变，引发问题。根源是ByteBuddy在JVM层面修改类，这些修改会在多轮测试中持续存在，而@DirtiesContext只能重置Spring上下文，管不了JVM类加载的残留。

1. 给每个测试用独立类加载器 #

别用系统类加载器，改用ByteBuddy的包装类加载器策略，让每个测试的动态类加载在独立的类加载器中，测试结束后类加载器可被GC回收，避免跨测试残留。修改代码：

private DynamicType.Loaded<? extends MyClassToRedefine> getLoadedTypeOfResource() {
    try (var unloadedResource = new ByteBuddy()
            .subclass(MyClassToRedefine.class)
            .annotateType(newAnnotation1)
            .annotateType(newAnnotation2)
            .annotateType(newAnnotation3)
            .make()) {
        // 使用包装类加载器替代系统类加载器
        return unloadedResource.load(getClass().getClassLoader(), ClassLoadingStrategy.Default.WRAPPER);
    }
}

同时去掉类级别的缓存判断（if (myClassToRedefineImplLoadedType == null)），确保每次测试都生成新的动态类。

2. 重置ByteBuddy缓存变量 #

如果必须复用动态类，在每个测试前后重置缓存的myClassToRedefineImplLoadedType变量。用JUnit的@AfterEach注解实现：

@AfterEach
void resetByteBuddyCache() {
    // 假设该变量是静态的，直接置空
    MyByteBuddyUtil.myClassToRedefineImplLoadedType = null;
}

3. 扩大@DirtiesContext的作用范围 #

把默认的方法级@DirtiesContext改成类级，指定每个测试方法结束后销毁上下文，虽然这不能直接解决类加载器的问题，但能配合前两点彻底清除Spring层面的残留：

@SpringBootTest(properties = "spring.main.allow-bean-definition-overriding=true")
@EnableKubeAPIServer(updateKubeConfigFile = true)
@DirtiesContext(classMode = DirtiesContext.ClassMode.AFTER_EACH_TEST_METHOD)
public class MyTest {
    // 测试代码
}

4. 若场景允许，用重定义+恢复原始类 #

如果是修改已有类而非创建子类，测试前保存原始类定义，测试后用ByteBuddy重新恢复：

private Class<?> originalMyClass;

@BeforeEach
void saveOriginalClass() {
    originalMyClass = MyClassToRedefine.class;
}

@AfterEach
void restoreOriginalClass() throws Exception {
    // 重新加载原始类定义，覆盖修改后的版本
    new ByteBuddy().redefine(originalMyClass).make()
        .load(originalMyClass.getClassLoader(), ClassLoadingStrategy.Default.REDEFINITION);
}

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