Example usage for org.springframework.mock.web MockHttpServletRequest 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:com.nebhale.cyclinglibrary.web.GzipFilterTest.java

@Test
public void gzipRequest() throws ServletException, IOException {
    MockHttpServletRequest request = new MockHttpServletRequest();
    request.addHeader("Content-Encoding", "gzip");
    request.setContent(gzipContent("test-request-content"));

    MockHttpServletResponse response = new MockHttpServletResponse();
    MockFilterChain filterChain = new MockFilterChain();

    this.filter.doFilterInternal(request, response, filterChain);
    writeContent("test-response-content", filterChain.getResponse().getOutputStream());

    assertEquals("test-request-content", readContent(filterChain.getRequest().getInputStream()));
    assertEquals("test-response-content", response.getContentAsString());
}

From source file:org.ambraproject.doi.ResolverServletTest.java

@DataProvider(name = "dois")
public Object[][] getDois() {
    //start at 100 so we don't conflict with the DAO service test ids
    insertArticleRow(101, "info:doi/10.1371/ambr.1234567");
    insertArticleRow(102, "info:doi/10.1371/ovrj.v06.i09");
    insertArticleRow(103, "info:doi/10.1371/ovrj.1234567");
    insertAnnotationRow(101, "info:doi/10.1371/test-annotation", 103, "Reply");

    MockHttpServletRequest article1 = new MockHttpServletRequest();
    article1.addHeader(HTTP_HEADER_REFERRER, "http://test.referral.com/test");
    article1.setPathInfo("/10.1371/ambr.1234567");

    MockHttpServletRequest encodedArticle1 = new MockHttpServletRequest();
    encodedArticle1.setPathInfo("%2F10.1371%2Fambr.1234567");

    MockHttpServletRequest article2 = new MockHttpServletRequest();
    article2.addHeader(HTTP_HEADER_REFERRER, "http://test2.referral2.com/test2");
    article2.setPathInfo("%2F10.1371%2Fovrj.v06.i09");

    MockHttpServletRequest figure = new MockHttpServletRequest();
    figure.setPathInfo("%2F10.1371%2Fambr.1234567.g001");

    MockHttpServletRequest representation = new MockHttpServletRequest();
    representation.setPathInfo("%2F10.1371%2Fovrj.1234567.pdf");

    MockHttpServletRequest annotation = new MockHttpServletRequest();
    annotation.setPathInfo("%2F10.1371%2Ftest-annotation");

    MockHttpServletRequest suppInfo = new MockHttpServletRequest();
    suppInfo.setPathInfo("%2F10.1371%2Fovrj.1234567.s001");

    return new Object[][] {
            { article1, FIRST_JOURNAL_URL + "article/info%3Adoi%2F10.1371%2Fambr.1234567",
                    "http://test.referral.com/test" },
            { encodedArticle1, FIRST_JOURNAL_URL + "article/info%3Adoi%2F10.1371%2Fambr.1234567", null },
            { figure, FIRST_JOURNAL_URL + "article/info%3Adoi%2F10.1371%2Fambr.1234567#ambr-1234567-g001",
                    null },//from www.  ja va2s. c om
            { article2, SECOND_JOURNAL_URL + "article/info%3Adoi%2F10.1371%2Fovrj.v06.i09",
                    "http://test2.referral2.com/test2" },
            { representation,
                    SECOND_JOURNAL_URL + "article/fetchObjectAttachment.action?"
                            + "uri=info%3Adoi%2F10.1371%2Fovrj.1234567&representation=PDF",
                    null },
            { annotation, SECOND_JOURNAL_URL + "annotation/listThread.action?root=101", null },
            { suppInfo, SECOND_JOURNAL_URL + "article/fetchSingleRepresentation.action?"
                    + "uri=info%3Adoi%2F10.1371%2Fovrj.1234567.s001", null }, };
}

From source file:org.openmrs.module.emrapi.web.controller.BaseEmrControllerTest.java

/**
 * Creates a request from the given parameters.
 * <p>/*  w  w  w  .j  a v  a2  s  .c  o m*/
 * The requestURI is automatically preceded with "/rest/" + RestConstants.VERSION_1.
 *
 * @param method
 * @param requestURI
 * @return
 */
public MockHttpServletRequest request(RequestMethod method, String requestURI) {
    MockHttpServletRequest request = new MockHttpServletRequest(method.toString(), requestURI);
    request.addHeader("content-type", "application/json");
    return request;
}

From source file:com.google.api.server.spi.IoUtilTest.java

@Test
public void testGetRequestInputStream_headerWithPlaintext() throws Exception {
    MockHttpServletRequest request = new MockHttpServletRequest();
    request.setContent("test".getBytes(StandardCharsets.UTF_8));
    request.addHeader("Content-Encoding", "gzip");
    InputStream stream = IoUtil.getRequestInputStream(request);
    assertThat(stream).isNotInstanceOf(GZIPInputStream.class);
    assertThat(IoUtil.readStream(stream)).isEqualTo("test");
}

From source file:org.surfnet.oaaas.auth.AuthorizationServerFilterTest.java

@Test
public void testCorsHeadersPreflight() throws IOException, ServletException {
    MockHttpServletRequest request = new MockHttpServletRequest();
    request.addHeader(HttpHeaders.AUTHORIZATION, "bearer dummy-access-token");
    request.addHeader("Origin", "http://www.test.org");
    request.addHeader("Access-Control-Request-Method", "PATCH");
    request.addHeader("Access-Control-Request-Headers", HttpHeaders.AUTHORIZATION + ", X-Custom-Header");
    request.setMethod("OPTIONS");

    MockHttpServletResponse response = new MockHttpServletResponse();
    //null FilterChain as chain.filter to proceed the flow is not called with preflight
    filter.doFilter(request, response, null);

    assertEquals("86400", response.getHeader("Access-Control-Max-Age"));
    assertEquals(HttpHeaders.AUTHORIZATION + ", X-Custom-Header",
            response.getHeader("Access-Control-Allow-Headers"));
    assertEquals("GET, OPTIONS, HEAD, PUT, PATCH, POST, DELETE",
            response.getHeader("Access-Control-Allow-Methods"));
    assertEquals("*", response.getHeader("Access-Control-Allow-Origin"));

}

From source file:com.nebhale.cyclinglibrary.web.GzipFilterTest.java

@Test
public void gzipResponse() throws ServletException, IOException {
    MockHttpServletRequest request = new MockHttpServletRequest();
    request.addHeader("Accept-Encoding", "gzip");
    request.setContent("test-request-content".getBytes("UTF-8"));

    MockHttpServletResponse response = new MockHttpServletResponse();

    FilterChain filterChain = mock(FilterChain.class);
    doAnswer(new Answer<Void>() {

        @Override/*from w  w  w  .j a va  2 s  .  co m*/
        public Void answer(InvocationOnMock invocation) throws Throwable {
            HttpServletRequest request = (HttpServletRequest) invocation.getArguments()[0];
            assertEquals("test-request-content", readContent(request.getInputStream()));

            HttpServletResponse response = (HttpServletResponse) invocation.getArguments()[1];
            writeContent("test-response-content", response.getOutputStream());
            return null;
        }
    }).when(filterChain).doFilter(any(ServletRequest.class), any(ServletResponse.class));

    this.filter.doFilterInternal(request, response, filterChain);

    assertEquals("test-response-content", gunzipContent(response.getContentAsByteArray()));
}

From source file:com.google.api.server.spi.IoUtilTest.java

@Test
public void testReadRequestBody() throws Exception {
    MockHttpServletRequest request = new MockHttpServletRequest();
    request.setContent(compress("test".getBytes(StandardCharsets.UTF_8)));
    request.addHeader("Content-Encoding", "gzip");
    assertThat(IoUtil.readRequestBody(request)).isEqualTo("test");
}

From source file:com.google.api.server.spi.auth.GoogleAuthTest.java

@Test
public void testGetAuthToken_fromHeaders() {
    for (String scheme : GoogleAuth.ALLOWED_AUTH_SCHEMES) {
        MockHttpServletRequest request = new MockHttpServletRequest();

        request.addHeader(GoogleAuth.AUTHORIZATION_HEADER, scheme + " some-value");
        assertEquals("some-value", GoogleAuth.getAuthToken(request));
    }//from  w ww.j a  va2 s . c o m
    MockHttpServletRequest request = new MockHttpServletRequest();
    request.addHeader(GoogleAuth.AUTHORIZATION_HEADER, "noSuchAuthScheme some-value");
    assertNull(GoogleAuth.getAuthToken(request));
}

From source file:com.google.api.server.spi.IoUtilTest.java

@Test
public void testGetRequestInputStream() throws Exception {
    MockHttpServletRequest request = new MockHttpServletRequest();
    request.setContent(compress("test".getBytes(StandardCharsets.UTF_8)));
    request.addHeader("Content-Encoding", "gzip");
    InputStream stream = IoUtil.getRequestInputStream(request);
    assertThat(stream).isInstanceOf(GZIPInputStream.class);
    assertThat(IoUtil.readStream(stream)).isEqualTo("test");
}

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

@Test
public void testHandleCacheableResponseWithIfNoneMatch() throws IOException {
    byte[] data = "the response data".getBytes();
    String etag = "\"0" + DigestUtils.md5DigestAsHex(data) + '"';
    String contentType = "application/javascript;charset=UTF-8";

    MockHttpServletRequest request = new MockHttpServletRequest();
    request.addHeader("If-None-Match", etag);
    MockHttpServletResponse response = new MockHttpServletResponse();

    ExtDirectSpringUtil.handleCacheableResponse(request, response, data, contentType);
    assertThat(response.getStatus()).isEqualTo(304);

    request = new MockHttpServletRequest();
    request.addHeader("If-None-Match", etag);
    response = new MockHttpServletResponse();
    data = "new response data".getBytes();
    etag = "\"0" + DigestUtils.md5DigestAsHex(data) + '"';
    ExtDirectSpringUtil.handleCacheableResponse(request, response, data, contentType);
    assertThat(response.getStatus()).isEqualTo(200);
    assertResponse(response, 5, etag, 6);
    assertThat(response.getContentLength()).isEqualTo(data.length);
    assertThat(response.getContentType()).isEqualTo(contentType);
    assertThat(response.getContentAsByteArray()).isEqualTo(data);
}