Example usage for org.springframework.mock.web MockHttpServletRequest addHeader

List of usage examples for org.springframework.mock.web MockHttpServletRequest addHeader

  1. HOME
  2. Java
  3. org.springframework
  4. org.springframework.mock.web.*
  5. MockHttpServletRequest
  6. addHeader

Introduction

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

Prototype

public void addHeader(String name, Object value)

Source Link

Document

Add an HTTP header entry for the given name.

Usage

From source file:org.openmrs.module.referenceapplication.page.controller.LoginPageControllerTest.java

/**
 * @see LoginPageController#get(org.openmrs.ui.framework.page.PageModel,
 *      org.openmrs.ui.framework.UiUtils, org.openmrs.ui.framework.page.PageRequest, String,
 *      org.openmrs.api.LocationService,
 *      org.openmrs.module.appframework.service.AppFrameworkService)
 *///  w  w w. j ava 2 s .  c  om
@Test
@Verifies(value = "should set the referer as the redirectUrl in the page model if no redirect param exists", method = "get(PageModel,UiUtils,PageRequest)")
public void get_shouldSetTheRefererAsTheRedirectUrlInThePageModelIfNoRedirectParamExists() throws Exception {
    when(Context.isAuthenticated()).thenReturn(false);

    String refererUrl = TEST_CONTEXT_PATH + "/referenceapplication/patient.page";
    MockHttpServletRequest request = new MockHttpServletRequest();
    request.setContextPath(TEST_CONTEXT_PATH);
    request.addHeader("Referer", refererUrl);
    PageModel pageModel = new PageModel();

    new LoginPageController().get(pageModel, uiUtils, createPageRequest(request, null), null, null,
            appFrameworkService);

    assertEquals(refererUrl, pageModel.get(REQUEST_PARAMETER_NAME_REDIRECT_URL));
}

From source file:org.openmrs.module.referenceapplication.page.controller.LoginPageControllerTest.java

/**
  * @see LoginPageController#get(PageModel,UiUtils,PageRequest,String,LocationService,AppFrameworkService)
  * @verifies set the referer as the redirectUrl in the page model if referer URL is within context path
  *///from   ww  w .  j a  va 2s  .  c om
@Test
public void get_shouldSetTheRefererAsTheRedirectUrlInThePageModelIfRefererUrlIsWithinContextPath()
        throws Exception {
    when(Context.isAuthenticated()).thenReturn(false);

    String redirectUrl = TEST_CONTEXT_PATH + "/demo/";
    String refererUrl = "http://openmrs.org" + redirectUrl;
    MockHttpServletRequest request = new MockHttpServletRequest();
    request.setContextPath(TEST_CONTEXT_PATH);
    request.addHeader("Referer", refererUrl);
    PageModel pageModel = new PageModel();
    new LoginPageController().get(pageModel, uiUtils, createPageRequest(request, null), null, null,
            appFrameworkService);

    assertEquals(redirectUrl, pageModel.get(REQUEST_PARAMETER_NAME_REDIRECT_URL));
}

From source file:org.hdiv.dataComposer.DataComposerFactoryTest.java

public void testNewInstanceAjaxNoParameter() {

    MockHttpServletRequest request = (MockHttpServletRequest) HDIVUtil.getHttpServletRequest();

    IDataComposer dataComposer = this.dataComposerFactory.newInstance(request);

    dataComposer.beginRequest("GET", "/ajax");
    String stateId = dataComposer.endRequest();
    dataComposer.endPage();/*from   w ww . jav  a  2s  .co  m*/

    // Create other instance
    request.addHeader("x-requested-with", "XMLHttpRequest");
    this.getConfig().setReuseExistingPageInAjaxRequest(true);

    dataComposer = this.dataComposerFactory.newInstance(request);

    dataComposer.beginRequest("GET", "/ajax");
    String stateId2 = dataComposer.endRequest();
    dataComposer.endPage();

    assertFalse(getPageId(stateId) == getPageId(stateId2));
}

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

@Test
public void verifyOKWithAuthHeader() throws Exception {
    final AccessToken accessToken = mock(AccessToken.class);
    when(accessToken.getId()).thenReturn(AT_ID);

    final List<PrincipalMetadata> principalMetas = Arrays.asList(
            new PrincipalMetadata(CLIENT_ID, PRINC_NAME_ONE, PRINC_DESCR_ONE),
            new PrincipalMetadata(CLIENT_ID, PRINC_NAME_TWO, PRINC_DESCR_TWO));

    final CentralOAuthService centralOAuthService = mock(CentralOAuthService.class);
    when(centralOAuthService.getToken(AT_ID, AccessToken.class)).thenReturn(accessToken);
    when(centralOAuthService.getPrincipalMetadata(accessToken)).thenReturn(principalMetas);

    final MockHttpServletRequest mockRequest = new MockHttpServletRequest("POST",
            CONTEXT + OAuthConstants.METADATA_URL);
    mockRequest.addHeader("Authorization", OAuthConstants.BEARER_TOKEN + " " + AT_ID);
    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   ww  w.j  ava 2  s  .  c o m
    assertEquals(HttpStatus.SC_OK, mockResponse.getStatus());
    assertEquals(CONTENT_TYPE, mockResponse.getContentType());

    final ObjectMapper mapper = new ObjectMapper();

    final String expected = "{\"data\":[" + "{\"client_id\":\"" + CLIENT_ID + "\",\"name\":\"" + PRINC_NAME_ONE
            + "\",\"description\":\"" + PRINC_DESCR_ONE + "\",\"scope\":[]}," + "{\"client_id\":\"" + CLIENT_ID
            + "\",\"name\":\"" + PRINC_NAME_TWO + "\",\"description\":\"" + PRINC_DESCR_TWO
            + "\",\"scope\":[]}]}";
    final JsonNode expectedObj = mapper.readTree(expected);
    final JsonNode receivedObj = mapper.readTree(mockResponse.getContentAsString());

    final JsonNode expectedData = expectedObj.get("data");
    final JsonNode receivedData = receivedObj.get("data");
    assertEquals(expectedData.size(), receivedData.size());

    final JsonNode expectedPrincipalOne = expectedData.get(0);
    final JsonNode receivedPrincipalOne = receivedData.get(0);
    assertEquals(expectedPrincipalOne.get("client_id"), receivedPrincipalOne.get("client_id"));
    assertEquals(expectedPrincipalOne.get("name"), receivedPrincipalOne.get("name"));
    assertEquals(expectedPrincipalOne.get("description"), receivedPrincipalOne.get("description"));
    assertEquals(expectedPrincipalOne.get("scope").size(), receivedPrincipalOne.get("scope").size());

    final JsonNode expectedPrincipalTwo = expectedData.get(1);
    final JsonNode receivedPrincipalTwo = receivedData.get(1);
    assertEquals(expectedPrincipalTwo.get("client_id"), receivedPrincipalTwo.get("client_id"));
    assertEquals(expectedPrincipalTwo.get("name"), receivedPrincipalTwo.get("name"));
    assertEquals(expectedPrincipalTwo.get("description"), receivedPrincipalTwo.get("description"));
    assertEquals(expectedPrincipalTwo.get("scope").size(), receivedPrincipalTwo.get("scope").size());
}

From source file:com.tasktop.c2c.server.common.service.tests.ajp.AjpProtocolTest.java

@Test
public void testPostData() throws IOException {
    Payload payload = new Payload(HttpServletResponse.SC_CREATED, "some content\none two three\n\nfour");
    payload.getResponseHeaders().put("foo", "bar");
    TestServlet.setResponsePayload(payload);

    String formContent = "a=b&c=def";
    byte[] requestContent = formContent.getBytes();

    MockHttpServletRequest request = new MockHttpServletRequest();
    request.setMethod("GET");
    request.setRequestURI("/testPostData");
    request.addHeader("Content-Type", "application/x-www-form-urlencoded");
    request.addHeader("Content-Length", requestContent.length);
    request.setContent(requestContent);/*ww w . j ava  2  s.  co  m*/

    MockHttpServletResponse response = new MockHttpServletResponse();
    protocol.forward(request, response);

    assertRequestIsExpected(request, TestServlet.getLastRequest());
    assertResponseIsExpected(payload, response);
}

From source file:org.sventon.web.HttpBasicAuthenticationHandlerTest.java

@Test
public void testParseCredentials() throws Exception {
    final MockHttpServletRequest request = new MockHttpServletRequest();
    final HttpBasicAuthenticationHandler handler = new HttpBasicAuthenticationHandler();

    Credentials credentials;// w  w  w. j a va2 s  . c o m

    try {
        handler.parseCredentials(request);
        fail("Should cause IllegalArgumentException");
    } catch (IllegalArgumentException iae) {
        // Expected
    }

    final String credentialString = "uid:pwd";

    request.addHeader(AbstractHttpAuthenticationHandler.AUTHORIZATION_HEADER,
            "Basic " + new String(Base64.encodeBase64(credentialString.getBytes())));
    credentials = handler.parseCredentials(request);
    assertFalse(credentials.isEmpty());
    assertEquals("uid", credentials.getUserName());
    assertEquals("pwd", credentials.getPassword());
}

From source file:com.ge.predix.web.cors.test.CORSFilterTest.java

@Test
public void testRequestWithForbiddenUri() throws ServletException, IOException {
    CORSFilter corsFilter = createConfiguredCORSFilter();

    MockHttpServletRequest request = new MockHttpServletRequest("GET", "/uaa/login");
    request.addHeader("Origin", "example.com");
    request.addHeader("X-Requested-With", "XMLHttpRequest");

    MockHttpServletResponse response = new MockHttpServletResponse();

    FilterChain filterChain = newMockFilterChain();

    corsFilter.doFilter(request, response, filterChain);

    Assert.assertEquals(403, response.getStatus());
}

From source file:com.enonic.cms.framework.util.HttpServletRangeUtilTest.java

@Test
public void test_bad_symbols_in_range() throws Exception {
    final MockHttpServletRequest httpServletRequest = new MockHttpServletRequest();
    httpServletRequest.setMethod("GET");
    httpServletRequest.setPathInfo("/input.dat");
    httpServletRequest.addHeader(HttpHeaders.RANGE, "bytes=bad");

    final MockHttpServletResponse mockHttpServletResponse = new MockHttpServletResponse();
    HttpServletRangeUtil.processRequest(httpServletRequest, mockHttpServletResponse, "input.dat",
            "application/pdf", INPUT_FILE, false);

    assertEquals("", mockHttpServletResponse.getContentAsString());

    assertEquals(HttpServletResponse.SC_REQUESTED_RANGE_NOT_SATISFIABLE, mockHttpServletResponse.getStatus());
}

From source file:com.enonic.cms.framework.util.HttpServletRangeUtilTest.java

@Test
public void test_bad_range() throws Exception {
    final MockHttpServletRequest httpServletRequest = new MockHttpServletRequest();
    httpServletRequest.setMethod("GET");
    httpServletRequest.setPathInfo("/input.dat");
    httpServletRequest.addHeader(HttpHeaders.RANGE, "bytes=5-1");

    final MockHttpServletResponse mockHttpServletResponse = new MockHttpServletResponse();
    HttpServletRangeUtil.processRequest(httpServletRequest, mockHttpServletResponse, "input.dat",
            "application/pdf", INPUT_FILE, false);

    assertEquals("", mockHttpServletResponse.getContentAsString());

    assertEquals(HttpServletResponse.SC_REQUESTED_RANGE_NOT_SATISFIABLE, mockHttpServletResponse.getStatus());
}

From source file:com.enonic.cms.framework.util.HttpServletRangeUtilTest.java

@Test
public void test_out_of_range() throws Exception {
    final MockHttpServletRequest httpServletRequest = new MockHttpServletRequest();
    httpServletRequest.setMethod("GET");
    httpServletRequest.setPathInfo("/input.dat");
    httpServletRequest.addHeader(HttpHeaders.RANGE, "bytes=50000-50100");

    final MockHttpServletResponse mockHttpServletResponse = new MockHttpServletResponse();
    HttpServletRangeUtil.processRequest(httpServletRequest, mockHttpServletResponse, "input.dat",
            "application/pdf", INPUT_FILE, false);

    assertEquals("", mockHttpServletResponse.getContentAsString());

    assertEquals(HttpServletResponse.SC_REQUESTED_RANGE_NOT_SATISFIABLE, mockHttpServletResponse.getStatus());
}