ConditionalBase64WriteInterceptorTest.java 文件源码

@Test
public void testWrapsOutputStreamAlways() throws IOException {

    WriterInterceptorContext context = mock(WriterInterceptorContext.class);
    OutputStream os = mock(OutputStream.class);
    when(context.getOutputStream()).thenReturn(os);

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

    alwaysBase64WriteInterceptor.aroundWriteTo(context);

    verify(alwaysBase64WriteInterceptor).isBase64(context);

    verifyZeroInteractions(os);

    verify(context).setOutputStream(updatedOsCapture.capture());
    verify(context).proceed();
    verify(context).getOutputStream();
    verifyNoMoreInteractions(context);
    OutputStream updatedOs = updatedOsCapture.getValue();

    // just make sure we have some wrapper
    assertNotSame(os, updatedOs);
    updatedOs.close();
    verify(os).close();
}