Join the 1-day Testing & QA Summit featuring 15+ Expert Speakers.Register for FREE! Join TestMu Conference

How to use DelegatingToConstructorsOngoingStubbing method of org.powermock.api.mockito.internal.expectation.DelegatingToConstructorsOngoingStubbing class

Best Powermock code snippet using org.powermock.api.mockito.internal.expectation.DelegatingToConstructorsOngoingStubbing.DelegatingToConstructorsOngoingStubbing

Source:DefaultConstructorExpectationSetup.java

...17 */18package org.powermock.api.mockito.expectation;19import org.mockito.ArgumentMatchers;20import org.mockito.stubbing.OngoingStubbing;21import org.powermock.api.mockito.internal.expectation.DelegatingToConstructorsOngoingStubbing;22import org.powermock.api.mockito.internal.invocation.MockitoNewInvocationControl;23import org.powermock.api.mockito.internal.mockcreation.DefaultMockCreator;24import org.powermock.core.MockRepository;25import org.powermock.core.spi.NewInvocationControl;26import org.powermock.core.spi.support.InvocationSubstitute;27import org.powermock.reflect.internal.WhiteboxImpl;28import org.powermock.tests.utils.ArrayMerger;29import org.powermock.tests.utils.impl.ArrayMergerImpl;30import java.lang.reflect.Constructor;31import java.lang.reflect.Method;32public class DefaultConstructorExpectationSetup<T> implements ConstructorExpectationSetup<T> {33    34    private final Class<T> mockType;35    private final ArrayMerger arrayMerger;36    private final DefaultMockCreator mockCreator;37    private Class<?>[] parameterTypes = null;38    private final InvocationSubstitute mock;39    40    public DefaultConstructorExpectationSetup(Class<T> mockType) {41        this.arrayMerger = new ArrayMergerImpl();42        this.mockType = mockType;43        this.mockCreator = new DefaultMockCreator();44        this.mock = getMockCreator().createMock(InvocationSubstitute.class, false, false, null, null, (Method[]) null);45    }46    47    @Override48    public OngoingStubbing<T> withArguments(Object firstArgument, Object... additionalArguments) throws Exception {49        return createNewSubstituteMock(mockType, parameterTypes, arrayMerger.mergeArrays(Object.class, new Object[]{firstArgument},50                                                                                         additionalArguments));51    }52    53    @SuppressWarnings({"unchecked", "rawtypes"})54    private OngoingStubbing<T> createNewSubstituteMock(Class<T> type, Class<?>[] parameterTypes, Object... arguments) throws Exception {55        if (type == null) {56            throw new IllegalArgumentException("type cannot be null");57        }58        59        final Class<T> unmockedType = (Class<T>) WhiteboxImpl.getOriginalUnmockedType(type);60        if (parameterTypes == null) {61            WhiteboxImpl.findUniqueConstructorOrThrowException(type, arguments);62        } else {63            WhiteboxImpl.getConstructor(unmockedType, parameterTypes);64        }65        66        NewInvocationControl<OngoingStubbing<T>> newInvocationControl = createNewInvocationControl(type, unmockedType);67        68        return newInvocationControl.expectSubstitutionLogic(arguments);69    }70    71    private NewInvocationControl<OngoingStubbing<T>> createNewInvocationControl(final Class<T> type, final Class<T> unmockedType) {72        /*73        * Check if this type has been mocked before74        */75        NewInvocationControl<OngoingStubbing<T>> newInvocationControl =76            (NewInvocationControl<OngoingStubbing<T>>) MockRepository.getNewInstanceControl(unmockedType);77        if (newInvocationControl == null) {78            newInvocationControl = createNewInvocationControl(mock);79            MockRepository.putNewInstanceControl(type, newInvocationControl);80            MockRepository.addObjectsToAutomaticallyReplayAndVerify(WhiteboxImpl.getOriginalUnmockedType(type));81        }82        return newInvocationControl;83    }84    85    @Override86    public OngoingStubbing<T> withAnyArguments() throws Exception {87        if (mockType == null) {88            throw new IllegalArgumentException("Class to expected cannot be null");89        }90        final Class<T> unmockedType = (Class<T>) WhiteboxImpl.getOriginalUnmockedType(mockType);91        final Constructor<?>[] allConstructors = WhiteboxImpl.getAllConstructors(unmockedType);92        final Constructor<?> constructor = allConstructors[0];93        final Class<?>[] parameterTypes = constructor.getParameterTypes();94        Object[] paramArgs = new Object[parameterTypes.length];95        for (int i = 0; i < parameterTypes.length; i++) {96            Class<?> paramType = parameterTypes[i];97            paramArgs[i] = createParamArgMatcher(paramType);98        }99        Constructor<?>[] otherCtors = new Constructor<?>[allConstructors.length - 1];100        System.arraycopy(allConstructors, 1, otherCtors, 0, allConstructors.length - 1);101        102        final OngoingStubbing<T> ongoingStubbing = createNewSubstituteMock(mockType, parameterTypes, paramArgs);103        return new DelegatingToConstructorsOngoingStubbing<T>(otherCtors, ongoingStubbing);104    }105    106    private Object createParamArgMatcher(Class<?> paramType) {107        return ArgumentMatchers.nullable(paramType);108    }109    110    @Override111    public OngoingStubbing<T> withNoArguments() throws Exception {112        return createNewSubstituteMock(mockType, parameterTypes);113    }114    115    @Override116    public WithExpectedArguments<T> withParameterTypes(Class<?> parameterType, Class<?>... additionalParameterTypes) {117        this.parameterTypes = arrayMerger.mergeArrays(Class.class, new Class<?>[]{parameterType}, additionalParameterTypes);...