如何更改通过引用传递给Mockito中的模拟的对象

Question

给出以下代码

@Mock
Client client;

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

如何指示Mockito填充baos对象？

Answer 1

您可以使用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);

但是，如果您有可能重构经过测试的代码，最好是让客户端返回OutputStream或可以放入此Output流的一些数据。这将是更好的设计。

Answer 2

试试这个

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