jak zmienić obiekt, który przepuszcza się przez odniesienie do makiety w Mockito

Question

Biorąc pod uwagę następujący kod

@Mock 
Client client; 

ByteArrayOutputStream baos = new ByteArrayOutputStream(); 
client.retrieveFile(baos); // client is supposed to fill boas with data 

jak mogę polecić Mockito wypełnić obiekt baos?

Answer 1

Możesz użyć Mockitos Answer.

doAnswer(new Answer() { 
    @Override 
    public Object answer(InvocationOnMock invocation) { 
     Object[] args = invocation.getArguments(); 
     ByteArrayOutputStream baos = (ByteArrayOutputStream)args[0]; 
     //fill baos with data 
     return null; 
    } 
}).when(client).retrieveFile(baos); 

Jednakże, jeśli masz możliwość byłaby testowanego kodu, lepiej jest, aby klient zwróci OutputStream lub niektóre dane, które mogą zostać wprowadzone do tego strumienia wyjściowego. To byłby znacznie lepszy projekt.

Answer 2

spróbować

 doAnswer(new Answer() { 
      @Override 
      public Object answer(InvocationOnMock invocation) { 
       ByteArrayOutputStream baos = (ByteArrayOutputStream) invocation.getArguments()[0]; 
       // fill it here 
       return null; 
      }}).when(client).retrieveFile(baos);