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

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

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

Introduction

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

Prototype

public void setCookies(@Nullable Cookie... cookies)

Source Link

Usage

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

private static Pair<ModelAndView, MockHttpServletResponse> doRequestUsingPersonUserCert(String certHeader,
        Object certAttribute, Cookie cookie) throws Exception {
    Map<String, String> queryParams = authnRequestParameters(Flow.AUTHZ_CODE);
    Map<String, String> formParams = new HashMap<String, String>();
    formParams.put("CastleAuthorization", "TLSClient ");
    MockHttpServletRequest request = TestUtil.createPostRequestWithQueryString(formParams, queryParams);
    if (certHeader != null) {
        request.addHeader("X-SSL-Client-Certificate", certHeader);
    }//from  w  ww  . ja  v a 2  s.c  o  m
    if (certAttribute != null) {
        request.setAttribute("javax.servlet.request.X509Certificate", certAttribute);
    }
    if (cookie != null) {
        request.setCookies(cookie);
    }

    MockHttpServletResponse response = new MockHttpServletResponse();
    ModelAndView modelView = authnController().authenticateByPersonUserCertificate(new ExtendedModelMap(),
            Locale.ENGLISH, request, response);
    return Pair.of(modelView, response);
}

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

private static Pair<ModelAndView, MockHttpServletResponse> doRequest(String loginString, String authzHeader,
        Cookie sessionCookie, CasIdmClient idmClient) throws Exception {
    Map<String, String> queryParams = authnRequestParameters(Flow.AUTHZ_CODE);

    MockHttpServletRequest request;
    if (loginString != null) {
        Map<String, String> formParams = new HashMap<String, String>();
        formParams.put("CastleAuthorization", loginString);
        request = TestUtil.createPostRequestWithQueryString(formParams, queryParams);
    } else {//  www .java  2 s  .c  o m
        request = TestUtil.createGetRequest(queryParams);
    }
    if (authzHeader != null) {
        request.addHeader("Authorization", authzHeader);
    }
    if (sessionCookie != null) {
        request.setCookies(sessionCookie);
    }

    MockHttpServletResponse response = new MockHttpServletResponse();
    AuthenticationController controller = authnController(idmClient);
    ModelAndView modelView = controller.authenticate(new ExtendedModelMap(), Locale.ENGLISH, request, response);
    return Pair.of(modelView, response);
}

From source file:org.jasig.cas.web.flow.SendTicketGrantingTicketActionTests.java

@Test
public void testTgtToSetRemovingOldTgt() throws Exception {
    final MockHttpServletResponse response = new MockHttpServletResponse();
    final MockHttpServletRequest request = new MockHttpServletRequest();
    final String TICKET_VALUE = "test";
    request.setCookies(new Cookie[] { new Cookie("TGT", "test5") });
    WebUtils.putTicketGrantingTicketInRequestScope(this.context, TICKET_VALUE);
    this.context.setExternalContext(new ServletExternalContext(new MockServletContext(), request, response));

    assertEquals("success", this.action.execute(this.context).getId());
    assertEquals(TICKET_VALUE, response.getCookies()[0].getValue());
}

From source file:org.jasig.cas.web.LogoutControllerTests.java

@Test
public void testLogoutCookie() throws Exception {
    MockHttpServletRequest request = new MockHttpServletRequest();
    Cookie cookie = new Cookie(COOKIE_TGC_ID, "test");
    request.setCookies(new Cookie[] { cookie });
    assertNotNull(this.logoutController.handleRequestInternal(request, new MockHttpServletResponse()));
}

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

public void testCookieRetrieve() {
    final MockHttpServletRequest request = new MockHttpServletRequest();
    final Cookie cookie = new Cookie("test", "test");
    cookie.setDomain("cas.org");
    cookie.setMaxAge(5);//from   w ww  .  j  a va2 s  .  co m
    request.setCookies(new Cookie[] { cookie });

    assertEquals("test", this.g.retrieveCookieValue(request));

}

From source file:org.jasig.cas.web.flow.GenerateServiceTicketActionTests.java

@Test
public void testServiceTicketFromCookie() throws Exception {
    MockRequestContext context = new MockRequestContext();
    context.getFlowScope().put("service", TestUtils.getService());
    context.getFlowScope().put("ticketGrantingTicketId", this.ticketGrantingTicket);
    MockHttpServletRequest request = new MockHttpServletRequest();
    context.setExternalContext(/*from  w ww  .j a  va 2  s . c o  m*/
            new ServletExternalContext(new MockServletContext(), request, new MockHttpServletResponse()));
    request.addParameter("service", "service");
    request.setCookies(new Cookie[] { new Cookie("TGT", this.ticketGrantingTicket) });

    this.action.execute(context);

    assertNotNull(WebUtils.getServiceTicketFromRequestScope(context));
}

From source file:eionet.webq.web.interceptor.CdrAuthorizationInterceptorTest.java

private MockHttpServletRequest requestWithCookies() {
    MockHttpServletRequest request = new MockHttpServletRequest();
    Cookie cookie = new Cookie("_ZopeId", "\"68673848A6sbSTxqyEQ\"");
    Cookie[] cookies = { cookie };/*from   w  w  w.jav  a  2  s  .com*/
    request.setCookies(cookies);
    return request;
}

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 ww w. j a v a  2 s  .  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);//from w  w  w. j  a  va  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:edu.internet2.middleware.shibboleth.idp.system.conf1.ShibbolethSSOTestCase.java

/** Tests second leg of the SSO request where request returns to SSO handler and AuthN statement is generated. */
public void testSecondAuthenticationLeg() throws Exception {
    MockServletContext servletContext = new MockServletContext();
    MockHttpServletRequest servletRequest = buildServletRequest();
    MockHttpServletResponse servletResponse = new MockHttpServletResponse();

    ProfileHandlerManager handlerManager = (ProfileHandlerManager) getApplicationContext()
            .getBean("shibboleth.HandlerManager");
    AbstractShibbolethProfileHandler handler = (AbstractShibbolethProfileHandler) handlerManager
            .getProfileHandler(servletRequest);
    assertNotNull(handler);/*from  w w  w.  j a v a2s . com*/

    HttpServletHelper.bindLoginContext(buildLoginContext(), handler.getStorageService(), servletContext,
            servletRequest, servletResponse);
    servletRequest.setCookies(servletResponse.getCookies());

    // Process request
    HTTPInTransport profileRequest = new HttpServletRequestAdapter(servletRequest);
    HTTPOutTransport profileResponse = new HttpServletResponseAdapter(servletResponse, false);
    handler.processRequest(profileRequest, profileResponse);

    String response = servletResponse.getContentAsString();
    assertTrue(response.contains("action=\"https&#x3a;&#x2f;&#x2f;example.org&#x2f;mySP\" method=\"post\""));
    assertTrue(response.contains("name=\"TARGET\" value=\"https&#x3a;&#x2f;&#x2f;example.org&#x2f;mySP\""));
    assertTrue(response.contains("SAMLResponse"));
}