ConditionalBase64WriteInterceptorTest.java 文件源码

private void testBase64Encoding(byte[] bytes, String expectedBase64) throws IOException {

        WriterInterceptorContext context = mock(WriterInterceptorContext.class);
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        when(context.getOutputStream()).thenReturn(baos);

        ArgumentCaptor<OutputStream> updatesOsCapture = ArgumentCaptor.forClass(OutputStream.class);

        alwaysBase64WriteInterceptor.aroundWriteTo(context);

        verify(context).setOutputStream(updatesOsCapture.capture());
        OutputStream updatedOs = updatesOsCapture.getValue();

        updatedOs.write(bytes);
        updatedOs.close();
        assertEquals(expectedBase64, baos.toString());
    }