làm thế nào để thay đổi một đối tượng mà được thông qua tham chiếu đến một mô hình trong Mockito

Question

Với đoạn mã sau

@Mock 
Client client; 

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

làm thế nào để hướng dẫn Mockito để điền vào các đối tượng baos?

Answer 1

Bạn có thể sử dụng 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); 

Tuy nhiên, nếu bạn có khả năng tái cấu trúc mã thử nghiệm, tốt hơn là để khách hàng trả về OutputStream hoặc một số dữ liệu có thể được đưa vào luồng đầu ra này. Đây sẽ là thiết kế tốt hơn nhiều.

Answer 2

thử này

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