How to use MockMethodInterceptor class of org.mockito.internal.creation.bytebuddy package

Best Mockito code snippet using org.mockito.internal.creation.bytebuddy.MockMethodInterceptor

Source:InlineByteBuddyMockMaker.java

...154        INSTRUMENTATION = instrumentation;155        INITIALIZATION_ERROR = initializationError;156    }157    private final BytecodeGenerator bytecodeGenerator;158    private final WeakConcurrentMap<Object, MockMethodInterceptor> mocks = new WeakConcurrentMap.WithInlinedExpunction<Object, MockMethodInterceptor>();159    public InlineByteBuddyMockMaker() {160        if (INITIALIZATION_ERROR != null) {161            throw new MockitoInitializationException(join(162                    "Could not initialize inline Byte Buddy mock maker. (This mock maker is not supported on Android.)",163                    "",164                    Platform.describe()), INITIALIZATION_ERROR);165        }166        bytecodeGenerator = new TypeCachingBytecodeGenerator(new InlineBytecodeGenerator(INSTRUMENTATION, mocks), true);167    }168    @Override169    public <T> T createMock(MockCreationSettings<T> settings, MockHandler handler) {170        Class<? extends T> type = createMockType(settings);171        Instantiator instantiator = Plugins.getInstantiatorProvider().getInstantiator(settings);172        try {173            T instance = instantiator.newInstance(type);174            MockMethodInterceptor mockMethodInterceptor = new MockMethodInterceptor(handler, settings);175            mocks.put(instance, mockMethodInterceptor);176            if (instance instanceof MockAccess) {177                ((MockAccess) instance).setMockitoInterceptor(mockMethodInterceptor);178            }179            return instance;180        } catch (org.mockito.creation.instance.InstantiationException e) {181            throw new MockitoException("Unable to create mock instance of type '" + type.getSimpleName() + "'", e);182        }183    }184    @Override185    public <T> Class<? extends T> createMockType(MockCreationSettings<T> settings) {186        try {187            return bytecodeGenerator.mockClass(MockFeatures.withMockFeatures(188                    settings.getTypeToMock(),189                    settings.getExtraInterfaces(),190                    settings.getSerializableMode(),191                    settings.isStripAnnotations()192            ));193        } catch (Exception bytecodeGenerationFailed) {194            throw prettifyFailure(settings, bytecodeGenerationFailed);195        }196    }197    private <T> RuntimeException prettifyFailure(MockCreationSettings<T> mockFeatures, Exception generationFailed) {198        if (mockFeatures.getTypeToMock().isArray()) {199            throw new MockitoException(join(200                    "Arrays cannot be mocked: " + mockFeatures.getTypeToMock() + ".",201                    ""202            ), generationFailed);203        }204        if (Modifier.isFinal(mockFeatures.getTypeToMock().getModifiers())) {205            throw new MockitoException(join(206                    "Mockito cannot mock this class: " + mockFeatures.getTypeToMock() + ".",207                    "Can not mock final classes with the following settings :",208                    " - explicit serialization (e.g. withSettings().serializable())",209                    " - extra interfaces (e.g. withSettings().extraInterfaces(...))",210                    "",211                    "You are seeing this disclaimer because Mockito is configured to create inlined mocks.",212                    "You can learn about inline mocks and their limitations under item #39 of the Mockito class javadoc.",213                    "",214                    "Underlying exception : " + generationFailed215            ), generationFailed);216        }217        if (Modifier.isPrivate(mockFeatures.getTypeToMock().getModifiers())) {218            throw new MockitoException(join(219                    "Mockito cannot mock this class: " + mockFeatures.getTypeToMock() + ".",220                    "Most likely it is a private class that is not visible by Mockito",221                    "",222                    "You are seeing this disclaimer because Mockito is configured to create inlined mocks.",223                    "You can learn about inline mocks and their limitations under item #39 of the Mockito class javadoc.",224                    ""225            ), generationFailed);226        }227        throw new MockitoException(join(228                "Mockito cannot mock this class: " + mockFeatures.getTypeToMock() + ".",229                "",230                "If you're not sure why you're getting this error, please report to the mailing list.",231                "",232                Platform.warnForVM(233                        "IBM J9 VM", "Early IBM virtual machine are known to have issues with Mockito, please upgrade to an up-to-date version.\n",234                        "Hotspot", Platform.isJava8BelowUpdate45() ? "Java 8 early builds have bugs that were addressed in Java 1.8.0_45, please update your JDK!\n" : ""235                ),236                Platform.describe(),237                "",238                "You are seeing this disclaimer because Mockito is configured to create inlined mocks.",239                "You can learn about inline mocks and their limitations under item #39 of the Mockito class javadoc.",240                "",241                "Underlying exception : " + generationFailed242        ), generationFailed);243    }244    @Override245    public MockHandler getHandler(Object mock) {246        MockMethodInterceptor interceptor = mocks.get(mock);247        if (interceptor == null) {248            return null;249        } else {250            return interceptor.handler;251        }252    }253    @Override254    public void resetMock(Object mock, MockHandler newHandler, MockCreationSettings settings) {255        MockMethodInterceptor mockMethodInterceptor = new MockMethodInterceptor(newHandler, settings);256        mocks.put(mock, mockMethodInterceptor);257        if (mock instanceof MockAccess) {258            ((MockAccess) mock).setMockitoInterceptor(mockMethodInterceptor);259        }260    }261    @Override262    public TypeMockability isTypeMockable(final Class<?> type) {263        return new TypeMockability() {264            @Override265            public boolean mockable() {266                return INSTRUMENTATION.isModifiableClass(type) && !EXCLUDES.contains(type);267            }268            @Override269            public String nonMockableReason() {...

Source:SubclassBytecodeGenerator.java

...15import net.bytebuddy.implementation.attribute.MethodAttributeAppender;16import net.bytebuddy.matcher.ElementMatcher;17import org.mockito.exceptions.base.MockitoException;18import org.mockito.internal.creation.bytebuddy.ByteBuddyCrossClassLoaderSerializationSupport.CrossClassLoaderSerializableMock;19import org.mockito.internal.creation.bytebuddy.MockMethodInterceptor.DispatcherDefaultingToRealMethod;20import org.mockito.mock.SerializableMode;21import java.io.IOException;22import java.io.ObjectInputStream;23import java.lang.annotation.Annotation;24import java.lang.reflect.Modifier;25import java.lang.reflect.Type;26import java.util.ArrayList;27import java.util.Random;28import static java.lang.Thread.currentThread;29import static net.bytebuddy.description.modifier.Visibility.PRIVATE;30import static net.bytebuddy.dynamic.Transformer.ForMethod.withModifiers;31import static net.bytebuddy.implementation.MethodDelegation.to;32import static net.bytebuddy.implementation.attribute.MethodAttributeAppender.ForInstrumentedMethod.INCLUDING_RECEIVER;33import static net.bytebuddy.matcher.ElementMatchers.*;34import static org.mockito.internal.util.StringUtil.join;35class SubclassBytecodeGenerator implements BytecodeGenerator {36    private final SubclassLoader loader;37    private final ByteBuddy byteBuddy;38    private final Random random;39    private final Implementation readReplace;40    private final ElementMatcher<? super MethodDescription> matcher;41    public SubclassBytecodeGenerator() {42        this(new SubclassInjectionLoader());43    }44    public SubclassBytecodeGenerator(SubclassLoader loader) {45        this(loader, null, any());46    }47    public SubclassBytecodeGenerator(Implementation readReplace, ElementMatcher<? super MethodDescription> matcher) {48        this(new SubclassInjectionLoader(), readReplace, matcher);49    }50    protected SubclassBytecodeGenerator(SubclassLoader loader, Implementation readReplace, ElementMatcher<? super MethodDescription> matcher) {51        this.loader = loader;52        this.readReplace = readReplace;53        this.matcher = matcher;54        byteBuddy = new ByteBuddy().with(TypeValidation.DISABLED);55        random = new Random();56    }57    @Override58    public <T> Class<? extends T> mockClass(MockFeatures<T> features) {59        DynamicType.Builder<T> builder =60                byteBuddy.subclass(features.mockedType)61                         .name(nameFor(features.mockedType))62                         .ignoreAlso(isGroovyMethod())63                         .annotateType(features.stripAnnotations64                             ? new Annotation[0]65                             : features.mockedType.getAnnotations())66                         .implement(new ArrayList<Type>(features.interfaces))67                         .method(matcher)68                           .intercept(to(DispatcherDefaultingToRealMethod.class))69                           .transform(withModifiers(SynchronizationState.PLAIN))70                           .attribute(features.stripAnnotations71                               ? MethodAttributeAppender.NoOp.INSTANCE72                               : INCLUDING_RECEIVER)73                         .method(isHashCode())74                           .intercept(to(MockMethodInterceptor.ForHashCode.class))75                         .method(isEquals())76                           .intercept(to(MockMethodInterceptor.ForEquals.class))77                         .serialVersionUid(42L)78                         .defineField("mockitoInterceptor", MockMethodInterceptor.class, PRIVATE)79                         .implement(MockAccess.class)80                           .intercept(FieldAccessor.ofBeanProperty());81        if (features.serializableMode == SerializableMode.ACROSS_CLASSLOADERS) {82            builder = builder.implement(CrossClassLoaderSerializableMock.class)83                             .intercept(to(MockMethodInterceptor.ForWriteReplace.class));84        }85        if (readReplace != null) {86            builder = builder.defineMethod("readObject", void.class, Visibility.PRIVATE)87                    .withParameters(ObjectInputStream.class)88                    .throwing(ClassNotFoundException.class, IOException.class)89                    .intercept(readReplace);90        }91        ClassLoader classLoader = new MultipleParentClassLoader.Builder()92            .append(features.mockedType)93            .append(features.interfaces)94            .append(currentThread().getContextClassLoader())95            .append(MockAccess.class, DispatcherDefaultingToRealMethod.class)96            .append(MockMethodInterceptor.class,97                MockMethodInterceptor.ForHashCode.class,98                MockMethodInterceptor.ForEquals.class).build(MockMethodInterceptor.class.getClassLoader());99        if (classLoader != features.mockedType.getClassLoader()) {100            assertVisibility(features.mockedType);101            for (Class<?> iFace : features.interfaces) {102                assertVisibility(iFace);103            }104            builder = builder.ignoreAlso(isPackagePrivate()105                .or(returns(isPackagePrivate()))106                .or(hasParameters(whereAny(hasType(isPackagePrivate())))));107        }108        return builder.make()109                      .load(classLoader, loader.getStrategy(features.mockedType))110                      .getLoaded();111    }112    private static ElementMatcher<MethodDescription> isGroovyMethod() {...

Source:Sample_2.java

...22import net.bytebuddy.implementation.FieldAccessor;23import net.bytebuddy.implementation.Implementation;24import net.bytebuddy.matcher.ElementMatcher;25import org.mockito.internal.creation.bytebuddy.ByteBuddyCrossClassLoaderSerializationSupport.CrossClassLoaderSerializableMock;26import org.mockito.internal.creation.bytebuddy.MockMethodInterceptor.DispatcherDefaultingToRealMethod;27import org.mockito.mock.SerializableMode;28import java.io.IOException;29import java.io.ObjectInputStream;30import java.lang.reflect.Type;31import java.util.ArrayList;32import java.util.Random;33import static java.lang.Thread.currentThread;34import static net.bytebuddy.description.modifier.Visibility.PRIVATE;35import static net.bytebuddy.dynamic.Transformer.ForMethod.withModifiers;36import static net.bytebuddy.implementation.MethodDelegation.to;37import static net.bytebuddy.implementation.attribute.MethodAttributeAppender.ForInstrumentedMethod.INCLUDING_RECEIVER;38import static net.bytebuddy.matcher.ElementMatchers.*;39class SubclassBytecodeGenerator implements BytecodeGenerator {40    private final ByteBuddy byteBuddy;41    private final Random random;42    private final Implementation readReplace;43    private final ElementMatcher<? super MethodDescription> matcher;44    public SubclassBytecodeGenerator() {45        this(null, any());46    }47    public SubclassBytecodeGenerator(Implementation readReplace, ElementMatcher<? super MethodDescription> matcher) {48        this.readReplace = readReplace;49        this.matcher = matcher;50        byteBuddy = new ByteBuddy().with(TypeValidation.DISABLED);51        random = new Random();52    }53    @Override54    public <T> Class<? extends T> mockClass(MockFeatures<T> features) {55        DynamicType.Builder<T> builder =56                byteBuddy.subclass(features.mockedType)57                         .name(nameFor(features.mockedType))58                         .ignoreAlso(isGroovyMethod())59                         .annotateType(features.mockedType.getAnnotations())60                         .implement(new ArrayList<Type>(features.interfaces))61                         .method(matcher)62                           .intercept(to(DispatcherDefaultingToRealMethod.class))63                           .transform(withModifiers(SynchronizationState.PLAIN))64                           .attribute(INCLUDING_RECEIVER)65                         .method(isHashCode())66                           .intercept(to(MockMethodInterceptor.ForHashCode.class))67                         .method(isEquals())68                           .intercept(to(MockMethodInterceptor.ForEquals.class))69                         .serialVersionUid(42L)70                         .defineField("mockitoInterceptor", MockMethodInterceptor.class, PRIVATE)71                         .implement(MockAccess.class)72                           .intercept(FieldAccessor.ofBeanProperty());73        if (features.serializableMode == SerializableMode.ACROSS_CLASSLOADERS) {74            builder = builder.implement(CrossClassLoaderSerializableMock.class)75                             .intercept(to(MockMethodInterceptor.ForWriteReplace.class));76        }77        if (readReplace != null) {78            builder = builder.defineMethod("readObject", void.class, Visibility.PRIVATE)79                    .withParameters(ObjectInputStream.class)80                    .throwing(ClassNotFoundException.class, IOException.class)81                    .intercept(readReplace);82        }83        return builder.make()84                      .load(new MultipleParentClassLoader.Builder()85                              .append(features.mockedType)86                              .append(features.interfaces)87                              .append(currentThread().getContextClassLoader())88                              .append(MockAccess.class, DispatcherDefaultingToRealMethod.class)89                              .append(MockMethodInterceptor.class,90                                      MockMethodInterceptor.ForHashCode.class,91                                      MockMethodInterceptor.ForEquals.class).build(),92                              ClassLoadingStrategy.Default.INJECTION.with(features.mockedType.getProtectionDomain()))93                      .getLoaded();94    }95    private static ElementMatcher<MethodDescription> isGroovyMethod() {96        return isDeclaredBy(named("groovy.lang.GroovyObjectSupport"));97    }98    // TODO inspect naming strategy (for OSGI, signed package, java.* (and bootstrap classes), etc...)99    private String nameFor(Class<?> type) {100        String typeName = type.getName();101        if (isComingFromJDK(type)102                || isComingFromSignedJar(type)103                || isComingFromSealedPackage(type)) {104            typeName = "codegen." + typeName;105        }...

Source:20ByteBuddyMockMaker.java

...27        Instantiator instantiator = new InstantiatorProvider().getInstantiator(settings);28        T mockInstance = null;29        try {30            mockInstance = instantiator.newInstance(mockedProxyType);31            MockMethodInterceptor.MockAccess mockAccess = (MockMethodInterceptor.MockAccess) mockInstance;32            mockAccess.setMockitoInterceptor(new MockMethodInterceptor(asInternalMockHandler(handler), settings));33            return ensureMockIsAssignableToMockedType(settings, mockInstance);34        } catch (ClassCastException cce) {35            throw new MockitoException(join(36                    "ClassCastException occurred while creating the mockito mock :",37                    "  class to mock : " + describeClass(mockedProxyType),38                    "  created class : " + describeClass(settings.getTypeToMock()),39                    "  proxy instance class : " + describeClass(mockInstance),40                    "  instance creation by : " + instantiator.getClass().getSimpleName(),41                    "",42                    "You might experience classloading issues, please ask the mockito mailing-list.",43                    ""44            ),cce);45        } catch (org.mockito.internal.creation.instance.InstantiationException e) {46            throw new MockitoException("Unable to create mock instance of type '" + mockedProxyType.getSuperclass().getSimpleName() + "'", e);47        }48    }49    private <T> T ensureMockIsAssignableToMockedType(MockCreationSettings<T> settings, T mock) {50        // Force explicit cast to mocked type here, instead of51        // relying on the JVM to implicitly cast on the client call site.52        // This allows us to catch the ClassCastException earlier53        Class<T> typeToMock = settings.getTypeToMock();54        return typeToMock.cast(mock);55    }56    private static String describeClass(Class type) {57        return type == null ? "null" : "'" + type.getCanonicalName() + "', loaded by classloader : '" + type.getClassLoader() + "'";58    }59    private static String describeClass(Object instance) {60        return instance == null ? "null" : describeClass(instance.getClass());61    }62    public MockHandler getHandler(Object mock) {63        if (!(mock instanceof MockMethodInterceptor.MockAccess)) {64            return null;65        }66        return ((MockMethodInterceptor.MockAccess) mock).getMockitoInterceptor().getMockHandler();67    }68    public void resetMock(Object mock, MockHandler newHandler, MockCreationSettings settings) {69        ((MockMethodInterceptor.MockAccess) mock).setMockitoInterceptor(70                new MockMethodInterceptor(asInternalMockHandler(newHandler), settings)71        );72    }73    private static ClassInstantiator initializeClassInstantiator() {74        try {75            Class<?> objenesisClassLoader = Class.forName("org.mockito.internal.creation.bytebuddy.ClassInstantiator$UsingObjenesis");76            Constructor<?> usingClassCacheConstructor = objenesisClassLoader.getDeclaredConstructor(boolean.class);77            return ClassInstantiator.class.cast(usingClassCacheConstructor.newInstance(new GlobalConfiguration().enableClassCache()));78        } catch (Throwable throwable) {79            // MockitoException cannot be used at this point as we are early in the classloading chain and necessary dependencies may not yet be loadable by the classloader80            throw new IllegalStateException(join(81                    "Mockito could not create mock: Objenesis is missing on the classpath.",82                    "Please add Objenesis on the classpath.",83                    ""84            ), throwable);...

Source:src_org_mockito_internal_creation_bytebuddy_ByteBuddyMockMaker.java

...27        Instantiator instantiator = new InstantiatorProvider().getInstantiator(settings);28        T mockInstance = null;29        try {30            mockInstance = instantiator.newInstance(mockedProxyType);31            MockMethodInterceptor.MockAccess mockAccess = (MockMethodInterceptor.MockAccess) mockInstance;32            mockAccess.setMockitoInterceptor(new MockMethodInterceptor(asInternalMockHandler(handler), settings));33            return ensureMockIsAssignableToMockedType(settings, mockInstance);34        } catch (ClassCastException cce) {35            throw new MockitoException(join(36                    "ClassCastException occurred while creating the mockito mock :",37                    "  class to mock : " + describeClass(mockedProxyType),38                    "  created class : " + describeClass(settings.getTypeToMock()),39                    "  proxy instance class : " + describeClass(mockInstance),40                    "  instance creation by : " + instantiator.getClass().getSimpleName(),41                    "",42                    "You might experience classloading issues, please ask the mockito mailing-list.",43                    ""44            ),cce);45        } catch (org.mockito.internal.creation.instance.InstantiationException e) {46            throw new MockitoException("Unable to create mock instance of type '" + mockedProxyType.getSuperclass().getSimpleName() + "'", e);47        }48    }49    private <T> T ensureMockIsAssignableToMockedType(MockCreationSettings<T> settings, T mock) {50        // Force explicit cast to mocked type here, instead of51        // relying on the JVM to implicitly cast on the client call site.52        // This allows us to catch the ClassCastException earlier53        Class<T> typeToMock = settings.getTypeToMock();54        return typeToMock.cast(mock);55    }56    private static String describeClass(Class type) {57        return type == null ? "null" : "'" + type.getCanonicalName() + "', loaded by classloader : '" + type.getClassLoader() + "'";58    }59    private static String describeClass(Object instance) {60        return instance == null ? "null" : describeClass(instance.getClass());61    }62    public MockHandler getHandler(Object mock) {63        if (!(mock instanceof MockMethodInterceptor.MockAccess)) {64            return null;65        }66        return ((MockMethodInterceptor.MockAccess) mock).getMockitoInterceptor().getMockHandler();67    }68    public void resetMock(Object mock, MockHandler newHandler, MockCreationSettings settings) {69        ((MockMethodInterceptor.MockAccess) mock).setMockitoInterceptor(70                new MockMethodInterceptor(asInternalMockHandler(newHandler), settings)71        );72    }73    private static ClassInstantiator initializeClassInstantiator() {74        try {75            Class<?> objenesisClassLoader = Class.forName("org.mockito.internal.creation.bytebuddy.ClassInstantiator$UsingObjenesis");76            Constructor<?> usingClassCacheConstructor = objenesisClassLoader.getDeclaredConstructor(boolean.class);77            return ClassInstantiator.class.cast(usingClassCacheConstructor.newInstance(new GlobalConfiguration().enableClassCache()));78        } catch (Throwable throwable) {79            // MockitoException cannot be used at this point as we are early in the classloading chain and necessary dependencies may not yet be loadable by the classloader80            throw new IllegalStateException(join(81                    "Mockito could not create mock: Objenesis is missing on the classpath.",82                    "Please add Objenesis on the classpath.",83                    ""84            ), throwable);...

Source:MockBytecodeGenerator.java

...11import net.bytebuddy.implementation.MethodDelegation;12import net.bytebuddy.implementation.attribute.MethodAttributeAppender;13import net.bytebuddy.matcher.ElementMatcher;14import org.mockito.internal.creation.bytebuddy.ByteBuddyCrossClassLoaderSerializationSupport.CrossClassLoaderSerializableMock;15import org.mockito.internal.creation.bytebuddy.MockMethodInterceptor.DispatcherDefaultingToRealMethod;16import org.mockito.internal.creation.bytebuddy.MockMethodInterceptor.MockAccess;17import java.lang.reflect.Type;18import java.util.ArrayList;19import java.util.Random;20import static net.bytebuddy.description.modifier.Visibility.PRIVATE;21import static net.bytebuddy.implementation.MethodDelegation.to;22import static net.bytebuddy.matcher.ElementMatchers.*;23class MockBytecodeGenerator {24    private final ByteBuddy byteBuddy;25    private final Random random;26    public MockBytecodeGenerator() {27        byteBuddy = new ByteBuddy().with(TypeValidation.DISABLED);28        random = new Random();29    }30    public <T> Class<? extends T> generateMockClass(MockFeatures<T> features) {31        DynamicType.Builder<T> builder =32                byteBuddy.subclass(features.mockedType)33                         .name(nameFor(features.mockedType))34                         .ignoreAlso(isGroovyMethod())35                         .annotateType(features.mockedType.getAnnotations())36                         .implement(new ArrayList<Type>(features.interfaces))37                         .method(any())38                           .intercept(MethodDelegation.to(DispatcherDefaultingToRealMethod.class))39                           .transform(Transformer.ForMethod.withModifiers(SynchronizationState.PLAIN))40                           .attribute(MethodAttributeAppender.ForInstrumentedMethod.INCLUDING_RECEIVER)41                         .serialVersionUid(42L)42                         .defineField("mockitoInterceptor", MockMethodInterceptor.class, PRIVATE)43                         .implement(MockAccess.class)44                           .intercept(FieldAccessor.ofBeanProperty())45                         .method(isHashCode())46                           .intercept(to(MockMethodInterceptor.ForHashCode.class))47                         .method(isEquals())48                           .intercept(to(MockMethodInterceptor.ForEquals.class));49        if (features.crossClassLoaderSerializable) {50            builder = builder.implement(CrossClassLoaderSerializableMock.class)51                             .intercept(to(MockMethodInterceptor.ForWriteReplace.class));52        }53        return builder.make()54                      .load(new MultipleParentClassLoader.Builder()55                              .append(features.mockedType)56                              .append(features.interfaces)57                              .append(Thread.currentThread().getContextClassLoader())58                              .append(MockAccess.class, DispatcherDefaultingToRealMethod.class)59                              .append(MockMethodInterceptor.class,60                                      MockMethodInterceptor.ForHashCode.class,61                                      MockMethodInterceptor.ForEquals.class).build(),62                              ClassLoadingStrategy.Default.INJECTION.with(features.mockedType.getProtectionDomain()))63                      .getLoaded();64    }65    private static ElementMatcher<MethodDescription> isGroovyMethod() {66        return isDeclaredBy(named("groovy.lang.GroovyObjectSupport"));67    }68    // TODO inspect naming strategy (for OSGI, signed package, java.* (and bootstrap classes), etc...)69    private String nameFor(Class<?> type) {70        String typeName = type.getName();71        if (isComingFromJDK(type)72                || isComingFromSignedJar(type)73                || isComingFromSealedPackage(type)) {74            typeName = "codegen." + typeName;75        }...

Source:MockMethodInterceptor.java

...17import net.bytebuddy.implementation.bind.annotation.Origin;18import net.bytebuddy.implementation.bind.annotation.RuntimeType;19import net.bytebuddy.implementation.bind.annotation.SuperCall;20import net.bytebuddy.implementation.bind.annotation.This;21public class MockMethodInterceptor implements Serializable {22    private static final long serialVersionUID = 7152947254057253027L;23    private final InternalMockHandler handler;24    private final MockCreationSettings mockCreationSettings;25    private final ByteBuddyCrossClassLoaderSerializationSupport serializationSupport;26    public MockMethodInterceptor(InternalMockHandler handler, MockCreationSettings mockCreationSettings) {27        this.handler = handler;28        this.mockCreationSettings = mockCreationSettings;29        serializationSupport = new ByteBuddyCrossClassLoaderSerializationSupport();30    }31    @RuntimeType32    @BindingPriority(BindingPriority.DEFAULT * 3)33    public Object interceptSuperCallable(@This Object mock,34                                         @Origin(cache = true) Method invokedMethod,35                                         @AllArguments Object[] arguments,36                                         @SuperCall(serializableProxy = true) Callable<?> superCall) throws Throwable {37        return doIntercept(38                mock,39                invokedMethod,40                arguments,41                new InterceptedInvocation.SuperMethod.FromCallable(superCall)42        );43    }44    @RuntimeType45    @BindingPriority(BindingPriority.DEFAULT * 2)46    public Object interceptDefaultCallable(@This Object mock,47                                           @Origin(cache = true) Method invokedMethod,48                                           @AllArguments Object[] arguments,49                                           @DefaultCall(serializableProxy = true) Callable<?> superCall) throws Throwable {50        return doIntercept(51                mock,52                invokedMethod,53                arguments,54                new InterceptedInvocation.SuperMethod.FromCallable(superCall)55        );56    }57    @RuntimeType58    public Object interceptAbstract(@This Object mock,59                                    @Origin(cache = true) Method invokedMethod,60                                    @AllArguments Object[] arguments) throws Throwable {61        return doIntercept(62                mock,63                invokedMethod,64                arguments,65                InterceptedInvocation.SuperMethod.IsIllegal.INSTANCE66        );67    }68    private Object doIntercept(Object mock,69                               Method invokedMethod,70                               Object[] arguments,71                               InterceptedInvocation.SuperMethod superMethod) throws Throwable {72        return handler.handle(new InterceptedInvocation(73                mock,74                createMockitoMethod(invokedMethod),75                arguments,76                superMethod,77                SequenceNumber.next()78        ));79    }80    private MockitoMethod createMockitoMethod(Method method) {81        if (mockCreationSettings.isSerializable()) {82            return new SerializableMethod(method);83        } else {84            return new DelegatingMethod(method);85        }86    }87    public MockHandler getMockHandler() {88        return handler;89    }90    public ByteBuddyCrossClassLoaderSerializationSupport getSerializationSupport() {91        return serializationSupport;92    }93    public static class ForHashCode {94        public static int doIdentityHashCode(@This Object thiz) {95            return System.identityHashCode(thiz);96        }97    }98    public static class ForEquals {99        public static boolean doIdentityEquals(@This Object thiz, @Argument(0) Object other) {100            return thiz == other;101        }102    }103    public static class ForWriteReplace {104        public static Object doWriteReplace(@This MockAccess thiz) throws ObjectStreamException {105            return thiz.getMockitoInterceptor().getSerializationSupport().writeReplace(thiz);106        }107    }108    public static interface MockAccess {109        MockMethodInterceptor getMockitoInterceptor();110        void setMockitoInterceptor(MockMethodInterceptor mockMethodInterceptor);111    }112}...

MockMethodInterceptor

Using AI Code Generation

1import org.mockito.internal.creation.bytebuddy.MockMethodInterceptor;2import org.mockito.invocation.InvocationOnMock;3import org.mockito.stubbing.Answer;4import java.lang.reflect.Method;5import java.lang.reflect.Modifier;6import java.util.Arrays;7import java.util.List;8import java.util.stream.Collectors;9import java.util.stream.Stream;10public class Test {11    public static void main(String[] args) {12        List list = MockMethodInterceptor.mock(List.class, new Answer() {13            public Object answer(InvocationOnMock invocation) throws Throwable {14                String methodName = invocation.getMethod().getName();15                Object[] args = invocation.getArguments();16                Class returnType = invocation.getMethod().getReturnType();17                System.out.println("Method name: " + methodName);18                System.out.println("Arguments: " + Arrays.toString(args));19                System.out.println("Return type: " + returnType);20                System.out.println("Return type: " + returnType);21                if (returnType == void.class) {22                    return null;23                }24                return getDefaultReturnValue(returnType);25            }26        });27        list.add("one");28        list.add("two");29        list.get(0);30        list.clear();31        list.size();32        list.add("three");33        list.add("four");34        list.get(1);35        list.clear();36        list.size();37    }38    private static Object getDefaultReturnValue(Class returnType) {39        if (returnType == boolean.class

MockMethodInterceptor

Using AI Code Generation

1public class MockMethodInterceptorTest {2    public static void main(String[] args) {3        MockMethodInterceptor mockMethodInterceptor = new MockMethodInterceptor();4        mockMethodInterceptor.intercept(null, null, null, null);5    }6}7public class MockMethodInterceptorTest {8    public static void main(String[] args) {9        MockMethodInterceptor mockMethodInterceptor = new MockMethodInterceptor();10        mockMethodInterceptor.intercept(null, null, null, null);11    }12}13	at MockMethodInterceptorTest.main(MockMethodInterceptorTest.java:6)14	at java.base/jdk.internal.loader.BuiltinClassLoader.loadClass(BuiltinClassLoader.java:583)15	at java.base/jdk.internal.loader.ClassLoaders$AppClassLoader.loadClass(ClassLoaders.java:178)16	at java.base/java.lang.ClassLoader.loadClass(ClassLoader.java:521)17	at MockMethodInterceptorTest.main(MockMethodInterceptorTest.java:6)18	at java.base/jdk.internal.loader.BuiltinClassLoader.loadClass(BuiltinClassLoader.java:583)19	at java.base/jdk.internal.loader.ClassLoaders$AppClassLoader.loadClass(ClassLoaders.java:178)20	at java.base/java.lang.ClassLoader.loadClass(ClassLoader.java:521)

MockMethodInterceptor

Using AI Code Generation

1package org.mockito.internal.creation.bytebuddy;2import org.mockito.invocation.InvocationOnMock;3import org.mockito.stubbing.Answer;4import java.io.Serializable;5import java.lang.reflect.Method;6import java.util.Arrays;7import java.util.List;8public class MockMethodInterceptor implements Answer<Object>, Serializable {9    private static final long serialVersionUID = 1L;10    private final MockMethodInterceptorFilter filter;11    private final List<Answer<Object>> delegates;12    public MockMethodInterceptor(MockMethodInterceptorFilter filter, List<Answer<Object>> delegates) {13        this.filter = filter;14        this.delegates = delegates;15    }16    public Object answer(InvocationOnMock invocation) throws Throwable {17        Method method = invocation.getMethod();18        if (filter.isMocked(method)) {19            for (Answer<Object> delegate : delegates) {20                Object result = delegate.answer(invocation);21                if (result != null) {22                    return result;23                }24            }25            return null;26        } else {27            return invocation.callRealMethod();28        }29    }30    public String toString() {31        return "MockMethodInterceptor{" +32                '}';33    }34}35package org.mockito.internal.creation.bytebuddy;36import org.mockito.invocation.InvocationOnMock;37import org.mockito.stubbing.Answer;38import java.io.Serializable;39import java.lang.reflect.Method;40import java.util.Arrays;41import java.util.List;42public class MockMethodInterceptor implements Answer<Object>, Serializable {43    private static final long serialVersionUID = 1L;44    private final MockMethodInterceptorFilter filter;45    private final List<Answer<Object>> delegates;46    public MockMethodInterceptor(MockMethodInterceptorFilter filter, List<Answer<Object>> delegates) {47        this.filter = filter;48        this.delegates = delegates;49    }50    public Object answer(InvocationOnMock invocation) throws Throwable {51        Method method = invocation.getMethod();52        if (filter.isMocked(method)) {53            for (Answer<Object> delegate : delegates) {54                Object result = delegate.answer(invocation);55                if (result != null) {56                    return result;57                }58            }59            return null;60        } else {61            return invocation.callRealMethod();62        }63    }64    public String toString() {

MockMethodInterceptor

Using AI Code Generation

1package com.example;2import org.mockito.internal.creation.bytebuddy.MockMethodInterceptor;3public class 1 {4    public static void main(String[] args) {5        MockMethodInterceptor mockMethodInterceptor = new MockMethodInterceptor();6    }7}8package com.example;9import org.mockito.internal.creation.bytebuddy.MockMethodInterceptor;10public class 2 {11    public static void main(String[] args) {12        MockMethodInterceptor mockMethodInterceptor = new MockMethodInterceptor();13    }14}15jar {16    manifest {17    }18    from {19        configurations.compile.collect { it.isDirectory() ? it : zipTree(it) }20    }21}22package com.example;23import org.mockito.internal.creation.bytebuddy.MockMethodInterceptor;24public class 3 {25    public static void main(String[] args) {26        MockMethodInterceptor mockMethodInterceptor = new MockMethodInterceptor();27    }28}

MockMethodInterceptor

Using AI Code Generation

1import org.mockito.internal.creation.bytebuddy.MockMethodInterceptor;2import org.mockito.invocation.InvocationOnMock;3import org.mockito.stubbing.Answer;4public class 1 {5    public static void main(String[] args) {6        MockMethodInterceptor mockMethodInterceptor = new MockMethodInterceptor();7        mockMethodInterceptor.setHandler(new Answer() {8            public Object answer(InvocationOnMock invocation) throws Throwable {9                return invocation.getArguments()[0];10            }11        });12        mockMethodInterceptor.intercept(null, null, null, null);13    }14}15How to mock a method call in Mockito with different arguments and return types using Answer and doAnswer() method?

MockMethodInterceptor

Using AI Code Generation

1public class 1 {2    public static void main(String[] args) {3        List<String> list = Mockito.mock(List.class);4        Mockito.when(list.get(0)).thenReturn("Mockito");5        Mockito.when(list.get(1)).thenReturn("PowerMock");6        Mockito.when(list.get(2)).thenReturn("Junit");7        Mockito.when(list.get(3)).thenReturn("EasyMock");8        Mockito.when(list.get(4)).thenReturn("TestNG");9        Mockito.when(list.get(5)).thenReturn("Selenium");10        Mockito.when(list.get(6)).thenReturn("Cucumber");

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.