import com.ning.http.client.cookie.Cookie; //导入依赖的package包/类
@Test
public void testGetCookie() throws Exception {
    NewCookie cookie1 = ParsecHttpUtil.getCookie(new Cookie(
        "cookie1_name",
        "cookie1_value",
        false,
        null,
        "cookie1_path",
        1,
        true,
        true
    ));

    assertEquals("cookie1_name", cookie1.getName());
    assertEquals("cookie1_value", cookie1.getValue());
    assertEquals(null, cookie1.getDomain());
    assertEquals("cookie1_path", cookie1.getPath());
    assertEquals(null, cookie1.getExpiry());
    assertEquals(1, cookie1.getMaxAge());
    assertTrue(cookie1.isSecure());
    assertTrue(cookie1.isHttpOnly());
}
 

import com.ning.http.client.cookie.Cookie; //导入依赖的package包/类
protected void recordCookie(com.ning.http.client.Request httpRequest, Trace trace) {
    if (cookieSampler.isSampling()) {
        Collection<Cookie> cookies = httpRequest.getCookies();

        if (cookies.isEmpty()) {
            return;
        }

        StringBuilder sb = new StringBuilder(cookieDumpSize * 2);
        Iterator<Cookie> iterator = cookies.iterator();
        while (iterator.hasNext()) {
            Cookie cookie = iterator.next();
            sb.append(cookie.getName()).append("=").append(cookie.getValue());
            if (iterator.hasNext()) {
                sb.append(",");
            }
        }
        trace.recordAttribute(AnnotationKey.HTTP_COOKIE, StringUtils.drop(sb.toString(), cookieDumpSize));
    }
}
 

import com.ning.http.client.cookie.Cookie; //导入依赖的package包/类
private void recordCookie(Request httpRequest, SpanEventRecorder recorder) {
    if (cookieSampler.isSampling()) {
        Collection<Cookie> cookies = httpRequest.getCookies();

        if (cookies.isEmpty()) {
            return;
        }

        StringBuilder sb = new StringBuilder(config.getCookieDumpSize() * 2);
        Iterator<Cookie> iterator = cookies.iterator();
        while (iterator.hasNext()) {
            Cookie cookie = iterator.next();
            sb.append(cookie.getName()).append('=').append(cookie.getValue());
            if (iterator.hasNext()) {
                sb.append(',');
            }
        }
        recorder.recordAttribute(AnnotationKey.HTTP_COOKIE, StringUtils.abbreviate(sb.toString(), config.getCookieDumpSize()));
    }
}
 

import com.ning.http.client.cookie.Cookie; //导入依赖的package包/类
public void storeCookies(TestCase testCase, List<Cookie> cookieLst) {
	int simNumber = testCase.getSimulationNumber()==null?0:testCase.getSimulationNumber();
	if(testCase.isServerApiAuth() || testCase.isServerApiTarget()) {
		simNumber = -1;
	} else if(testCase.isExternalApi()) {
		simNumber = -2;
	} else if(testCase.getSimulationNumber()==null) {
		simNumber = 0;
	}
	if(cookieLst!=null && cookies.get(simNumber)!=null)
	{
		for (Cookie cookie : cookieLst) {
			if(cookie.getValue()!=null)
			{
				cookies.get(simNumber).put(cookie.getName(), cookie.getValue());
			}
		}
	}
}
 

import com.ning.http.client.cookie.Cookie; //导入依赖的package包/类
/**
 * Get {@link NewCookie} from Ning {@link Cookie}.
 *
 * @param ningCookie Ning {@link Cookie}
 * @return {@link NewCookie}
 */
public static NewCookie getCookie(final Cookie ningCookie) {
    return new NewCookie(
        ningCookie.getName(),
        ningCookie.getValue(),
        ningCookie.getPath(),
        ningCookie.getDomain(),
        "",
        (int) ningCookie.getMaxAge(),
        ningCookie.isSecure(),
        ningCookie.isHttpOnly()
    );
}
 

import com.ning.http.client.cookie.Cookie; //导入依赖的package包/类
private void setCookies(Collection<Cookie> cookies) {
    this.cookies.clear();

    if (cookies != null) {
        this.cookies.addAll(cookies);
    }
}
 

import com.ning.http.client.cookie.Cookie; //导入依赖的package包/类
private AsyncHttpClient.BoundRequestBuilder addContext(AsyncHttpClient.BoundRequestBuilder builder) {
    if (builder != null) {
        if (this.realm != null) {
            builder.setRealm(this.realm);
        }
        for (Cookie cookie : this.cookies) {
            builder.addCookie(cookie);
        }
    }
    return builder;
}
 

import com.ning.http.client.cookie.Cookie; //导入依赖的package包/类
public void setResponse(String endpoint, Response response) throws IOException {

        Integer statusCode = response.getStatusCode();
        String httpResponse = response.getResponseBody();
        List<Cookie> cookies = response.getCookies();
        this.response = new HttpResponse(statusCode, httpResponse, cookies);
    }
 

import com.ning.http.client.cookie.Cookie; //导入依赖的package包/类
public void setDCOSCookie(String DCOSsecret, String email) throws Exception {
    final JWTSigner signer = new JWTSigner(DCOSsecret);
    final HashMap<String, Object> claims = new HashMap();
    claims.put("uid", email);
    final String jwt = signer.sign(claims);
    Cookie cookie = new Cookie("dcos-acs-auth-cookie", jwt, false, "", "", 99999, false, false);
    List<Cookie> cookieList = new ArrayList<Cookie>();
    cookieList.add(cookie);
    commonspec.setCookies(cookieList);
}
 

import com.ning.http.client.cookie.Cookie; //导入依赖的package包/类
/**
 * Generate token to authenticate in gosec SSO
 * @param ssoHost current sso host
 * @param userName username
 * @param passWord password
 * @throws Exception exception
 */
@Given("^I set sso token using host '(.+?)' with user '(.+?)' and password '(.+?)'$")
public void setGoSecSSOCookie(String ssoHost, String userName, String passWord) throws Exception {
    HashMap<String, String> ssoCookies = new GosecSSOUtils(ssoHost, userName, passWord).ssoTokenGenerator();
    String[] tokenList = {"user", "dcos-acs-auth-cookie"};
    List<Cookie> cookiesAtributes = addSsoToken(ssoCookies, tokenList);

    commonspec.setCookies(cookiesAtributes);
}
 

import com.ning.http.client.cookie.Cookie; //导入依赖的package包/类
public List<Cookie> addSsoToken(HashMap<String, String> ssoCookies, String[] tokenList) {
    List<Cookie> cookiesAttributes = new ArrayList<>();

    for (String tokenKey : tokenList) {
        cookiesAttributes.add(new Cookie(tokenKey, ssoCookies.get(tokenKey),
                false, null,
                null, 999999, false, false));
    }
    return cookiesAttributes;
}
 

import com.ning.http.client.cookie.Cookie; //导入依赖的package包/类
/**
 * Get only one cookie, using the cookie name.
 */
@Override
public Cookie getCookie(String name) {
    for (Cookie cookie : ahcResponse.getCookies()) {
        // safe -- cookie.getName() will never return null
        if (cookie.getName().equals(name)) {
            return cookie;
        }
    }
    return null;
}
 

import com.ning.http.client.cookie.Cookie; //导入依赖的package包/类
@Test
public void testOptionalCachingOnJSESSIONIDCookie() throws Exception {
    Map<String,String> filterInitParams = new HashMap<String,String>(1,1.0f) {{
        put(PublishToMemcachedFilter.MEMCACHED_KEY_PARAM,"$scheme$request_method$uri$args?$cookie_jsessionid?");
        put(PublishToMemcachedFilter.MEMCACHED_EXPIRY,"10");
    }};
    server.setupServlet3Filter("localhost:" + memcached.getPort(),null,filterInitParams);
    String url = server.setupServlet("/date/*","date","org.greencheek.web.filter.memcached.servlets.JSESSIONIDServlet",true);

    assertTrue(server.startTomcat());
    url = server.replacePort(url);
    Response response = executeGetRequest(url);
    assertEquals(CacheConfigGlobals.DEFAULT_CACHE_MISS_HEADER_VALUE,getCacheHeader(response));

    List<Cookie> cookies = response.getCookies();
    String sessionId = getSessionID(response);

    response = executeGetRequest(url,cookies);
    assertEquals(CacheConfigGlobals.DEFAULT_CACHE_MISS_HEADER_VALUE,getCacheHeader(response));

    response = executeGetRequest(url,cookies);
    assertEquals(CacheConfigGlobals.DEFAULT_CACHE_HIT_HEADER_VALUE,getCacheHeader(response));

    response = executeGetRequest(url);
    assertEquals(CacheConfigGlobals.DEFAULT_CACHE_HIT_HEADER_VALUE,getCacheHeader(response));
    assertEquals(sessionId, getSessionID(response));

    Cookie c = cookies.get(0);
    Cookie madeupJsessionId = new Cookie("JSESSIONID","xx","xx",c.getDomain(),c.getPath(),
            c.getExpires(),c.getMaxAge(),c.isSecure(),c.isHttpOnly());

    response = executeGetRequest(url, Collections.singletonList(madeupJsessionId));
    assertEquals(CacheConfigGlobals.DEFAULT_CACHE_MISS_HEADER_VALUE,getCacheHeader(response));
    assertNotEquals(sessionId, getSessionID(response));

}
 

import com.ning.http.client.cookie.Cookie; //导入依赖的package包/类
private Response executeGetRequest(String url,List<Cookie> cookies) throws Exception {
    AsyncHttpClient.BoundRequestBuilder rBuilder = server.getHttpClient().prepareGet(url);
    for(Cookie c : cookies) {
        rBuilder.addCookie(c);
    }
    Request r = rBuilder.build();
    return server.getHttpClient().executeRequest(r).get();
}
 

import com.ning.http.client.cookie.Cookie; //导入依赖的package包/类
private String getCookieString(Response response, String key) {
    for(Cookie c : response.getCookies()) {
        if(c.getName().equalsIgnoreCase(key)) {
            return c.getValue();
        }
    }
    return "";
}
 

import com.ning.http.client.cookie.Cookie; //导入依赖的package包/类
/**
 * Buile new {@link ParsecAsyncHttpRequest} instance.
 *
 * @return {@link ParsecAsyncHttpRequest}
 */
@SuppressWarnings("PMD.NPathComplexity")
public ParsecAsyncHttpRequest build() {
    ningRequestBuilder = new RequestBuilder(method)
        .setContentLength(contentLength)
        .setFollowRedirects(followRedirect)
        .setHeaders(headers)
        .setProxyServer(proxyServer)
        .setRangeOffset(rangeOffset)
        .setRequestTimeout(requestTimeout)
        .setVirtualHost(virtualHost)
        .setUri(new Uri(
            uri.getScheme(),
            uri.getUserInfo(),
            uri.getHost(),
            uri.getPort(),
            uri.getPath(),
            uri.getRawQuery()
    ));

    if (body != null) {
        ningRequestBuilder.setBody(body);
    } else if (byteBody != null) {
        ningRequestBuilder.setBody(byteBody);
    }

    if (bodyEncoding != null) {
        ningRequestBuilder.setBodyEncoding(bodyEncoding);
    }

    if (cookies != null && !cookies.isEmpty()) {
        for (NewCookie cookie : cookies) {
            ningRequestBuilder.addCookie(new Cookie(
                cookie.getName(),
                cookie.getValue(),
                false,
                cookie.getDomain(),
                cookie.getPath(),
                cookie.getMaxAge(),
                cookie.isSecure(),
                cookie.isHttpOnly()
            ));
        }
    }

    if (formParams != null) {
        ningRequestBuilder.setFormParams(formParams);
    }

    if (queryParams != null) {
        ningRequestBuilder.setQueryParams(queryParams);
    }

    if (acceptCompression && !headers.containsKey(ACCEPT_ENCODING_HEADER)) {
        ningRequestBuilder.addHeader(ACCEPT_ENCODING_HEADER, COMPRESSION_TYPE);
    }

    for (Part part: bodyParts) {
        ningRequestBuilder.addBodyPart(part);
    }

    return new ParsecAsyncHttpRequest(this);
}
 

import com.ning.http.client.cookie.Cookie; //导入依赖的package包/类
@Test
public void testGetCookies() throws Exception {
    List<Cookie> ningCookies = new ArrayList<>();

    ningCookies.add(Cookie.newValidCookie(
        "cookie1_name",
        "cookie1_value",
        false,
        null,
        "cookie1_path",
        1,
        true,
        true
    ));

    ningCookies.add(Cookie.newValidCookie(
        "cookie2_name",
        "cookie2_value",
        false,
        null,
        "cookie2_path",
        2,
        false,
        false
    ));

    List<NewCookie> cookies = ParsecHttpUtil.getCookies(ningCookies);
    assertEquals(2, cookies.size());

    NewCookie cookie1 = cookies.get(0);
    assertEquals("cookie1_name", cookie1.getName());
    assertEquals("cookie1_value", cookie1.getValue());
    assertEquals(null, cookie1.getDomain());
    assertEquals("cookie1_path", cookie1.getPath());
    assertEquals(null, cookie1.getExpiry());
    assertEquals(1, cookie1.getMaxAge());
    assertTrue(cookie1.isSecure());
    assertTrue(cookie1.isHttpOnly());

    NewCookie cookie2 = cookies.get(1);
    assertEquals("cookie2_name", cookie2.getName());
    assertEquals("cookie2_value", cookie2.getValue());
    assertEquals(null, cookie2.getDomain());
    assertEquals("cookie2_path", cookie2.getPath());
    assertEquals(null, cookie2.getExpiry());
    assertEquals(2, cookie2.getMaxAge());
    assertFalse(cookie2.isSecure());
    assertFalse(cookie2.isHttpOnly());
}
 

import com.ning.http.client.cookie.Cookie; //导入依赖的package包/类
@Test
public void testGet() throws Exception {
    FluentCaseInsensitiveStringsMap responseHeaders = new FluentCaseInsensitiveStringsMap();

    responseHeaders.add("header1", "header1_value1");
    responseHeaders.add("header2", "header2_value1");

    List<Cookie> ningCookies = new ArrayList<>();

    ningCookies.add(new Cookie(
        "cookie1_name",
        "cookie1_value",
        false,
        null,
        "cookie1_path",
        1,
        true,
        true
    ));

    ningCookies.add(new Cookie(
        "cookie2_name",
        "cookie2_value",
        false,
        null,
        "cookie2_path",
        2,
        false,
        false
    ));

    when(mockNingResponse.getHeaders()).thenReturn(responseHeaders);
    when(mockNingResponse.getContentType()).thenReturn(MediaType.APPLICATION_JSON);
    when(mockNingResponse.getCookies()).thenReturn(ningCookies);
    when(mockNingResponse.getStatusCode()).thenReturn(200);
    when(mockNingResponse.hasResponseHeaders()).thenReturn(true);

    when(mockNingFuture.get()).thenReturn(mockNingResponse);
    when(mockNingFuture.get(anyLong(), any(TimeUnit.class))).thenReturn(mockNingResponse);

    ParsecCompletableFuture<Response> future = new ParsecCompletableFuture<>(mockNingFuture);
    Response response = future.get();

    assertEquals(response.getContentType(), MediaType.APPLICATION_JSON);
    assertEquals(response.getHeaders().size(), 2);

    response = future.get(3, TimeUnit.SECONDS);

    assertEquals(response.getContentType(), MediaType.APPLICATION_JSON);
    assertEquals(response.getHeaders().size(), 2);
}
 

import com.ning.http.client.cookie.Cookie; //导入依赖的package包/类
public List<Cookie> getCookies() {
    return cookies;
}
 

import com.ning.http.client.cookie.Cookie; //导入依赖的package包/类
public void setCookies(List<Cookie> cookies) {
    this.cookies = cookies;
}