如何使用参数captor来验证写入httpservletresponse的字节

Question

i有一个控制器，该控制器提供了下载文件的功能。

@ResponseBody
public void downloadRecycleResults(String batchName, HttpServletResponse response) throws Exception {
    File finalResultFile = null;
    // code here generates and initializes finalResultFile for batchName
    response.setContentType("application/octet-stream");
    response.setHeader("Content-Disposition", "attachment;filename=" + finalResultFile.getName());
    IOUtils.copy(new FileReader(finalResultFile), response.getOutputStream());
}

我无法弄清楚如何编写测试，可以验证写入response的内容。我已经使用了 ArgumentCaptor批次，但某种程度上似乎不适合这里。

controller.downloadRecycleResults("batchName", mock(HttpServletResponse.class));
verify(response).getOutputStream(); // but how to capture content?

Answer 1

，大卫建议，我能够做到。

ServletOutputStream opStreamMock = mock(ServletOutputStream.class);
when(response.getOutputStream()).thenReturn(opStreamMock);
ArgumentCaptor<byte[]> captor = ArgumentCaptor.forClass(byte[].class);
verify(opStreamMock).write(captor.capture(), Mockito.anyInt(), Mockito.anyInt());
//can create reader now.
BufferedReader reader = new BufferedReader(new InputStreamReader(new ByteArrayInputStream(captor.getValue())));

Answer 2

做到这一点的一种方法是模拟response及其输出流，然后验证呼叫对您模拟的输出流的write方法。