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

List of usage examples for org.springframework.mock.web MockHttpServletResponse getHeader

Introduction

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

Prototype

@Override
@Nullable
public String getHeader(String name) 

Source Link

Document

Return the primary value for the given header as a String, if any.

Usage

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

private static void assertResponse(MockHttpServletResponse response, int noOfHeaders, String etag, int month) {
    assertThat(response.getHeaderNames()).hasSize(noOfHeaders);
    assertThat(response.getHeader("ETag")).isEqualTo(etag);
    assertThat(response.getHeader("Cache-Control")).isEqualTo("public, max-age=" + month * 30 * 24 * 60 * 60);

    Long expiresMillis = (Long) response.getHeaderValue("Expires");
    DateTime expires = new DateTime(expiresMillis, DateTimeZone.UTC);
    DateTime inSixMonths = DateTime.now(DateTimeZone.UTC).plusSeconds(month * 30 * 24 * 60 * 60);
    assertThat(expires.getYear()).isEqualTo(inSixMonths.getYear());
    assertThat(expires.getMonthOfYear()).isEqualTo(inSixMonths.getMonthOfYear());
    assertThat(expires.getDayOfMonth()).isEqualTo(inSixMonths.getDayOfMonth());
    assertThat(expires.getHourOfDay()).isEqualTo(inSixMonths.getHourOfDay());
    assertThat(expires.getMinuteOfDay()).isEqualTo(inSixMonths.getMinuteOfDay());
}

From source file:com.vmware.identity.openidconnect.server.LoginTest.java

private static void assertErrorResponse(String loginString, String authzHeader, String expectedError,
        String expectedAuthzResponseHeader, String expectedAuthenticateHeader, CasIdmClient idmClient)
        throws Exception {
    Pair<ModelAndView, MockHttpServletResponse> result = doRequest(loginString, authzHeader,
            null /* sessionCookie */, idmClient);
    ModelAndView modelView = result.getLeft();
    MockHttpServletResponse response = result.getRight();
    Assert.assertNull("modelView", modelView);
    Assert.assertNull("sessionCookie", response.getCookie(SESSION_COOKIE_NAME));
    Assert.assertEquals("status", 401, response.getStatus());
    Object errorResponseHeader = response.getHeader("CastleError");
    Assert.assertNotNull("errorResponseHeader", errorResponseHeader);
    Assert.assertEquals("errorMessage", expectedError, response.getErrorMessage());

    if (expectedAuthzResponseHeader != null) {
        Object authzResponseHeader = response.getHeader("CastleAuthorization");
        Assert.assertNotNull("authzResponseHeader", authzResponseHeader);
        Assert.assertEquals("expectedAuthzResponseHeader", expectedAuthzResponseHeader,
                authzResponseHeader.toString());
    }//from   www  .  java 2  s.co  m

    if (expectedAuthenticateHeader != null) {
        Object wwwAuthenticateHeader = response.getHeader("WWW-Authenticate");
        Assert.assertNotNull("wwwAuthenticateHeader", wwwAuthenticateHeader);
        Assert.assertEquals("expectedAuthenticateHeader", expectedAuthenticateHeader,
                wwwAuthenticateHeader.toString());
    }
}

From source file:com.vmware.identity.openidconnect.server.LoginTest.java

private static void assertErrorResponseUsingPersonUserCert(String certHeader, Object certAttribute,
        Cookie cookie, String expectedError) throws Exception {
    Pair<ModelAndView, MockHttpServletResponse> result = doRequestUsingPersonUserCert(certHeader, certAttribute,
            cookie);/*from   ww  w  .j  a  v  a2  s. c o  m*/
    ModelAndView modelView = result.getLeft();
    MockHttpServletResponse response = result.getRight();

    Assert.assertNull("modelView", modelView);
    Assert.assertNull("sessionCookie", response.getCookie(SESSION_COOKIE_NAME));
    Assert.assertEquals("status", 401, response.getStatus());
    Assert.assertNotNull("errorResponseHeader", response.getHeader("CastleError"));
    Assert.assertEquals("errorMessage", expectedError, response.getErrorMessage());
}

From source file:com.mtt.myapp.common.util.FileDownloadUtilsTest.java

@Test
public void testDownloadFileHttpServletResponseString() throws IOException {
    File downFile = new ClassPathResource("TEST_USER.zip").getFile();
    String filePath = downFile.getAbsolutePath();
    MockHttpServletResponse resp = new MockHttpServletResponse();
    FileDownloadUtil.downloadFile(resp, filePath);
    String lengthHeader = resp.getHeader("Content-Length");

    assertThat(lengthHeader).isEqualTo(String.valueOf(downFile.length()));
}

From source file:org.ngrinder.common.util.FileDownloadUtilTest.java

@Test
public void testDownloadFileHttpServletResponseString() throws IOException {
    File downFile = new ClassPathResource("TEST_USER.zip").getFile();
    String filePath = downFile.getAbsolutePath();
    MockHttpServletResponse resp = new MockHttpServletResponse();
    FileDownloadUtil.downloadFile(resp, filePath);
    String lengthHeader = resp.getHeader("Content-Length");

    assertThat(lengthHeader, is(String.valueOf(downFile.length())));
}

From source file:org.ngrinder.common.util.FileDownloadUtilsTest.java

@Test
public void testDownloadFileHttpServletResponseString() throws IOException {
    File downFile = new ClassPathResource("TEST_USER.zip").getFile();
    String filePath = downFile.getAbsolutePath();
    MockHttpServletResponse resp = new MockHttpServletResponse();
    FileDownloadUtils.downloadFile(resp, filePath);
    String lengthHeader = resp.getHeader("Content-Length");

    assertThat(lengthHeader, is(String.valueOf(downFile.length())));
}

From source file:com.thoughtworks.go.server.web.FileViewTest.java

private long getContentLength(MockHttpServletResponse mockResponse) {
    return Long.parseLong(mockResponse.getHeader("Content-Length"));
}

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

@Test
public void testSendChallenge() throws Exception {
    final MockHttpServletResponse response = new MockHttpServletResponse();
    handler.sendChallenge(response);//  w ww .j av a 2s.c  o m
    assertEquals("basic realm=\"realm\"",
            response.getHeader(AbstractHttpAuthenticationHandler.AUTHENTICATE_HEADER));
    assertEquals(HttpServletResponse.SC_UNAUTHORIZED, response.getStatus());
}

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

@Test(dataProvider = "dois")
public void testDoGet(HttpServletRequest request, String expectedRedirect, String expectedReferral) {
    MockHttpServletResponse response = new MockHttpServletResponse();
    resolverServlet.doGet(request, response);

    assertEquals(response.getHeader(HTTP_HEADER_REFERRER), expectedReferral, "referral did not match");
    assertEquals(response.getRedirectedUrl(), expectedRedirect, "servlet didn't redirect correctly");
}

From source file:com.xemantic.tadedon.servlet.CacheDisablingFilterTest.java

@Test
public void shouldAddResponseHeaders() throws Exception {
    // given//from   w ww  . j  a va  2s.  c o m
    CacheDisablingFilter filter = new CacheDisablingFilter();
    MockHttpServletRequest request = new MockHttpServletRequest();
    MockHttpServletResponse response = new MockHttpServletResponse();
    MockFilterChain chain = new MockFilterChain();

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

    // then
    assertThat((String) response.getHeader("Pragma"), is("no-cache"));
    //TODO write more assertions
}