Add test.

raphw · Oct 31, 2024 · 810fd97 · 810fd97
1 parent 9e919eb
commit 810fd97
Showing 1 changed file with 43 additions and 0 deletions.
diff --git a/...ddy-dep/src/test/java/net/bytebuddy/agent/builder/AgentBuilderDefaultApplicationTest.java b/...ddy-dep/src/test/java/net/bytebuddy/agent/builder/AgentBuilderDefaultApplicationTest.java
@@ -7,6 +7,7 @@
 import net.bytebuddy.dynamic.DynamicType;
 import net.bytebuddy.dynamic.TargetType;
 import net.bytebuddy.dynamic.loading.ByteArrayClassLoader;
+import net.bytebuddy.dynamic.loading.ClassInjector;
 import net.bytebuddy.dynamic.loading.ClassLoadingStrategy;
 import net.bytebuddy.implementation.FixedValue;
 import net.bytebuddy.implementation.MethodDelegation;
@@ -838,6 +839,33 @@ public void testAdviceTransformer() throws Exception {
         }
     }
 
+    @Test
+    @IntegrationRule.Enforce
+    public void testAdviceTransformerWithInjection() throws Exception {
+        Instrumentation instrumentation = ByteBuddyAgent.install();
+        assertThat(instrumentation, instanceOf(Instrumentation.class));
+        ClassFileTransformer classFileTransformer = new AgentBuilder.Default()
+                .with(poolStrategy)
+                .with(new AgentBuilder.InjectionStrategy.UsingUnsafe.OfFactory(ClassInjector.UsingUnsafe.Factory.resolve(instrumentation)))
+                .ignore(none())
+                .with(AgentBuilder.InitializationStrategy.NoOp.INSTANCE)
+                .type(ElementMatchers.is(Foo.class), ElementMatchers.is(classLoader)).transform(new AgentBuilder.Transformer.ForAdvice()
+                        .with(poolStrategy)
+                        .with(AgentBuilder.LocationStrategy.ForClassLoader.STRONG)
+                        .include(BazAdvice.class.getClassLoader())
+                        .with(Assigner.DEFAULT)
+                        .withExceptionHandler(new Advice.ExceptionHandler.Simple(Removal.SINGLE))
+                        .advice(named(FOO), BazAdvice.class.getName())
+                        .auxiliary(BazAdviceAuxiliary.class.getName()))
+                .installOnByteBuddyAgent();
+        try {
+            Class<?> type = classLoader.loadClass(Foo.class.getName());
+            assertThat(type.getDeclaredMethod(FOO).invoke(type.getDeclaredConstructor().newInstance()), is((Object) (FOO + BAR)));
+        } finally {
+            assertThat(ByteBuddyAgent.getInstrumentation().removeTransformer(classFileTransformer), is(true));
+        }
+    }
+
     private static class FooTransformer implements AgentBuilder.Transformer {
 
         public DynamicType.Builder<?> transform(DynamicType.Builder<?> builder,
@@ -1016,4 +1044,19 @@ private static void exit(@Advice.Return(readOnly = false) String value) {
             value += QUX;
         }
     }
+
+    private static class BazAdvice {
+
+        @Advice.OnMethodExit
+        private static void exit(@Advice.Return(readOnly = false) String value) {
+            value += BazAdviceAuxiliary.value();
+        }
+    }
+
+    public static class BazAdviceAuxiliary {
+
+        public static String value() {
+            return BAR;
+        }
+    }
 }