如何在 zipoutputstream 中断言响应

Question

我正在尝试使用 MockitoJUnitRunner 编写 JUnit。 我正在将文件 ID 传递给我的函数，该函数正在从云下载文件并将 zip 文件作为下载返回。 这是我的代码

public void getLogFile(HttpServletResponse response, String id) throws IOException {

    response.setContentType("Content-type: application/zip");
    response.setHeader("Content-Disposition", "attachment; filename=LogFiles.zip");

    ServletOutputStream out = response.getOutputStream();
    ZipOutputStream zos = new ZipOutputStream(new BufferedOutputStream(out));

    zos.putNextEntry(new ZipEntry(id));

    InputStream inputStream = someDao.getFile(id);

    BufferedInputStream fif = new BufferedInputStream(inputStream);

    int data = 0;

    while ((data = fif.read()) != -1) {
        zos.write(data);
    }

    fif.close();

    zos.closeEntry();
    zos.close();
}

而我的 JUnit 函数是

@Mock
private MockHttpServletResponse mockHttpServletResponse;

anyInputStream = new ByteArrayInputStream("test data".getBytes());

    @Test
    public void shouldDownloadFile() throws IOException {

    ServletOutputStream outputStream = mock(ServletOutputStream.class);

        when(mockHttpServletResponse.getOutputStream()).thenReturn(outputStream);


 =>          when(someDao.download(anyString())).thenReturn(anyInputStream);


        controller.getLogFile(mockHttpServletResponse, id);

        verify(mockHttpServletResponse).setContentType("Content-type: application/zip");

        verify(mockHttpServletResponse).setHeader("Content-Disposition","attachment; filename=LogFiles.zip");

        verify(atmosdao).download(atmosFilePath);
    }

这个单元测试通过了，但是我想验证 outputStream 上写了什么，我该怎么做？当我将“测试数据”写入模拟的 outputStream 时，就像

anyInputStream = new ByteArrayInputStream("test data".getBytes());

when(someDao.download(anyString())).thenReturn(anyInputStream);

mockHttpServletResponse.getContentAsString() 给我 null ！

是否可以断言使用 zipoutputStream 编写的MockHttpServletResponse？如果是，那我该怎么做？

谢谢。

Answer 1

我得到了我想要的东西... 这就是我使用 powerMockito 断言写入 zipoutputStream 的数据所做的。

@Test
public void ShouldAttemptToWriteDownloadedFileToZipOutputStream() throws Exception {

    InputStream anyInputStream = new ByteArrayInputStream("test data".getBytes());

    ServletOutputStream outputStream = mock(ServletOutputStream.class);

    BufferedOutputStream bufferedOutputStream = Mockito.mock(BufferedOutputStream.class);
    PowerMockito.whenNew(BufferedOutputStream.class).withArguments(outputStream).thenReturn(bufferedOutputStream);

    ZipOutputStream zipOutputStream = Mockito.mock(ZipOutputStream.class);
    PowerMockito.whenNew(ZipOutputStream.class).withArguments(bufferedOutputStream).thenReturn(zipOutputStream);

    BufferedInputStream bufferedInputStream = new BufferedInputStream(anyInputStream);
    PowerMockito.whenNew(BufferedInputStream.class).withArguments(anyInputStream).thenReturn(bufferedInputStream);

    subjectUnderTest.getLogFile(mockHttpServletResponse, "12345");

    int data = 0;

    while ((data = bufferedInputStream.read()) != -1) {

        verify(zipOutputStream).write(data);
    }
}

感谢雨果的帮助！

Answer 2

您可以创建一个自定义的，而不是模拟您的 OutputStream：

public class CustomOutputStream extends ServletOutputStream {
    private ByteArrayOutputStream out = new ByteArrayOutputStream();
    private String content;

    @Override
    public void write(int b) throws IOException {
        out.write(b);
    }

    @Override
    public void close() throws IOException {
        content = new String(out.toByteArray());
        out.close();
        super.close();
    }

    public String getContentAsString() {
        return this.content; 
    }
}

这个类将存储所有写入它的字节并将它们保存在content字段中。

然后你替换这个：

ServletOutputStream outputStream = mock(ServletOutputStream.class);

通过这个：

CustomOutputStream outputStream = new CustomOutputStream();

当您的 servlet 调用 getOutputStream() 时，它将使用自定义的，最后 getContentAsString() 将返回您写入 servlet 的输出。

注意：输出是压缩的，所以字符串会包含奇怪的字符。如果您想要原始字符串，则必须将其解压缩（在这种情况下，我将使用 out.toByteArray() 返回的字节数组而不是字符串，因为当您以这种方式创建字符串时，您可能会遇到编码问题打电话给string.getBytes())