mockito-返回与传入方法相同的对象

Question

public SomeEntity makeSthWithEntity(someArgs){
    SomeEntity entity = new SomeEntity();
    /**
     * here goes some logic concerning the entity
     */
    return repository.merge(entity);
}

when(repository.merge(any(SomeEntity.class))).thenReturn(objectPassedAsArgument);

然后mocked repository返回makessthwithEntity传递给它的内容，我可以轻松地测试它。

有什么想法吗？如何强制mockito返回objectpassedasargument？

Answer 1

您可以实现答案然后改用ThenAnswer()。

类似于：

when(mock.someMethod(anyString())).thenAnswer(new Answer() {
    public Object answer(InvocationOnMock invocation) {
        return invocation.getArguments()[0];
    }
});

当然，一旦有了这些，您就可以将答案重构为一个可重用的答案，称为ReturnFirstArgument或类似的答案。