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

Introduction

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

Prototype

@Override
    @Nullable
    public Cookie[] getCookies() 

Source Link

Usage

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  ww . j  a  v  a  2  s .  co  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.springframework.test.web.servlet.htmlunit.HtmlUnitRequestBuilderTest.java

@Test
public void buildRequestCookiesNull() {
    MockHttpServletRequest actualRequest = requestBuilder.buildRequest(servletContext);

    assertThat(actualRequest.getCookies()).isNull();
}

---

From source file:org.springframework.test.web.servlet.htmlunit.HtmlUnitRequestBuilderTest.java

@Test
public void buildRequestCookiesSingle() {
    webRequest.setAdditionalHeader("Cookie", "name=value");

    MockHttpServletRequest actualRequest = requestBuilder.buildRequest(servletContext);

    Cookie[] cookies = actualRequest.getCookies();
    assertThat(cookies.length).isEqualTo(1);
    assertThat(cookies[0].getName()).isEqualTo("name");
    assertThat(cookies[0].getValue()).isEqualTo("value");
}

---

From source file:org.springframework.test.web.servlet.htmlunit.HtmlUnitRequestBuilderTest.java

@Test
public void buildRequestCookiesMulti() {
    webRequest.setAdditionalHeader("Cookie", "name=value; name2=value2");

    MockHttpServletRequest actualRequest = requestBuilder.buildRequest(servletContext);

    Cookie[] cookies = actualRequest.getCookies();
    assertThat(cookies.length).isEqualTo(2);
    Cookie cookie = cookies[0];//from w  w w. java 2s.  co  m
    assertThat(cookie.getName()).isEqualTo("name");
    assertThat(cookie.getValue()).isEqualTo("value");
    cookie = cookies[1];
    assertThat(cookie.getName()).isEqualTo("name2");
    assertThat(cookie.getValue()).isEqualTo("value2");
}

---

From source file:org.springframework.test.web.servlet.htmlunit.HtmlUnitRequestBuilderTests.java

@Test
public void buildRequestCookiesNull() {
    MockHttpServletRequest actualRequest = requestBuilder.buildRequest(servletContext);

    assertThat(actualRequest.getCookies(), nullValue());
}

---

From source file:org.springframework.test.web.servlet.htmlunit.HtmlUnitRequestBuilderTests.java

@Test
public void buildRequestCookiesSingle() {
    webRequest.setAdditionalHeader("Cookie", "name=value");

    MockHttpServletRequest actualRequest = requestBuilder.buildRequest(servletContext);

    Cookie[] cookies = actualRequest.getCookies();
    assertThat(cookies.length, equalTo(1));
    assertThat(cookies[0].getName(), equalTo("name"));
    assertThat(cookies[0].getValue(), equalTo("value"));
}

---

From source file:org.springframework.test.web.servlet.htmlunit.HtmlUnitRequestBuilderTests.java

@Test
public void buildRequestCookiesMulti() {
    webRequest.setAdditionalHeader("Cookie", "name=value; name2=value2");

    MockHttpServletRequest actualRequest = requestBuilder.buildRequest(servletContext);

    Cookie[] cookies = actualRequest.getCookies();
    assertThat(cookies.length, equalTo(2));
    Cookie cookie = cookies[0];//from w  w  w.  j  a va 2  s .co  m
    assertThat(cookie.getName(), equalTo("name"));
    assertThat(cookie.getValue(), equalTo("value"));
    cookie = cookies[1];
    assertThat(cookie.getName(), equalTo("name2"));
    assertThat(cookie.getValue(), equalTo("value2"));
}

---

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  av  a  2  s. c om*/
    }
    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));
    }
}

---