java - Mockito - 返回与传递给方法相同的对象

Question

假设我在某个服务类中有以下方法:

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

我想测试此方法的行为，因此想以以下方式模拟 repository.merge:

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

然后模拟存储库返回 makesSthWithEntity 传递给它的内容，我可以轻松地对其进行测试。

任何想法如何强制 mockito 返回 objectPassedAsArgument ？

Answer 1

您可以使用 Mockito 提供的答案:

when(mock.something()).then(AdditionalAnswers.returnsFirstArg())

其中 AdditionalAnswers.returnsFirstArg() 可以静态导入。