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

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

Introduction

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

Prototype

@Nullable
    public Cookie getCookie(String name) 

Source Link

Usage

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 www .  j  av  a 2 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.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());
    }//  w  w  w .j  a v  a  2 s .  c  o m

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

From source file:org.jasig.cas.web.support.CookieRetrievingCookieGeneratorTests.java

public void testCookieAddWithoutRememberMe() {
    final MockHttpServletRequest request = new MockHttpServletRequest();
    final MockHttpServletResponse response = new MockHttpServletResponse();

    this.g.addCookie(request, response, "test");

    final Cookie c = response.getCookie("test");
    assertEquals(5, c.getMaxAge());/*from  ww w .ja va 2s  . co  m*/
    assertEquals("test", c.getValue());
}

From source file:org.jasig.cas.web.support.CookieRetrievingCookieGeneratorTests.java

public void testCookieAddWithRememberMe() {
    final MockHttpServletRequest request = new MockHttpServletRequest();
    request.addParameter(RememberMeCredentials.REQUEST_PARAMETER_REMEMBER_ME, "true");
    final MockHttpServletResponse response = new MockHttpServletResponse();

    this.g.addCookie(request, response, "test");

    final Cookie c = response.getCookie("test");
    assertEquals(100, c.getMaxAge());/*  w w  w .  java 2s  . c  om*/
    assertEquals("test", c.getValue());
}

From source file:au.gov.dto.dibp.appointments.security.csrf.CookieBasedCsrfTokenRepositoryTest.java

@Test
public void testSaveAndLoadToken() throws Exception {
    CookieBasedCsrfTokenRepository repo = new CookieBasedCsrfTokenRepository();
    MockHttpServletRequest request = new MockHttpServletRequest();
    MockHttpServletResponse response = new MockHttpServletResponse();
    CsrfToken token = repo.generateToken(request);
    repo.saveToken(token, request, response);

    Cookie cookie = response.getCookie(token.getParameterName());
    assertNotNull(cookie);/*from  w  w  w .j  av  a2s  . c  o m*/
    assertEquals(token.getToken(), cookie.getValue());
    assertEquals(true, cookie.isHttpOnly());

    request.setCookies(cookie);

    CsrfToken saved = repo.loadToken(request);
    assertEquals(token.getToken(), saved.getToken());
    assertEquals(token.getHeaderName(), saved.getHeaderName());
    assertEquals(token.getParameterName(), saved.getParameterName());
}

From source file:au.gov.dto.springframework.security.web.csrf.CookieCsrfTokenRepositoryTest.java

@Test
public void testSaveAndLoadToken() throws Exception {
    CookieCsrfTokenRepository csrfTokenRepository = new CookieCsrfTokenRepository();
    String csrfCookieName = "csrfCookie";
    csrfTokenRepository.setCsrfCookieName(csrfCookieName);
    MockHttpServletRequest request = new MockHttpServletRequest();
    MockHttpServletResponse response = new MockHttpServletResponse();
    CsrfToken token = csrfTokenRepository.generateToken(request);
    csrfTokenRepository.saveToken(token, request, response);

    Cookie cookie = response.getCookie(csrfCookieName);
    assertNotNull(cookie);//ww w  .java 2  s.c o  m
    assertEquals(token.getToken(), cookie.getValue());
    assertEquals(true, cookie.isHttpOnly());

    request.setCookies(cookie);

    CsrfToken saved = csrfTokenRepository.loadToken(request);
    assertEquals(token.getToken(), saved.getToken());
    assertEquals(token.getHeaderName(), saved.getHeaderName());
    assertEquals(token.getParameterName(), saved.getParameterName());
}

From source file:org.unidle.social.SignInAdapterImplTest.java

@Test
public void testSignIn() throws Exception {
    final ConnectionStub<Object> connection = new ConnectionStub<>(
            new ConnectionData("provider id", "provider user id", "display name", "profile url", "image url",
                    "access token", "secret", "refresh token", 1234L));

    final MockHttpServletResponse response = new MockHttpServletResponse();

    subject.signIn("userId", connection, new ServletWebRequest(new MockHttpServletRequest(), response));

    final Object principal = SecurityContextHolder.getContext().getAuthentication().getPrincipal();

    assertThat(principal).isEqualTo("userId");
    assertThat(response.getCookie(LAST_LOGIN_SOURCE.getName()).getValue()).isEqualTo("provider id");
}

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

@Test
public void testMissingLogin() throws Exception {
    Pair<ModelAndView, MockHttpServletResponse> result = doRequest(null /* loginString */,
            null /* sessionCookie */);
    ModelAndView modelView = result.getLeft();
    MockHttpServletResponse response = result.getRight();
    Assert.assertNotNull("modelView", modelView); // logon form should be served
    Assert.assertNull("sessionCookie", response.getCookie(SESSION_COOKIE_NAME));
    Assert.assertEquals("status", 200, response.getStatus());
}

From source file:fr.mby.portal.coreimpl.EndToEndTest.java

/**
 * End to end test.//from  w ww  .j a v a2s  . c  om
 * 
 * @throws Exception
 */
@Test
public void testDispatch() throws Exception {
    final MockHttpServletRequest request = new MockHttpServletRequest();
    final MockHttpServletResponse response = new MockHttpServletResponse();

    this.sessionManager.initPortalSession(request, response);

    this.userActionDispatcher.dispatch(request, response);

    // Test headers
    final String actionHeader1 = response.getHeader("actionProp1");
    final String renderHeader1 = response.getHeader("renderProp1");
    Assert.assertEquals("Bad action header value !", "actionVal1", actionHeader1);
    Assert.assertEquals("Bad render header value !", "renderVal1", renderHeader1);

    final Cookie actionCookie1 = response.getCookie("actionCookie1");
    final Cookie renderCookie1 = response.getCookie("renderCookie1");
    Assert.assertNotNull("Action cookie is null !", actionCookie1);
    Assert.assertNotNull("Render cookie is null !", renderCookie1);
    Assert.assertEquals("Bad action cookie value !", "actionCookieVal1", actionCookie1.getValue());
    Assert.assertEquals("Bad render cookie value !", "renderCookieVal1", renderCookie1.getValue());

    // Test response
    response.flushBuffer();
    final String reponseOutputStream = response.getContentAsString();
    Assert.assertEquals("Bad response output stream !", "<html><body><h1>Test</h1></body></html>",
            reponseOutputStream);
}

From source file:fragment.web.AuthenticationControllerTest.java

@Test
public void testLoggedOut() throws Exception {
    asRoot();//w w  w.  j  av  a2s.  c  o m
    User user = getSystemTenant().getOwner();
    MockHttpServletResponse response = new MockHttpServletResponse();
    MockHttpServletRequest request = new MockHttpServletRequest();
    String view = controller.loggedout(user.getUuid(), map, session, response, request);
    Cookie cookie = response.getCookie("JforumSSO");
    Assert.assertEquals(cookie.getValue(), "");
    Assert.assertTrue(session.isInvalid());
    if (config.getAuthenticationService().compareToIgnoreCase("cas") == 0) {
        Assert.assertEquals("redirect:" + config.getCasLogoutUrl() + "?service="
                + URLEncoder.encode(config.getCasServiceUrl(), "UTF-8"), view);
    } else {
        Assert.assertEquals("redirect:/j_spring_security_logout", view);
    }

}