All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.mockito.internal.stubbing.defaultanswers.TriesToReturnSelf Maven / Gradle / Ivy

Go to download

Mockito mock objects library core API and implementation

There is a newer version: 5.12.0
Show newest version
/*
 * Copyright (c) 2016 Mockito contributors
 * This program is made available under the terms of the MIT License.
 */
package org.mockito.internal.stubbing.defaultanswers;

import java.io.Serializable;

import org.mockito.internal.util.MockUtil;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;

public class TriesToReturnSelf implements Answer, Serializable {

    private final ReturnsEmptyValues defaultReturn = new ReturnsEmptyValues();

    @Override
    public Object answer(InvocationOnMock invocation) throws Throwable {
        Class methodReturnType = invocation.getMethod().getReturnType();
        Object mock = invocation.getMock();
        Class mockType = MockUtil.getMockHandler(mock).getMockSettings().getTypeToMock();

        if (methodReturnType.isAssignableFrom(mockType) && methodReturnType != Object.class) {
            return invocation.getMock();
        }

        return defaultReturn.returnValueFor(methodReturnType);
    }
}
 















© 2015 - 2024 Weber Informatics LLC | Privacy Policy