come cambiare un oggetto che viene passato in riferimento ad un finto in Mockito

Question

Dato il seguente codice

@Mock 
Client client; 

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

come faccio istruire Mockito per riempire l'oggetto baos?

Answer 1

È possibile utilizzare 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); 

Tuttavia, se avete la possibilità di refactoring del codice testato, meglio è quello di rendere cliente restituire l'OutputStream o alcuni dati che possono essere messi a questo flusso di uscita. Questo sarebbe un design molto migliore.

Answer 2

provare questo

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