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:io.jmnarloch.spring.request.correlation.filter.RequestCorrelationFilterTest.java

@Test
public void shouldUseExistingCorrelationId() throws IOException, ServletException {

    // given/*from ww w .  ja  v  a  2  s. c  om*/
    final String requestId = UUID.randomUUID().toString();
    final MockHttpServletRequest request = new MockHttpServletRequest();
    final MockHttpServletResponse response = new MockHttpServletResponse();
    final MockFilterChain chain = new MockFilterChain();

    request.addHeader(RequestCorrelationConsts.HEADER_NAME, requestId);

    // when
    instance.doFilter(request, response, chain);

    // then
    final Object requestCorrelation = request.getAttribute(RequestCorrelationConsts.ATTRIBUTE_NAME);
    assertNotNull(requestCorrelation);
    assertEquals(requestId, ((RequestCorrelation) requestCorrelation).getRequestId());

    final String header = ((HttpServletRequest) chain.getRequest())
            .getHeader(RequestCorrelationConsts.HEADER_NAME);
    assertNotNull(header);
    assertEquals(requestId, header);
}

From source file:io.jmnarloch.spring.request.correlation.filter.RequestCorrelationFilterTest.java

@Test
public void shouldUseCustomHeader() throws IOException, ServletException {

    // given/*w  w w  . j  a  v a2s  . c  om*/
    final String headerName = "X-TraceId";
    final String requestId = UUID.randomUUID().toString();
    final MockHttpServletRequest request = new MockHttpServletRequest();
    final MockHttpServletResponse response = new MockHttpServletResponse();
    final MockFilterChain chain = new MockFilterChain();

    request.addHeader(headerName, requestId);
    properties.setHeaderName(headerName);

    // when
    instance.doFilter(request, response, chain);

    // then
    final Object requestCorrelation = request.getAttribute(RequestCorrelationConsts.ATTRIBUTE_NAME);
    assertNotNull(requestCorrelation);
    assertEquals(requestId, ((RequestCorrelation) requestCorrelation).getRequestId());

    final String header = ((HttpServletRequest) chain.getRequest()).getHeader(headerName);
    assertNotNull(header);
    assertEquals(requestId, header);
}

From source file:ch.ralscha.extdirectspring.util.ExtDirectSpringUtilTest.java

@Test
public void testIsMultipart() {
    MockHttpServletRequest request = new MockHttpServletRequest();
    request.setMethod("POST");
    request.setRequestURI("/demo/controller/router");

    request.addHeader("Host", "eds.rasc.ch");
    request.addHeader("Connection", "keep-alive");
    request.addHeader("Content-Length", "8277");
    request.addHeader("Cache-Control", "max-age=0");
    request.addHeader("Origin", "http://eds.rasc.ch");
    request.addHeader("User-Agent",
            "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.1 (KHTML, like Gecko) Chrome/21.0.1180.75 Safari/537.1");
    request.addHeader("Content-Type", "multipart/form-data; boundary=----WebKitFormBoundaryux6D0mMa2PlsY016");

    assertThat(ExtDirectSpringUtil.isMultipart(request)).isTrue();

    request = new MockHttpServletRequest();
    request.setMethod("POST");
    request.setRequestURI("/demo/controller/router");

    request.addHeader("Host", "eds.rasc.ch");
    request.addHeader("Connection", "keep-alive");
    request.addHeader("Content-Length", "165");
    request.addHeader("Origin", "http://eds.rasc.ch");
    request.addHeader("X-Requested-With", "XMLHttpRequest");
    request.addHeader("User-Agent",
            "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.1 (KHTML, like Gecko) Chrome/21.0.1180.75 Safari/537.1");
    request.addHeader("Content-Type", "application/x-www-form-urlencoded; charset=UTF-8");

    assertThat(ExtDirectSpringUtil.isMultipart(request)).isFalse();

    request = new MockHttpServletRequest();
    request.setMethod("GET");
    request.setRequestURI("/demo/controller/router");
    request.addHeader("Host", "eds.rasc.ch");
    request.addHeader("Connection", "keep-alive");
    request.addHeader("User-Agent",
            "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.1 (KHTML, like Gecko) Chrome/21.0.1180.75 Safari/537.1");

    assertThat(ExtDirectSpringUtil.isMultipart(request)).isFalse();

}

From source file:org.cloudfoundry.identity.uaa.security.web.UaaRequestMatcherTests.java

@Test
public void pathMatcherMatchesWithSingleHeader() throws Exception {
    UaaRequestMatcher matcher = new UaaRequestMatcher("/somePath");
    matcher.setHeaders(Collections.singletonMap("Authorization", Collections.singletonList("Basic")));
    MockHttpServletRequest testRequest = request("/somePath", String.format("%s,%s",
            MediaType.APPLICATION_JSON.toString(), MediaType.APPLICATION_XML.toString()));
    testRequest.addHeader("Authorization", "Basic abc");
    assertTrue(matcher.matches(testRequest));
}

From source file:org.cloudfoundry.identity.uaa.security.web.UaaRequestMatcherTests.java

@Test
public void pathMatcherMatchesOneOfMultipleHeaders() throws Exception {
    UaaRequestMatcher matcher = new UaaRequestMatcher("/somePath");
    Map<String, List<String>> configMap = new HashMap<String, List<String>>();
    configMap.put("Authorization", Arrays.asList(new String[] { "Basic", "Bearer" }));
    matcher.setHeaders(configMap);/*from w  w w. j a  v a  2 s  . c  om*/
    MockHttpServletRequest testRequest = request("/somePath", String.format("%s,%s",
            MediaType.APPLICATION_JSON.toString(), MediaType.APPLICATION_XML.toString()));
    testRequest.addHeader("Authorization", "Basic abc");
    assertFalse(matcher.matches(testRequest));
}

From source file:org.cloudfoundry.identity.uaa.security.web.UaaRequestMatcherTests.java

@Test
public void pathMatcherDoesNotMatchOneOfMultipleHeaders() throws Exception {
    UaaRequestMatcher matcher = new UaaRequestMatcher("/somePath");
    Map<String, List<String>> configMap = new HashMap<String, List<String>>();
    configMap.put("Authorization", Arrays.asList(new String[] { "Basic", "Bearer" }));
    matcher.setHeaders(configMap);/*from  w  w  w .  j a  v  a2 s.c om*/
    MockHttpServletRequest testRequest = request("/somePath", String.format("%s,%s",
            MediaType.APPLICATION_JSON.toString(), MediaType.APPLICATION_XML.toString()));
    testRequest.addHeader("Authorization", "non matching header value");
    assertFalse(matcher.matches(testRequest));
}

From source file:org.cloudfoundry.identity.uaa.security.web.UaaRequestMatcherTests.java

private MockHttpServletRequest request(String path, String accept, String... parameters) {
    MockHttpServletRequest request = new MockHttpServletRequest();
    request.setContextPath("/ctx");
    request.setRequestURI("/ctx" + path);
    if (accept != null) {
        request.addHeader("Accept", accept);
    }/*from w w  w  .  j ava 2 s  . c om*/
    for (int i = 0; i < parameters.length; i += 2) {
        String key = parameters[i];
        String value = parameters[i + 1];
        request.addParameter(key, value);
    }
    return request;
}

From source file:com.github.jrialland.ajpclient.servlet.TestServletProxy.java

@Test
public void testMultiple() throws Exception {

    final int nTasks = 10;

    for (int i = 0; i < nTasks; i++) {
        final MockHttpServletRequest request = new MockHttpServletRequest();
        request.setMethod("GET");
        request.setRequestURI("/long_request");
        request.setMethod("POST");
        request.addHeader("Content-Type", "application/x-www-form-urlencoded");
        request.setContent("duration=1000".getBytes());
        final MockHttpServletResponse response = new MockHttpServletResponse();
        AjpServletProxy.forHost("localhost", getPort()).forward(request, response);
        Assert.assertEquals(HttpServletResponse.SC_OK, response.getStatus());
    }/*w w  w.  ja v a 2s  .c o  m*/
}

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

@Test
public void verifyNoTokenAndAuthHeaderIsBlank() throws Exception {
    final MockHttpServletRequest mockRequest = new MockHttpServletRequest("POST",
            CONTEXT + OAuthConstants.REVOKE_URL);
    mockRequest.addHeader("Authorization", "");
    mockRequest.setParameter(OAuthConstants.CLIENT_ID, CLIENT_ID);
    final MockHttpServletResponse mockResponse = new MockHttpServletResponse();

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

    final ModelAndView modelAndView = oauth20WrapperController.handleRequest(mockRequest, mockResponse);
    assertNull(modelAndView);/*from  w w w.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\":\""
            + OAuthConstants.MISSING_ACCESS_TOKEN_DESCRIPTION + "\"}";

    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.jasig.cas.support.oauth.web.OAuth20RevokeClientPrincipalTokensControllerTests.java

@Test
public void verifyNoTokenAndAuthHeaderIsMalformed() throws Exception {
    final MockHttpServletRequest mockRequest = new MockHttpServletRequest("POST",
            CONTEXT + OAuthConstants.REVOKE_URL);
    mockRequest.addHeader("Authorization", "Let me in i am authorized");
    mockRequest.setParameter(OAuthConstants.CLIENT_ID, CLIENT_ID);
    final MockHttpServletResponse mockResponse = new MockHttpServletResponse();

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

    final ModelAndView modelAndView = oauth20WrapperController.handleRequest(mockRequest, mockResponse);
    assertNull(modelAndView);//  ww  w .  ja va  2  s.co m
    assertEquals(HttpStatus.SC_BAD_REQUEST, mockResponse.getStatus());
    assertEquals(CONTENT_TYPE, mockResponse.getContentType());

    final String expected = "{\"error\":\"" + OAuthConstants.INVALID_REQUEST + "\",\"error_description\":\""
            + OAuthConstants.MISSING_ACCESS_TOKEN_DESCRIPTION + "\"}";

    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());
}