Mockito - restituendo lo stesso oggetto passato nel metodo

Question

Immaginiamo Ho un seguente metodo in qualche classe di servizio:

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

mi piacerebbe testare il comportamento di questo metodo e, quindi, vogliono prendere in giro il repository.merge in modo seguente:

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

rendimenti repository Poi deriso che ciò che makesSthWithEntity passato ad esso e posso facilmente testarlo.

Qualche idea, come posso forzare il mockito a restituire objectPassedAsArgument?

Answer 1

È possibile implementare uno Answer e quindi utilizzare thenAnswer().

Qualcosa di simile a:

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

Naturalmente, una volta che hai questo si può refactoring la risposta in una risposta riutilizzabile chiamato ReturnFirstArgument o simili.

Answer 2

o meglio utilizzando Mockito spediti risposte

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

Dove AdditionalAnswers.returnsFirstArg() potrebbero essere importati in modo statico.