mockit.internal.faking.FakeInvocation Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2006 JMockit developers
 * This file is subject to the terms of the MIT license (see LICENSE.txt).
 */
package mockit.internal.faking;

import static mockit.internal.util.Utilities.NO_ARGS;

import edu.umd.cs.findbugs.annotations.NonNull;
import edu.umd.cs.findbugs.annotations.Nullable;

import java.lang.reflect.Constructor;
import java.lang.reflect.Member;

import mockit.internal.BaseInvocation;
import mockit.internal.state.TestRun;

import org.checkerframework.checker.index.qual.NonNegative;

/**
 * An invocation to a {@linkplain mockit.Mock fake} method.
 */
public final class FakeInvocation extends BaseInvocation {
    @NonNull
    private final FakeState fakeState;
    @NonNull
    private final String fakedClassDesc;
    @NonNull
    private final String fakedMethodName;
    @NonNull
    private final String fakedMethodDesc;
    boolean proceeding;

    @NonNull // called by generated bytecode
    public static FakeInvocation create(@Nullable Object invokedInstance, @Nullable Object[] invokedArguments,
            @NonNull String fakeClassDesc, @NonNegative int fakeStateIndex, @NonNull String fakedClassDesc,
            @NonNull String fakedMethodName, @NonNull String fakedMethodDesc) {
        Object fake = TestRun.getFake(fakeClassDesc, invokedInstance);
        assert fake != null;
        FakeState fakeState = TestRun.getFakeStates().getFakeState(fake, fakeStateIndex);
        Object[] args = invokedArguments == null ? NO_ARGS : invokedArguments;
        return new FakeInvocation(invokedInstance, args, fakeState, fakedClassDesc, fakedMethodName, fakedMethodDesc);
    }

    FakeInvocation(@Nullable Object invokedInstance, @NonNull Object[] invokedArguments, @NonNull FakeState fakeState,
            @NonNull String fakedClassDesc, @NonNull String fakedMethodName, @NonNull String fakedMethodDesc) {
        super(invokedInstance, invokedArguments, fakeState.getTimesInvoked());
        this.fakeState = fakeState;
        this.fakedClassDesc = fakedClassDesc;
        this.fakedMethodName = fakedMethodName;
        this.fakedMethodDesc = fakedMethodDesc;
    }

    @NonNull
    @Override
    protected Member findRealMember() {
        Object invokedInstance = getInvokedInstance();

        if (invokedInstance != null) {
            Class mockedClass = invokedInstance.getClass();
            return fakeState.getRealMethodOrConstructor(mockedClass, fakedMethodName, fakedMethodDesc);
        }

        return fakeState.getRealMethodOrConstructor(fakedClassDesc, fakedMethodName, fakedMethodDesc);
    }

    @SuppressWarnings("WeakerAccess") // called from generated bytecode
    public boolean shouldProceedIntoConstructor() {
        if (proceeding && getInvokedMember() instanceof Constructor) {
            fakeState.clearProceedIndicator();
            return true;
        }

        return false;
    }

    @Override
    public void prepareToProceed() {
        fakeState.prepareToProceed(this);
        proceeding = true;
    }

    public void prepareToProceedFromNonRecursiveMock() {
        fakeState.prepareToProceedFromNonRecursiveFake(this);
        proceeding = true;
    }

    @Override
    public void cleanUpAfterProceed() {
        fakeState.clearProceedIndicator();
    }
}