Java Code Examples for com.gargoylesoftware.htmlunit.WebClient#setRefreshHandler()

The following examples show how to use com.gargoylesoftware.htmlunit.WebClient#setRefreshHandler() . You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.
Example 1
Source File: HtmlPageTest.java    From htmlunit with Apache License 2.0 6 votes vote down vote up
/**
 * Test auto-refresh from a meta tag with no URL.
 * @throws Exception if the test fails
 */
@Test
public void refresh_MetaTag_NoUrl() throws Exception {
    final String firstContent = "<html><head><title>first</title>\n"
        + "<META HTTP-EQUIV=\"Refresh\" CONTENT=\"1\">\n"
        + "</head><body></body></html>";

    final WebClient client = getWebClient();
    final List<Object> collectedItems = new ArrayList<>();
    client.setRefreshHandler(new LoggingRefreshHandler(collectedItems));

    final MockWebConnection webConnection = new MockWebConnection();
    webConnection.setResponse(URL_FIRST, firstContent);
    client.setWebConnection(webConnection);

    client.getPage(URL_FIRST);

    // avoid using equals() on URL because it takes to much time (due to ip resolution)
    assertEquals("first", collectedItems.get(0));
    assertEquals(URL_FIRST, (URL) collectedItems.get(1));
    assertEquals(Integer.valueOf(1), collectedItems.get(2));
}
 
Example 2
Source File: HtmlPageTest.java    From htmlunit with Apache License 2.0 6 votes vote down vote up
/**
 * Test auto-refresh from a meta tag with a refresh handler that doesn't refresh.
 * @throws Exception if the test fails
 */
@Test
public void refresh_MetaTag_CustomRefreshHandler() throws Exception {
    final String firstContent = "<html><head><title>first</title>\n"
        + "<META HTTP-EQUIV=\"Refresh\" CONTENT=\"3;URL=" + URL_SECOND + "\">\n"
        + "</head><body></body></html>";
    final String secondContent = "<html><head><title>second</title></head><body></body></html>";

    final WebClient client = getWebClient();
    final List<Object> collectedItems = new ArrayList<>();
    client.setRefreshHandler(new LoggingRefreshHandler(collectedItems));

    final MockWebConnection webConnection = new MockWebConnection();
    webConnection.setResponse(URL_FIRST, firstContent);
    webConnection.setResponse(URL_SECOND, secondContent);
    client.setWebConnection(webConnection);

    final HtmlPage page = client.getPage(URL_FIRST);

    assertEquals("first", page.getTitleText());

    // avoid using equals() on URL because it takes to much time (due to ip resolution)
    assertEquals("first", collectedItems.get(0));
    assertEquals(URL_SECOND, (URL) collectedItems.get(1));
    assertEquals(Integer.valueOf(3), collectedItems.get(2));
}
 
Example 3
Source File: HiddenBrowserDriver.java    From ats-framework with Apache License 2.0 4 votes vote down vote up
/**
 * 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);
        }
    }
}