WebActionBase64WriteInterceptorTest.java 文件源码

public void testWrapsInputStream(String contentType) throws WebApplicationException, IOException {
    WriterInterceptorContext context = mockContext(contentType);
    OutputStream os = mock(OutputStream.class);
    when(context.getOutputStream()).thenReturn(os);

    writeInterceptor.aroundWriteTo(context);

    verifyZeroInteractions(os);

    ArgumentCaptor<OutputStream> updatedOsCapture = ArgumentCaptor.forClass(OutputStream.class);
    verify(context).setOutputStream(updatedOsCapture.capture());
    verify(context).getMediaType();
    verify(context).getOutputStream();
    verify(context).proceed();
    verifyNoMoreInteractions(context);

    OutputStream updatedOs = updatedOsCapture.getValue();

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