Java Code Examples for com.gargoylesoftware.htmlunit.util.Cookie#getExpires()

private void storeCookies(WebRequest webRequest, javax.servlet.http.Cookie[] cookies) {
	Date now = new Date();
	CookieManager cookieManager = this.webClient.getCookieManager();
	for (javax.servlet.http.Cookie cookie : cookies) {
		if (cookie.getDomain() == null) {
			cookie.setDomain(webRequest.getUrl().getHost());
		}
		Cookie toManage = createCookie(cookie);
		Date expires = toManage.getExpires();
		if (expires == null || expires.after(now)) {
			cookieManager.addCookie(toManage);
		}
		else {
			cookieManager.removeCookie(toManage);
		}
	}
}
 

private void storeCookies(WebRequest webRequest, javax.servlet.http.Cookie[] cookies) {
	Date now = new Date();
	CookieManager cookieManager = this.webClient.getCookieManager();
	for (javax.servlet.http.Cookie cookie : cookies) {
		if (cookie.getDomain() == null) {
			cookie.setDomain(webRequest.getUrl().getHost());
		}
		Cookie toManage = createCookie(cookie);
		Date expires = toManage.getExpires();
		if (expires == null || expires.after(now)) {
			cookieManager.addCookie(toManage);
		}
		else {
			cookieManager.removeCookie(toManage);
		}
	}
}
 

/**
 * Clears all cookies that have expired before supplied date.
 * If disabled, this returns false.
 * @param date the date to use for comparison when clearing expired cookies
 * @return whether any cookies were found expired, and were cleared
 */
public synchronized boolean clearExpired(final Date date) {
    if (!isCookiesEnabled()) {
        return false;
    }

    if (date == null) {
        return false;
    }

    boolean foundExpired = false;
    for (final Iterator<Cookie> iter = cookies_.iterator(); iter.hasNext();) {
        final Cookie cookie = iter.next();
        if (cookie.getExpires() != null && date.after(cookie.getExpires())) {
            iter.remove();
            foundExpired = true;
        }
    }
    return foundExpired;
}
 

/**
 * Clears all cookies that have expired before supplied date.
 * If disabled, this returns false.
 * @param date the date to use for comparison when clearing expired cookies
 * @return whether any cookies were found expired, and were cleared
 */
public synchronized boolean clearExpired(final Date date) {
    if (!isCookiesEnabled()) {
        return false;
    }

    if (date == null) {
        return false;
    }

    boolean foundExpired = false;
    for (final Iterator<Cookie> iter = cookies_.iterator(); iter.hasNext();) {
        final Cookie cookie = iter.next();
        if (cookie.getExpires() != null && date.after(cookie.getExpires())) {
            iter.remove();
            foundExpired = true;
        }
    }
    return foundExpired;
}
 

/**
 * Adds the specified cookie.
 * If disabled, this does nothing.
 * @param cookie the cookie to add
 */
public synchronized void addCookie(final Cookie cookie) {
    if (!isCookiesEnabled()) {
        return;
    }

    cookies_.remove(cookie);

    // don't add expired cookie
    if (cookie.getExpires() == null || cookie.getExpires().after(new Date())) {
        cookies_.add(cookie);
    }
}
 

/**
 * Adds the specified cookie.
 * If disabled, this does nothing.
 * @param cookie the cookie to add
 */
public synchronized void addCookie(final Cookie cookie) {
    if (!isCookiesEnabled()) {
        return;
    }

    cookies_.remove(cookie);

    // don't add expired cookie
    if (cookie.getExpires() == null || cookie.getExpires().after(new Date())) {
        cookies_.add(cookie);
    }
}
 

/**
 * Fixing refresh handler to skip Refresh meta tags
 * Allowing connections to any host, regardless of whether they have valid certificates or not
 * Fixing JSESSIONID cookie value
 * Some applications expect double quotes in the beginning and at the end of the JSESSIONID cookie value
 */
private void fixHtmlUnitBehaviour() {

    Field webClientField = null;
    boolean fieldAccessibleState = false;
    try {
        TargetLocator targetLocator = webDriver.switchTo();
        webClientField = targetLocator.getClass().getDeclaringClass().getDeclaredField("webClient");
        fieldAccessibleState = webClientField.isAccessible();
        webClientField.setAccessible(true);
        final WebClient webClient = (WebClient) webClientField.get(targetLocator.defaultContent());

        // Allowing connections to any host, regardless of whether they have valid certificates or not
        webClient.getOptions().setUseInsecureSSL(true);

        // Set Http connection timeout (in milliseconds). The default value is 90 seconds, because in Firefox >= 16
        // the "network.http.connection-timeout" property is 90. But this value is not enough for some cases.
        // NOTE: use 0 for infinite timeout
        webClient.getOptions().setTimeout(5 * 60 * 1000);

        webClient.getOptions().setRedirectEnabled(true);
        webClient.getOptions().setJavaScriptEnabled(true);
        webClient.getOptions().setThrowExceptionOnScriptError(true);
        webClient.getOptions().setPrintContentOnFailingStatusCode(true);

        // Hide CSS Warnings
        webClient.setCssErrorHandler(new SilentCssErrorHandler());

        // Suppress warnings like: "Expected content type ... but got ..."
        webClient.setIncorrectnessListener(new IncorrectnessListener() {

            //                private final Log log = LogFactory.getLog( this.getClass() );

            @Override
            public void notify( final String message, final Object origin ) {

                //                    log.warn( message );
            }
        });

        if (!Boolean.parseBoolean(System.getProperty(ALLOW_META_REFRESH_TAG))) {

            /*
             * Fix for refresh meta tags eg. "<meta http-equiv="refresh" content="300">"
             * The default refresh handler is with Thread.sleep(refreshSecondsFromMetaTag) in the main thread!!!
                 *
                 * Maybe we should check and test this handler: webClient.setRefreshHandler( new ThreadedRefreshHandler() );
             */
            webClient.setRefreshHandler(new RefreshHandler() {

                @Override
                public void handleRefresh( Page page, URL url, int seconds ) throws IOException {

                }
            });
        }

        /*
         * Fix JSessionId
         */

        // WebConnectionWrapper constructs a WebConnection object wrapping the connection of the WebClient
        // and places itself (in the constructor) as connection of the WebClient.
        new WebConnectionWrapper(webClient) {

            public WebResponse getResponse( WebRequest request ) throws IOException {

                Cookie jsCookie = webClient.getCookieManager().getCookie("JSESSIONID");
                if (jsCookie != null && (!jsCookie.getValue().startsWith("\"")
                                         && !jsCookie.getValue().endsWith("\""))) {

                    Cookie newCookie = new Cookie(jsCookie.getDomain(), jsCookie.getName(),
                                                  "\"" + jsCookie.getValue() + "\"", jsCookie.getPath(),
                                                  jsCookie.getExpires(), jsCookie.isSecure());

                    webClient.getCookieManager().removeCookie(jsCookie);
                    webClient.getCookieManager().addCookie(newCookie);
                }
                return super.getResponse(request);
            }
        };
    } catch (Exception e) {

        throw new SeleniumOperationException("Error retrieving internal Selenium web client", e);
    } finally {

        if (webClientField != null) {
            webClientField.setAccessible(fieldAccessibleState);
        }
    }
}