package net.bytebuddy.implementation.bind.annotation;
import net.bytebuddy.description.field.FieldDescription;
import net.bytebuddy.description.field.FieldList;
import net.bytebuddy.description.method.MethodDescription;
import net.bytebuddy.description.type.TypeDescription;
import net.bytebuddy.description.type.TypeList;
import net.bytebuddy.implementation.Implementation;
import net.bytebuddy.implementation.bind.MethodDelegationBinder;
import net.bytebuddy.implementation.bytecode.assign.Assigner;
import net.bytebuddy.test.utility.ObjectPropertyAssertion;
import org.junit.Before;
import org.junit.Test;
import org.mockito.Mock;
import static net.bytebuddy.matcher.ElementMatchers.named;
import static org.hamcrest.CoreMatchers.is;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.mockito.Mockito.*;
public class MorphBinderTest extends AbstractAnnotationBinderTest<Morph> {
@Mock
private MethodDescription morphMethod;
@Mock
private MethodDescription.SignatureToken morphToken;
@Mock
private TypeDescription morphType, defaultType;
@Mock
private TypeDescription.Generic genericMorphType;
@Mock
private MethodDescription.SignatureToken sourceToken;
@Mock
private Implementation.SpecialMethodInvocation specialMethodInvocation;
public MorphBinderTest() {
super(Morph.class);
}
@Override
protected TargetMethodAnnotationDrivenBinder.ParameterBinder<Morph> getSimpleBinder() {
return new Morph.Binder(morphMethod);
}
@Override
@Before
public void setUp() throws Exception {
super.setUp();
when(genericMorphType.asErasure()).thenReturn(morphType);
when(defaultType.asErasure()).thenReturn(defaultType);
when(source.asSignatureToken()).thenReturn(sourceToken);
}
@Test(expected = IllegalStateException.class)
public void testIllegalType() throws Exception {
when(target.getType()).thenReturn(genericMorphType);
when(morphMethod.getDeclaringType()).thenReturn(mock(TypeDescription.class));
new Morph.Binder(morphMethod).bind(annotationDescription, source, target, implementationTarget, assigner, Assigner.Typing.STATIC);
}
@Test
public void testSuperMethodCallInvalid() throws Exception {
when(target.getType()).thenReturn(genericMorphType);
when(morphMethod.getDeclaringType()).thenReturn(morphType);
doReturn(void.class).when(annotation).defaultTarget();
when(implementationTarget.invokeSuper(sourceToken)).thenReturn(specialMethodInvocation);
MethodDelegationBinder.ParameterBinding<?> parameterBinding = new Morph.Binder(morphMethod)
.bind(annotationDescription, source, target, implementationTarget, assigner, Assigner.Typing.STATIC);
assertThat(parameterBinding.isValid(), is(false));
verify(specialMethodInvocation).isValid();
}
@Test
public void testSuperMethodCallValid() throws Exception {
when(target.getType()).thenReturn(genericMorphType);
when(morphMethod.getDeclaringType()).thenReturn(morphType);
doReturn(void.class).when(annotation).defaultTarget();
when(implementationTarget.invokeSuper(sourceToken)).thenReturn(specialMethodInvocation);
when(specialMethodInvocation.isValid()).thenReturn(true);
MethodDelegationBinder.ParameterBinding<?> parameterBinding = new Morph.Binder(morphMethod)
.bind(annotationDescription, source, target, implementationTarget, assigner, Assigner.Typing.STATIC);
assertThat(parameterBinding.isValid(), is(true));
verify(specialMethodInvocation).isValid();
}
@Test
public void testDefaultMethodCallImplicitInvalid() throws Exception {
when(source.asSignatureToken()).thenReturn(morphToken);
when(target.getType()).thenReturn(genericMorphType);
when(morphMethod.getDeclaringType()).thenReturn(morphType);
when(annotation.defaultMethod()).thenReturn(true);
doReturn(void.class).when(annotation).defaultTarget();
when(implementationTarget.invokeDefault(morphToken)).thenReturn(specialMethodInvocation);
MethodDelegationBinder.ParameterBinding<?> parameterBinding = new Morph.Binder(morphMethod)
.bind(annotationDescription, source, target, implementationTarget, assigner, Assigner.Typing.STATIC);
assertThat(parameterBinding.isValid(), is(false));
verify(specialMethodInvocation).isValid();
}
@Test
public void testDefaultMethodCallImplicitValid() throws Exception {
when(source.asSignatureToken()).thenReturn(morphToken);
when(target.getType()).thenReturn(genericMorphType);
when(morphMethod.getDeclaringType()).thenReturn(morphType);
when(annotation.defaultMethod()).thenReturn(true);
doReturn(void.class).when(annotation).defaultTarget();
when(implementationTarget.invokeDefault(morphToken)).thenReturn(specialMethodInvocation);
when(specialMethodInvocation.isValid()).thenReturn(true);
MethodDelegationBinder.ParameterBinding<?> parameterBinding = new Morph.Binder(morphMethod)
.bind(annotationDescription, source, target, implementationTarget, assigner, Assigner.Typing.STATIC);
assertThat(parameterBinding.isValid(), is(true));
verify(specialMethodInvocation).isValid();
}
@Test
public void testDefaultMethodCallExplicitInvalid() throws Exception {
when(source.asSignatureToken()).thenReturn(morphToken);
when(instrumentedType.getInterfaces()).thenReturn(new TypeList.Generic.ForLoadedTypes(Foo.class));
when(target.getType()).thenReturn(genericMorphType);
when(morphMethod.getDeclaringType()).thenReturn(morphType);
when(annotation.defaultMethod()).thenReturn(true);
doReturn(Foo.class).when(annotation).defaultTarget();
when(implementationTarget.invokeDefault(morphToken, new TypeDescription.ForLoadedType(Foo.class)))
.thenReturn(specialMethodInvocation);
MethodDelegationBinder.ParameterBinding<?> parameterBinding = new Morph.Binder(morphMethod)
.bind(annotationDescription, source, target, implementationTarget, assigner, Assigner.Typing.STATIC);
assertThat(parameterBinding.isValid(), is(false));
verify(specialMethodInvocation).isValid();
}
@Test
public void testDefaultMethodCallExplicitValid() throws Exception {
when(source.asSignatureToken()).thenReturn(morphToken);
when(instrumentedType.getInterfaces()).thenReturn(new TypeList.Generic.ForLoadedTypes(Foo.class));
when(target.getType()).thenReturn(genericMorphType);
when(morphMethod.getDeclaringType()).thenReturn(morphType);
when(annotation.defaultMethod()).thenReturn(true);
doReturn(Foo.class).when(annotation).defaultTarget();
when(implementationTarget.invokeDefault(morphToken, new TypeDescription.ForLoadedType(Foo.class)))
.thenReturn(specialMethodInvocation);
when(specialMethodInvocation.isValid()).thenReturn(true);
MethodDelegationBinder.ParameterBinding<?> parameterBinding = new Morph.Binder(morphMethod)
.bind(annotationDescription, source, target, implementationTarget, assigner, Assigner.Typing.STATIC);
assertThat(parameterBinding.isValid(), is(true));
verify(specialMethodInvocation).isValid();
}
@Test
@SuppressWarnings("unchecked")
public void testObjectProperties() throws Exception {
ObjectPropertyAssertion.of(Morph.Binder.class).apply();
ObjectPropertyAssertion.of(Morph.Binder.RedirectionProxy.class).apply();
ObjectPropertyAssertion.of(Morph.Binder.RedirectionProxy.MethodCall.class).apply();
ObjectPropertyAssertion.of(Morph.Binder.RedirectionProxy.MethodCall.Appender.class).refine(new ObjectPropertyAssertion.Refinement<Implementation.Target>() {
@Override
public void apply(Implementation.Target mock) {
when(mock.getInstrumentedType()).thenReturn(mock(TypeDescription.class));
}
}).apply();
ObjectPropertyAssertion.of(Morph.Binder.RedirectionProxy.StaticFieldConstructor.class).apply();
ObjectPropertyAssertion.of(Morph.Binder.RedirectionProxy.InstanceFieldConstructor.class).apply();
ObjectPropertyAssertion.of(Morph.Binder.RedirectionProxy.InstanceFieldConstructor.Appender.class).refine(new ObjectPropertyAssertion.Refinement<Implementation.Target>() {
@Override
public void apply(Implementation.Target mock) {
TypeDescription typeDescription = mock(TypeDescription.class);
FieldList<?> fieldList = mock(FieldList.class);
FieldList<?> filteredFieldList = mock(FieldList.class);
when(fieldList.filter(named(Morph.Binder.RedirectionProxy.FIELD_NAME))).thenReturn((FieldList) filteredFieldList);
when(filteredFieldList.getOnly()).thenReturn(mock(FieldDescription.class));
when(typeDescription.getDeclaredFields()).thenReturn((FieldList) fieldList);
when(mock.getInstrumentedType()).thenReturn(typeDescription);
}
}).apply();
ObjectPropertyAssertion.of(Morph.Binder.DefaultMethodLocator.Implicit.class).apply();
ObjectPropertyAssertion.of(Morph.Binder.DefaultMethodLocator.Explicit.class).apply();
}
private interface Foo {
}
}