Search in sources :

Example 11 with AbstractResourceWebScript

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());
}
Also used : ApiAssistant(org.alfresco.rest.framework.tools.ApiAssistant) WithResponse(org.alfresco.rest.framework.webscripts.WithResponse) AbstractResourceWebScript(org.alfresco.rest.framework.webscripts.AbstractResourceWebScript) WebScriptResponse(org.springframework.extensions.webscripts.WebScriptResponse) ResourceWebScriptDelete(org.alfresco.rest.framework.webscripts.ResourceWebScriptDelete) Test(org.junit.Test)

Aggregations

AbstractResourceWebScript (org.alfresco.rest.framework.webscripts.AbstractResourceWebScript)11 Test (org.junit.Test)9 WebScriptResponse (org.springframework.extensions.webscripts.WebScriptResponse)8 ResourceWithMetadata (org.alfresco.rest.framework.core.ResourceWithMetadata)5 Matchers.anyString (org.mockito.Matchers.anyString)5 WebScriptRequest (org.springframework.extensions.webscripts.WebScriptRequest)4 HashMap (java.util.HashMap)3 Goat (org.alfresco.rest.framework.tests.api.mocks.Goat)2 Sheep (org.alfresco.rest.framework.tests.api.mocks.Sheep)2 ByteArrayOutputStream (java.io.ByteArrayOutputStream)1 RetryingTransactionHelper (org.alfresco.repo.transaction.RetryingTransactionHelper)1 Api (org.alfresco.rest.framework.Api)1 Grass (org.alfresco.rest.framework.tests.api.mocks.Grass)1 ApiAssistant (org.alfresco.rest.framework.tools.ApiAssistant)1 ResourceWebScriptDelete (org.alfresco.rest.framework.webscripts.ResourceWebScriptDelete)1 WithResponse (org.alfresco.rest.framework.webscripts.WithResponse)1 TransactionService (org.alfresco.service.transaction.TransactionService)1 JSONObject (org.json.JSONObject)1 Before (org.junit.Before)1 InvocationOnMock (org.mockito.invocation.InvocationOnMock)1