如何用ByteBuddy获取AbstractAssert的protected字段actual值？遇类加载问题

错误原因 #

你遇到的Class already loaded: class org.assertj.core.api.AbstractAssert错误，是因为ByteBuddy的redefine方法默认只能修改未被类加载器加载的类。而JUnit测试场景中，AssertJ的核心类通常在你的代码执行前就已经被系统类加载器加载完成，直接调用redefine会触发这个限制。

解决方案 #

针对你的JUnit测试报告需求，推荐两种可行方案：

方案一：Java Agent运行时修改已加载类（优先推荐） #

通过Java Agent机制可以在类加载后动态修改字节码，适配AssertJ类已加载的场景。

1. 编写Advice捕获protected字段
   用ByteBuddy的@Advice.FieldValue可以直接访问目标类的protected字段，无需反射：

   public class AssertAdvice {
       @Advice.OnMethodExit
       public static void captureActual(@Advice.FieldValue("actual") Object actualValue) {
           // 这里执行你的测试报告逻辑，比如记录actual值
           System.out.println("捕获到AssertJ actual字段值：" + actualValue);
       }
   }
   

2. 实现Java Agent
   编写Agent类，指定要修改的类和对应的Advice：

   public class AssertAgent {
       public static void premain(String agentArgs, Instrumentation inst) {
           new AgentBuilder.Default()
                   .type(ElementMatchers.named("org.assertj.core.api.AbstractAssert"))
                   .transform((builder, typeDesc, classLoader, module) ->
                           builder.visit(Advice.to(AssertAdvice.class).on(ElementMatchers.isMethod()))
                   )
                   .installOn(inst);
       }
   }
   

3. 打包Agent并配置JUnit运行参数

   • 打包Agent jar时，在MANIFEST.MF中添加：

     Premain-Class: com.yourpackage.AssertAgent
     Can-Redefine-Classes: true
     

   • 在JUnit运行的VM参数中添加：

     -javaagent:/path/to/your-agent.jar
     

方案二：自定义类加载器提前修改类（仅适用于类未加载场景） #

如果能确保AbstractAssert还未被系统类加载器加载，可以用自定义类加载器加载修改后的类：

public class AssertClassLoader extends ClassLoader {
    public AssertClassLoader(ClassLoader parent) {
        super(parent);
    }

    public Class<?> loadModifiedAssertClass() throws IOException, ClassNotFoundException {
        byte[] modifiedBytes = new ByteBuddy()
                .redefine(AbstractAssert.class)
                .visit(Advice.to(AssertAdvice.class).on(ElementMatchers.isMethod()))
                .make()
                .getBytes();
        return defineClass("org.assertj.core.api.AbstractAssert", modifiedBytes, 0, modifiedBytes.length);
    }
}

使用时需在AssertJ类加载前执行：

AssertClassLoader classLoader = new AssertClassLoader(Thread.currentThread().getContextClassLoader());
Class<?> modifiedAssertClass = classLoader.loadModifiedAssertClass();
// 后续通过该类加载器使用AssertJ API，即可触发Advice逻辑

核心注意事项 #

• 访问protected字段：ByteBuddy的@Advice.FieldValue可以直接注入目标类的protected字段，无需手动设置setAccessible(true)，只要字段名称匹配即可。
• 类加载优先级：系统类加载器加载的类无法被自定义类加载器覆盖，因此JUnit场景下优先选择Java Agent方案。

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