Example usage for org.springframework.mock.web MockHttpServletResponse MockHttpServletResponse

Introduction

In this page you can find the example usage for org.springframework.mock.web MockHttpServletResponse MockHttpServletResponse.

Prototype

MockHttpServletResponse

Source Link

Usage

From source file:org.ambraproject.wombat.util.HttpMessageUtilTest.java

@Test
public void testCopyResponseWithHeaders() throws IOException {
    byte[] testContent = "Test content".getBytes();

    HttpResponse input = new BasicHttpResponse(null, HttpStatus.OK.value(), "");
    BasicHttpEntity outputEntity = new BasicHttpEntity();
    outputEntity.setContent(new ByteArrayInputStream(testContent));
    input.setEntity(outputEntity);/* www . jav a2s  .co  m*/
    input.setHeader("includeMe", "foo");
    input.setHeader("excludeMe", "bar");
    input.setHeader("alterMe", "toBeAltered");

    HeaderFilter headerFilter = header -> {
        String name = header.getName();
        if ("includeMe".equalsIgnoreCase(name))
            return header.getValue();
        if ("alterMe".equalsIgnoreCase(name))
            return "altered";
        return null;
    };

    MockHttpServletResponse output = new MockHttpServletResponse();

    copyResponseWithHeaders(input, output, headerFilter);

    assertEquals(output.getContentAsByteArray(), testContent);

    assertEquals(output.getHeaderNames().size(), 2);
    assertEquals(output.getHeaders("includeMe"), ImmutableList.of("foo"));
    assertEquals(output.getHeaders("alterMe"), ImmutableList.of("altered"));
    assertEquals(output.getHeaders("excludeMe"), ImmutableList.of());
}

From source file:org.jasig.cas.support.oauth.web.OAuth20RevokeClientTokensControllerTests.java

@Test
public void verifyNoClientId() throws Exception {
    final CentralOAuthService centralOAuthService = mock(CentralOAuthService.class);
    when(centralOAuthService.revokeClientTokens(CLIENT_ID, CLIENT_SECRET)).thenReturn(true);

    final MockHttpServletRequest mockRequest = new MockHttpServletRequest("POST",
            CONTEXT + OAuthConstants.REVOKE_URL);
    mockRequest.setParameter(OAuthConstants.CLIENT_ID, "");
    mockRequest.setParameter(OAuthConstants.CLIENT_SECRET, CLIENT_SECRET);
    final MockHttpServletResponse mockResponse = new MockHttpServletResponse();

    final OAuth20WrapperController oauth20WrapperController = new OAuth20WrapperController();
    oauth20WrapperController.setCentralOAuthService(centralOAuthService);
    oauth20WrapperController.afterPropertiesSet();

    final ModelAndView modelAndView = oauth20WrapperController.handleRequest(mockRequest, mockResponse);
    assertNull(modelAndView);//from www.  ja v  a 2 s  .  com
    assertEquals(HttpStatus.SC_BAD_REQUEST, mockResponse.getStatus());
    assertEquals(CONTENT_TYPE, mockResponse.getContentType());

    final String expected = "{\"error\":\"" + OAuthConstants.INVALID_REQUEST + "\",\"error_description\":\""
            + "Invalid or missing parameter 'client_id'\"}";

    final ObjectMapper mapper = new ObjectMapper();
    final JsonNode expectedObj = mapper.readTree(expected);
    final JsonNode receivedObj = mapper.readTree(mockResponse.getContentAsString());
    assertEquals(expectedObj.get("error").asText(), receivedObj.get("error").asText());
    assertEquals(expectedObj.get("error_description").asText(), receivedObj.get("error_description").asText());
}

From source file:org.openmrs.contrib.metadatarepository.webapp.controller.PackageFormControllerTest.java

@Test
public void testRemove() throws Exception {
    request = newPost("/packageform");
    request.addParameter("delete", "");
    pkg = new MetadataPackage();
    pkg.setId(1L);/*from   w w  w. jav a2s . c om*/

    BindingResult errors = new DataBinder(pkg).getBindingResult();
    form.onSubmit(pkg, errors, request, new MockHttpServletResponse(), new ModelMap());

    assertNotNull(request.getSession().getAttribute("successMessages"));
}

From source file:org.jasig.cas.web.LogoutControllerTests.java

@Test
public void testLogoutNoCookie() throws Exception {
    assertNotNull(this.logoutController.handleRequestInternal(new MockHttpServletRequest(),
            new MockHttpServletResponse()));
}

From source file:org.openxdata.server.servlet.StudyExportServletTest.java

@Before
public void createServletAndMocks() {
    PowerMock.mockStatic(StudyExport.class);
    PowerMock.mockStatic(FormExport.class);
    PowerMock.mockStatic(VersionExport.class);
    dataExportMock = PowerMock.createMock(DataExportService.class);
    request = new MockHttpServletRequest();
    response = new MockHttpServletResponse();

    servlet = new StudyExportServlet();
    servlet.setDataExportService(dataExportMock);
}

From source file:org.openmrs.contrib.metadatarepository.webapp.controller.UserFormControllerTest.java

@Test
public void testCancel() throws Exception {
    log.debug("testing cancel...");
    request = newPost("/userform.html");
    request.addParameter("cancel", "");

    BindingResult errors = new DataBinder(user).getBindingResult();
    String view = c.onSubmit(user, errors, request, new MockHttpServletResponse());

    assertEquals("redirect:/mainMenu", view);
}

From source file:org.openmrs.module.webservices.rest.web.v1_0.controller.PrivilegeControllerTest.java

/**
 * @see PrivilegeController#create(SimpleObject, javax.servlet.http.HttpServletRequest, HttpServletResponse)
 * @verifies create a new Privilege/*from  w ww .j ava  2s.  c  o  m*/
 */
@Test
public void createPrivilege_shouldCreateANewPrivilege() throws Exception {
    int before = Context.getUserService().getAllPrivileges().size();
    String json = "{\"name\":\"test\",\"description\":\"This is a test privilege for PrivilegeControllerTest.\"}}";
    SimpleObject post = new ObjectMapper().readValue(json, SimpleObject.class);
    Object newPrivilege = new PrivilegeController().create(post, new MockHttpServletRequest(),
            new MockHttpServletResponse());
    Assert.assertEquals(before + 1, Context.getUserService().getAllPrivileges().size());
}

From source file:piazza.services.ingest.test.ControllerTests.java

/**
 * Tests ES initialization// w  ww . ja v a  2s .c  om
 */
@Test
public void testInit() throws Exception {
    // Ensuring no Exceptions are thrown
    MockHttpServletResponse mockResponse = new MockHttpServletResponse();
    controller.init();
}

From source file:com.xemantic.tadedon.guice.servlet.mock.FakeServletContainerTest.java

/**
 * Tests {@link UriReturningServlet#doGet(javax.servlet.http.HttpServletRequest, javax.servlet.http.HttpServletResponse)}.
 *
 * @throws IOException if an input or output error is detected during request handling
 * @throws ServletException if container cannot handle request. 
 *///from   w  w  w  . j  a  va  2  s.  c o  m
@Test
public void shouldReturnUriAndAddFooHeader() throws IOException, ServletException {
    // given
    MockHttpServletRequest request = servletContainer.newRequest("GET", "/UriReturningServlet");
    MockHttpServletResponse response = new MockHttpServletResponse();

    // when
    servletContainer.service(request, response);

    // then
    assertThat(response.getContentAsString(), is("/UriReturningServlet"));
    assertThat((String) response.getHeader("Foo"), is("foo"));
}

From source file:com.hp.autonomy.frontend.find.idol.view.IdolViewControllerTest.java

@Test
public void noReferenceField() {
    when(configService.getConfig())//from   w w  w.  j ava  2s  .  c  o  m
            .thenReturn(new IdolFindConfig.Builder().setView(new ViewConfig.Builder().build()).build());

    assertNotNull(viewController.handleViewNoReferenceFieldException(
            new ViewNoReferenceFieldException("some reference", "some field"), new MockHttpServletRequest(),
            new MockHttpServletResponse()));
}