use of org.alfresco.rest.framework.webscripts.AbstractResourceWebScript in project alfresco-remote-api by Alfresco.
the class WithResponseTest method testSetResponse.
@Test
public void testSetResponse() throws Exception {
AbstractResourceWebScript responseWriter = new ResourceWebScriptDelete();
responseWriter.setAssistant(new ApiAssistant());
WithResponse wr = new WithResponse(Status.STATUS_OK, ResponseWriter.DEFAULT_JSON_CONTENT, ResponseWriter.CACHE_NEVER);
WebScriptResponse response = mock(WebScriptResponse.class);
responseWriter.setResponse(response, wr.getStatus(), wr.getCache(), wr.getContentInfo(), wr.getHeaders());
verify(response, times(1)).setStatus(anyInt());
verify(response, times(1)).setCache((Cache) any());
verify(response, times(1)).setContentType(anyString());
verify(response, times(0)).setHeader(anyString(), anyString());
response = mock(WebScriptResponse.class);
responseWriter.setResponse(response, wr.getStatus(), null, null, null);
verify(response, times(1)).setStatus(anyInt());
verify(response, times(0)).setCache((Cache) any());
verify(response, times(0)).setContentType(anyString());
verify(response, times(0)).setHeader(anyString(), anyString());
response = mock(WebScriptResponse.class);
wr.addHeader("king", "can");
wr.addHeader("king", "kong");
responseWriter.setResponse(response, wr.getStatus(), null, null, wr.getHeaders());
verify(response, times(1)).setStatus(anyInt());
verify(response, times(0)).setCache((Cache) any());
verify(response, times(0)).setContentType(anyString());
verify(response, times(1)).setHeader(eq("king"), anyString());
verify(response, times(1)).addHeader(eq("king"), anyString());
response = mock(WebScriptResponse.class);
wr.addHeader("king", "kin");
wr.setHeader("ping", "ping");
responseWriter.setResponse(response, wr.getStatus(), null, null, wr.getHeaders());
verify(response, times(1)).setStatus(anyInt());
verify(response, times(0)).setCache((Cache) any());
verify(response, times(0)).setContentType(anyString());
verify(response, times(1)).setHeader(eq("king"), anyString());
verify(response, times(1)).setHeader(eq("ping"), anyString());
verify(response, times(2)).addHeader(eq("king"), anyString());
}
Aggregations