jhc.redsniff.webdriver.SeleniumController.java Source code

Introduction

Here is the source code for jhc.redsniff.webdriver.SeleniumController.java

Source

/*******************************************************************************
 * Copyright 2014 JHC Systems Limited
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *******************************************************************************/
package jhc.redsniff.webdriver;

import java.lang.reflect.Field;

import jhc.redsniff.action.Controller;

import org.openqa.selenium.Keys;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.htmlunit.HtmlUnitWebElement;
import org.openqa.selenium.interactions.Actions;
import org.openqa.selenium.remote.RemoteWebDriver;

import com.gargoylesoftware.htmlunit.html.HtmlElement;

public class SeleniumController implements Controller<WebElement> {

    private final WebDriver driver;

    public SeleniumController(WebDriver driver) {
        this.driver = driver;
    }

    @Override
    public void quit() {
        driver.quit();
    }

    @Override
    public void goTo(String url) {
        driver.get(url);
    }

    @Override
    public void click(WebElement element) {
        element.click();
    }

    @Override
    public void submit(WebElement element) {
        element.submit();
    }

    @Override
    public void type(String input, WebElement element) {
        element.sendKeys(input);
    }

    @Override
    public void clear(WebElement element) {
        element.clear();
    }

    @Override
    public boolean enabled(WebElement element) {
        return element.isEnabled();
    }

    public WebDriver getDriver() {
        return driver;
    }

    @Override
    public void tab(WebElement element) {
        //XXX shame we have to do it like this..
        if (element instanceof HtmlUnitWebElement)
            tabOnHtmlUnitElement(element);
        else {
            blurUsingJavascript(element);
            //blurByTypingTabKey(element);
        }
    }

    protected void blurByTypingTabKey(WebElement element) {
        new Actions(driver).click(element).sendKeys(Keys.TAB).perform();
    }

    private void blurUsingJavascript(WebElement element) {
        String id = element.getAttribute("id");
        String script = "document.getElementById('" + id + "').blur()";
        ((RemoteWebDriver) driver).executeScript(script);
    }

    private void tabOnHtmlUnitElement(WebElement element) {
        getHtmlUnitElement((HtmlUnitWebElement) element).blur();
    }

    //XXX
    //gets private field
    private HtmlElement getHtmlUnitElement(HtmlUnitWebElement webElement) {
        try {
            boolean changed = false;
            Field field = HtmlUnitWebElement.class.getDeclaredField("element");
            if (!field.isAccessible()) {

                field.setAccessible(true);
                changed = true;
            }

            HtmlElement htmlElement = (HtmlElement) field.get(webElement);
            if (changed)
                field.setAccessible(false);
            return htmlElement;
        } catch (Exception e) {
            throw new Error(e.getMessage(), e);
        }
    }
}