使用Mockito时是否可以为所有答案添加拦截器？

Question

假设我的Interface方法有一个验证注释来验证输入参数和返回值.目前是否有可能(V 1.9.5)告诉Mockito在调用过程中调用此验证器？

背景是通过以违反指定验证器的方式模拟给定接口来防止开发人员编写不切实际的测试.

所以我想要的是注册类似的东西

class MyAnswerInterceptor<T> implements AnswerInterceptor<T> {
    @Override
    public Answer<T> intercept(final Answer<T> answer) {
        return new Answer<T>() {
            @Override
            public T answer(InvocationOnMock invocation) throws Throwable {
                validateArguments(invocation);
                T result = answer.answer(invocation);
                validateReturnValue(result);
                return result;
            }
        }
    }
}

在给定模拟的每个答案上被调用.这有可能吗？我已经查看了代码,也检查了我是否可以在某些时候进行攻击(甚至使用反射等),但似乎是由于实例创建和逻辑的纠缠,它几乎不可能实现我想要的(即类似的东西MockHandler mockHandler = new MockHandlerFactory().create(settings);使得无法挂钩并将自定义的东西放在顶部而无需修补和部署整个东西......)

任何见解都将受到高度赞赏:-)

Answer 1

您可以通过创建自定义MockMaker.

MockMaker 是一个扩展点，可以使用自定义动态代理并避免使用默认的 cglib/asm/objenesis 实现

我们的自定义实现将所有复杂的事情委托给默认的MockMaker: CglibMockMaker。它createMock通过注册settings参数 an来仅“装饰”方法InvocationListener。invocation当允许使用调用 和validateArguments时，该侦听器将收到通知validateReturnValue。

import org.mockito.internal.creation.CglibMockMaker;
import org.mockito.invocation.Invocation;
import org.mockito.invocation.MockHandler;
import org.mockito.listeners.InvocationListener;
import org.mockito.listeners.MethodInvocationReport;
import org.mockito.mock.MockCreationSettings;
import org.mockito.plugins.MockMaker;

public class ValidationMockMaker implements MockMaker {
    private final MockMaker delegate = new CglibMockMaker();

    public ValidationMockMaker() {
    }
    @Override
    public <T> T createMock(MockCreationSettings<T> settings, MockHandler handler) {
        settings.getInvocationListeners().add(new InvocationListener() {

            @Override
            public void reportInvocation(MethodInvocationReport methodInvocationReport) {
                Invocation invocation = (Invocation) methodInvocationReport.getInvocation();
                validateArguments(invocation.getArguments());
                validateReturnValue(methodInvocationReport.getReturnedValue());
            }
        });
        return delegate.createMock(settings, handler);
    }

    @Override
    public MockHandler getHandler(Object mock) {
        return delegate.getHandler(mock);
    }

    @Override
    public void resetMock(Object mock, MockHandler newHandler, MockCreationSettings settings) {
        delegate.resetMock(mock, newHandler, settings);
    }

    protected void validateArguments(Object... arguments) {
//        Arrays.stream(arguments).forEach(Objects::requireNonNull);
    }

    private void validateReturnValue(Object result) {
//        Objects.requireNonNull(result);
    }
}

最后但并非最不重要的一点是，我们需要告诉 Mockito 使用我们的实现。这可以通过添加文件来实现

mockito-extensions/org.mockito.plugins.MockMaker 

包含我们的 MockMaker 类名：

ValidationMockMaker

请参阅Using the extension pointjavadoc 中的部分。