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

Вы можете использовать Answer, а затем использовать thenAnswer().

Нечто похожее на:

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

Конечно, как только вы это вы можете реорганизовать ответ на многоразовой ответ называется ReturnFirstArgument или аналогичный.

Answer 2

Или лучше с помощью Mockito погружено ответы

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

Где AdditionalAnswers.returnsFirstArg() может быть статически импортирован.