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:org.javalite.activeweb.RequestBuilder.java

private void addCookiesInternal(MockHttpServletRequest request) {
    List<javax.servlet.http.Cookie> servletCookieList = new ArrayList<javax.servlet.http.Cookie>();
    for (org.javalite.activeweb.Cookie cookie : cookies) {
        servletCookieList.add(org.javalite.activeweb.Cookie.toServletCookie(cookie));
    }/*from   w  ww . j a v a  2  s.c om*/
    javax.servlet.http.Cookie[] arr = servletCookieList.toArray(new javax.servlet.http.Cookie[0]);
    request.setCookies(arr);
}

From source file:edu.internet2.middleware.shibboleth.idp.system.conf1.SAML2SSOTestCase.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("urn:example.org:sp1");
    MockHttpServletResponse servletResponse = new MockHttpServletResponse();

    ProfileHandlerManager handlerManager = (ProfileHandlerManager) getApplicationContext()
            .getBean("shibboleth.HandlerManager");
    AbstractShibbolethProfileHandler handler = (AbstractShibbolethProfileHandler) handlerManager
            .getProfileHandler(servletRequest);
    assertNotNull(handler);//  w  ww  . ja  v  a2s.c  o  m

    HttpServletHelper.bindLoginContext(buildLoginContext("urn:example.org:sp1"), 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("SAMLResponse"));
}

From source file:edu.internet2.middleware.shibboleth.idp.system.conf1.ShibbolethSSOTestCase.java

/** Tests initial leg of the SSO request where request is decoded and sent to the authentication engine. */
public void testFirstAuthenticationLeg() 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 ww.  ja v  a 2 s  .co  m*/

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

    servletRequest.setCookies(servletResponse.getCookies());
    ShibbolethSSOLoginContext loginContext = (ShibbolethSSOLoginContext) HttpServletHelper
            .getLoginContext(handler.getStorageService(), servletContext, servletRequest);

    assertNotNull(loginContext);
    assertEquals(false, loginContext.getAuthenticationAttempted());
    assertEquals(false, loginContext.isForceAuthRequired());
    assertEquals(false, loginContext.isPassiveAuthRequired());
    assertEquals("/AuthnEngine", loginContext.getAuthenticationEngineURL());
    assertEquals("/shibboleth/SSO", loginContext.getProfileHandlerURL());
    assertEquals("urn:example.org:sp1", loginContext.getRelyingPartyId());
    assertEquals(0, loginContext.getRequestedAuthenticationMethods().size());
    assertEquals("https://example.org/mySP", loginContext.getSpAssertionConsumerService());
    assertEquals("https://example.org/mySP", loginContext.getSpAssertionConsumerService());

    assertTrue(servletResponse.getRedirectedUrl().endsWith("/AuthnEngine"));
}

From source file:edu.internet2.middleware.shibboleth.idp.system.conf1.SAML2SSOTestCase.java

/** Tests initial leg of the SSO request where request is decoded and sent to the authentication engine. */
public void testFirstAuthenticationLeg() throws Exception {
    MockServletContext servletContext = new MockServletContext();

    MockHttpServletRequest servletRequest = buildServletRequest("urn:example.org:sp1");
    MockHttpServletResponse servletResponse = new MockHttpServletResponse();

    ProfileHandlerManager handlerManager = (ProfileHandlerManager) getApplicationContext()
            .getBean("shibboleth.HandlerManager");
    AbstractShibbolethProfileHandler handler = (AbstractShibbolethProfileHandler) handlerManager
            .getProfileHandler(servletRequest);
    assertNotNull(handler);/*ww  w.j a  v  a  2 s  .  c  o m*/

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

    servletRequest.setCookies(servletResponse.getCookies());
    Saml2LoginContext loginContext = (Saml2LoginContext) HttpServletHelper
            .getLoginContext(handler.getStorageService(), servletContext, servletRequest);

    assertNotNull(loginContext);
    assertEquals(false, loginContext.getAuthenticationAttempted());
    assertEquals(false, loginContext.isForceAuthRequired());
    assertEquals(false, loginContext.isPassiveAuthRequired());
    assertEquals("/AuthnEngine", loginContext.getAuthenticationEngineURL());
    assertEquals("/saml2/POST/SSO", loginContext.getProfileHandlerURL());
    assertEquals("urn:example.org:sp1", loginContext.getRelyingPartyId());
    assertEquals(0, loginContext.getRequestedAuthenticationMethods().size());

    assertTrue(servletResponse.getRedirectedUrl().endsWith("/AuthnEngine"));
}

From source file:org.hdiv.filter.ValidatorHelperTest.java

/**
 * Test for cookies integrity./*from   w w w  .  j av  a 2  s .c  o  m*/
 */
public void testValidateCookiesIntegrity() {

    MockHttpServletRequest request = (MockHttpServletRequest) HDIVUtil.getHttpServletRequest();
    RequestWrapper requestWrapper = new RequestWrapper(request);

    MockHttpServletResponse response = new MockHttpServletResponse();
    ResponseWrapper responseWrapper = new ResponseWrapper(response);

    responseWrapper.addCookie(new Cookie("name", "value"));

    this.dataComposer.beginRequest(this.targetName);
    this.dataComposer.compose("param1", "value1", false);
    String pageState = this.dataComposer.endRequest();
    assertNotNull(pageState);
    request.addParameter(hdivParameter, pageState);

    this.dataComposer.endPage();

    // Modify cookie value on client
    request.setCookies(new Cookie[] { new Cookie("name", "changedValue") });

    requestWrapper = new RequestWrapper(request);
    boolean result = helper.validate(requestWrapper).isValid();
    assertFalse(result);
}

From source file:org.cateproject.test.functional.mockmvc.HtmlUnitRequestBuilder.java

private void cookies(MockHttpServletRequest result) {
    String cookieHeaderValue = header("Cookie");
    Cookie[] parentCookies = result.getCookies();
    List<Cookie> cookies = new ArrayList<Cookie>();
    if (cookieHeaderValue != null) {
        StringTokenizer tokens = new StringTokenizer(cookieHeaderValue, "=;");
        while (tokens.hasMoreTokens()) {
            String cookieName = tokens.nextToken().trim();
            if (!tokens.hasMoreTokens()) {
                throw new IllegalArgumentException("Expected value for cookie name " + cookieName
                        + ". Full cookie was " + cookieHeaderValue);
            }/*from w w w . j  a  v  a 2 s  . c o m*/
            String cookieValue = tokens.nextToken().trim();
            processCookie(result, cookies, new Cookie(cookieName, cookieValue));
        }
    }

    Set<com.gargoylesoftware.htmlunit.util.Cookie> managedCookies = cookieManager
            .getCookies(webRequest.getUrl());
    for (com.gargoylesoftware.htmlunit.util.Cookie cookie : managedCookies) {
        processCookie(result, cookies, new Cookie(cookie.getName(), cookie.getValue()));
    }
    if (parentCookies != null) {
        for (Cookie cookie : parentCookies) {
            cookies.add(cookie);
        }
    }
    if (!cookies.isEmpty()) {
        result.setCookies(cookies.toArray(new Cookie[0]));
    }
}

From source file:org.cateproject.test.functional.mockmvc.HtmlUnitRequestBuilder.java

private void parent(MockHttpServletRequest result, RequestBuilder parent) {
    if (parent == null) {
        return;//from   w ww.  j a  v a2 s  . c o  m
    }
    MockHttpServletRequest parentRequest = parent.buildRequest(result.getServletContext());

    // session
    HttpSession parentSession = parentRequest.getSession(false);
    if (parentSession != null) {
        Enumeration<String> attrNames = parentSession.getAttributeNames();
        while (attrNames.hasMoreElements()) {
            String attrName = attrNames.nextElement();
            Object attrValue = parentSession.getAttribute(attrName);
            result.getSession().setAttribute(attrName, attrValue);
        }
    }

    // header
    Enumeration<String> headerNames = parentRequest.getHeaderNames();
    while (headerNames.hasMoreElements()) {
        String attrName = headerNames.nextElement();
        Enumeration<String> attrValues = parentRequest.getHeaders(attrName);
        while (attrValues.hasMoreElements()) {
            String attrValue = attrValues.nextElement();
            result.addHeader(attrName, attrValue);
        }
    }

    // parameter
    Map<String, String[]> parentParams = parentRequest.getParameterMap();
    for (Map.Entry<String, String[]> parentParam : parentParams.entrySet()) {
        String paramName = parentParam.getKey();
        String[] paramValues = parentParam.getValue();
        result.addParameter(paramName, paramValues);
    }

    // cookie
    Cookie[] parentCookies = parentRequest.getCookies();
    if (parentCookies != null) {
        result.setCookies(parentCookies);
    }

    // request attribute
    Enumeration<String> parentAttrNames = parentRequest.getAttributeNames();
    while (parentAttrNames.hasMoreElements()) {
        String parentAttrName = parentAttrNames.nextElement();
        result.setAttribute(parentAttrName, parentRequest.getAttribute(parentAttrName));
    }
}

From source file:org.jasig.portal.portlet.container.services.PortletCookieServiceImplTest.java

/**
 * Test {@link PortletCookieServiceImpl#getOrCreatePortalCookie(javax.servlet.http.HttpServletRequest)}.
 * that results in returning an existing portalcookie from the request cookies
 *//*w  w  w. ja  v a 2s . c o m*/
@Test
public void testGetOrCreatePortalCookieGetExistingFromRequestCookies() {
    IPortletCookieDao portletCookieDao = EasyMock.createMock(IPortletCookieDao.class);
    MockPortalCookie portalCookie = new MockPortalCookie();
    portalCookie.setValue("ABCDEF");

    EasyMock.expect(portletCookieDao.getPortalCookie("ABCDEF")).andReturn(portalCookie);
    EasyMock.replay(portletCookieDao);

    PortletCookieServiceImpl cookieService = new PortletCookieServiceImpl();
    cookieService.setPortletCookieDao(portletCookieDao);

    MockHttpServletRequest request = new MockHttpServletRequest();
    Cookie[] cookies = new Cookie[1];
    Cookie cookie = new Cookie(IPortletCookieService.DEFAULT_PORTAL_COOKIE_NAME, "ABCDEF");
    cookies[0] = cookie;
    request.setCookies(cookies);

    IPortalCookie result = cookieService.getOrCreatePortalCookie(request);
    Assert.assertEquals(portalCookie, result);
    EasyMock.verify(portletCookieDao);
}

From source file:org.springframework.security.web.authentication.rememberme.TokenBasedRememberMeServicesTests.java

@Test
public void autoLoginIgnoresUnrelatedCookie() throws Exception {
    Cookie cookie = new Cookie("unrelated_cookie", "foobar");
    MockHttpServletRequest request = new MockHttpServletRequest();
    request.setCookies(cookie);
    MockHttpServletResponse response = new MockHttpServletResponse();

    Authentication result = services.autoLogin(request, response);

    assertThat(result).isNull();//from ww  w.j  av  a  2 s .c o m
    assertThat(response.getCookie(SPRING_SECURITY_REMEMBER_ME_COOKIE_KEY)).isNull();
}

From source file:org.springframework.security.web.authentication.rememberme.TokenBasedRememberMeServicesTests.java

@Test
public void autoLoginReturnsNullForExpiredCookieAndClearsCookie() throws Exception {
    Cookie cookie = new Cookie(SPRING_SECURITY_REMEMBER_ME_COOKIE_KEY, generateCorrectCookieContentForToken(
            System.currentTimeMillis() - 1000000, "someone", "password", "key"));
    MockHttpServletRequest request = new MockHttpServletRequest();
    request.setCookies(cookie);

    MockHttpServletResponse response = new MockHttpServletResponse();

    assertThat(services.autoLogin(request, response)).isNull();
    Cookie returnedCookie = response.getCookie(SPRING_SECURITY_REMEMBER_ME_COOKIE_KEY);
    assertThat(returnedCookie).isNotNull();
    assertThat(returnedCookie.getMaxAge()).isZero();
}