Java Selenium封装--RemoteWebElement

package com.liuke.selenium.driver;

import java.sql.SQLException;
import java.util.List;
import org.json.JSONException;
import org.openqa.selenium.By;
import org.openqa.selenium.Dimension;
import org.openqa.selenium.JavascriptExecutor;
import org.openqa.selenium.NoSuchElementException;
import org.openqa.selenium.Point;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.interactions.internal.Coordinates;
import org.openqa.selenium.remote.RemoteWebElement;
import org.openqa.selenium.support.ui.Select;

public class JSWebElement {
    private RemoteWebElement we = null;
    private JavascriptExecutor jse = null;
    
    public JSWebElement(){}
    
    public JSWebElement(RemoteWebElement we){
        this.we = we;
    }
    
    ///
    ///通过元素ID定位元素
    ///
    public JSWebElement findElementById(String using) {
        try {
            return new JSWebElement((RemoteWebElement)we.findElementById(using));
        }catch (NoSuchElementException e){
            return new JSWebElement();
        }
    }
    
    ///
    ///通过元素CSS表达式定位元素
    ///
    public JSWebElement findElementByCssSelector(String using) {
        try {
            return new JSWebElement((RemoteWebElement)we.findElementByCssSelector(using));
        }catch (NoSuchElementException e){
            return new JSWebElement();
        }
    }

    ///
    ///通过元素Xpath表达式定位元素
    ///    
    public JSWebElement findElementByXPath(String using) {
        try {
            return new JSWebElement((RemoteWebElement)we.findElementByXPath(using));
        }catch (NoSuchElementException e){
            return new JSWebElement();
        }
    }

    ///
    ///通过链接的文字定位元素
    ///    
    public JSWebElement findElementByLinkText(String using) {
        try {
            return new JSWebElement((RemoteWebElement)we.findElementByLinkText(using));
        }catch (NoSuchElementException e){
            return new JSWebElement();
        }
    }

    ///
    ///通过元素DOM表达式定位元素
    ///    
    public JSWebElement findElementByDom(String using) {
        try {
            JavascriptExecutor js = this.getJSE();
            WebElement we = (WebElement)js.executeScript(String.format("return %s", using));            
            return new JSWebElement((RemoteWebElement)we);
        }catch (NoSuchElementException e){
            return new JSWebElement();
        }
    }

    ///
    ///判断元素是否存在
    ///
    public Boolean isExist(){
        if (we != null){
            return true;
        }else{
            return false;
        }
    }

    ///
    ///获取元素的HTML内容
    ///
    public String getHtml(){
        return we.getAttribute("outerHTML");
    }

    ///
    ///获取元素的文本内容
    ///    
    public String getText(){
        return we.getText();
    }

    ///
    ///获取元素的value值
    ///    
    public String getValue(){
        return this.getAttribute("value");
    } 
    
    ///
    ///获取元素的特定属性值
    ///
    public String getAttribute(String name){
        return we.getAttribute(name);
    } 
        
    ///
    ///向可输入元素发送内容，如：text、textarea、filefield等
    ///
    public void sendKeys(String string){
        String old_bg = this.setBackground("yellow");
        try {
            Thread.sleep(800);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        we.sendKeys(string);
        this.setBackground(old_bg);
    }
    
    ///
    ///判断元素是否可用
    ///
    public boolean isEnabled(){
        return we.isEnabled();
    }
    
    ///
    ///判断元素是否可见
    ///
    public boolean isVisible(){
        return we.isDisplayed();
    }
    
    ///
    ///清空可编辑元素的内容。不可编辑元素次操作会抛异常
    ///
    public void clear(){
        we.clear();
    }
    
    ///
    ///对元素进行点击操作
    ///
    public void click(){
        we.click();
    }
    
    ///
    ///检查元素的特定属性值
    ///
    public void checkAttr(String attribute, JSWebUtils utils) throws SQLException, JSONException
    {
        String [] attributes=attribute.split("=", 2);
        String actual = this.we.getAttribute(attributes[0]);
        if (actual == null){ actual = "null"; }
        utils.checkPointBase(actual,attributes[1]);
    }
    
    ///
    ///获取元素的CSS值
    ///
    public String getCssValue(String name)
    {
        return we.getCssValue(name);
    }
    
    ///
    ///判断元素是否被选中
    ///
    public boolean isSelected()
    {
        return we.isSelected();
    }    
    
    ///
    ///可选元素进行选中操作；如：select
    ///
    public void select(String by, String value) throws Exception
    {
        if (we.getTagName().equals("select")){
            Select select = new Select(we);
            if (by.equals("index")){
                select.selectByIndex(Integer.parseInt(value));
            }else if (by.equals("value")){
                select.selectByValue(value);
            }else if (by.equals("text")){
                select.selectByVisibleText(value);                
            }
        }else{
            Exception e = new Exception("The element is not SELECT Object");
            throw e;
        }
    }
    
    ///
    ///对可选中元素进行取消选择操作；如:SELECT in multiple type
    ///
    public void deSelect(String by, String...value) throws Exception
    {
        if (we.getTagName().equals("select")){
            Select select = new Select(we);
            if (by.equals("index")){
                select.deselectByIndex(Integer.parseInt(value[0]));
            }else if (by.equals("value")){
                select.deselectByValue(value[0]);
            }else if (by.equals("text")){
                select.deselectByVisibleText(value[0]);
            }else if (by.equals("*")){
                select.deselectAll();
            }
        }else{
            Exception e = new Exception("The element is not SELECT Object");
            throw e;
        }
    }
    
    ///
    ///判断下拉框是否为多选
    ///
    public boolean isMultiple() throws Exception
    {
        if (we.getTagName().equals("select")){
            Select select = new Select(we);
            if (select.isMultiple()){
                return true;
            }else{
                return false;
            }
        }else{
            Exception e = new Exception("The element is not SELECT Object");
            throw e;
        }
    }    
    
    ///
    ///获取select的当前选中值
    ///
    public String getSelectedText() throws Exception
    {
        if (we.getTagName().equals("select")){
            String text = "";
            Select select = new Select(we);
            List<WebElement> options = select.getAllSelectedOptions();
            for (WebElement w : options){
                text += w.getText() + "
";
            }
            return text;
        }else{
            Exception e = new Exception("The element is not SELECT Object");
            throw e;
        }
    }
    
    ///
    ///判断指定项是否存在
    ///
    public boolean isInclude(String name) throws Exception
    {
        if (we.getTagName().equals("select")){
            Select select = new Select(we);
            List<WebElement> options = select.getOptions();
            for (WebElement w : options){
                if (w.getText().equals(name)){
                    return true;
                }
            }
            return false;
        }else{
            Exception e = new Exception("The element is not SELECT Object");
            throw e;
        }
    }
    
    ///
    ///获取元素的tagname
    ///
    public String getTagName(){
        return we.getTagName();
    }
    
    ///
    ///获取元素的id
    ///
    public String getId(){
        return we.getId();
    }
    
    ///
    ///获取元素的绝对位置
    ///
    public Point getLocation(){
        return we.getLocation();
    }
    
    ///
    ///获取元素的出现在屏幕可见区时的位置
    ///
    public Point getLocationOnScreenOnceScrolledIntoView(){
        return we.getLocationOnScreenOnceScrolledIntoView();
    }
    
    ///
    ///获取元素的坐标
    ///    
    public Coordinates getCoordinates(){
        return we.getCoordinates();
    }
    
    ///
    ///获取元素的大小
    ///
    public Dimension getSize(){
        return we.getSize();
    }

    ///
    ///提交元素所在form的内容
    ///    
    public void submit()
    {
        we.submit();
    }
    
    ///
    ///勾选radio、checkbox
    ///
    public void check(String...values) throws Exception
    {
        if (we.getTagName().equals("input")){
            if (we.getAttribute("type").equals("radio")){
                WebDriver wd = we.getWrappedDriver();
                List<WebElement> wl = wd.findElements(By.name(we.getAttribute("name")));
                if (values[0].equals("index")){
                    wl.get(Integer.parseInt(values[1])).click();
                }else if (values[0].equals("value")){
                    for (WebElement w : wl){
                        if (w.getAttribute("value").equals(values[1])){
                            w.click();
                            break;
                        }
                    }
                }
            }else if (we.getAttribute("type").equals("checkbox")){
                if (!we.isSelected()){
                    we.click();
                }    
            }else{
                Exception e = new Exception("The element is not Radio or CheckBox Object");
                throw e;                
            }
        }else{
            Exception e = new Exception("The element is not INPUT Object");
            throw e;
        }
    }
    
    ///
    ///取消勾选checkbox
    ///
    public void unCheck() throws Exception
    {
        if (we.getTagName().equals("input") && we.getAttribute("type").equals("checkbox")){
                if (we.isSelected()){
                    we.click();
                }                
        }else{
            Exception e = new Exception("The element is not CheckBox Object");
            throw e;
        }
    }
    
    ///
    ///checkbox、radio是否勾选
    ///
    public boolean isChecked(String...values) throws Exception
    {
        if (we.getTagName().equals("input")){
            if (we.getAttribute("type").equals("radio")){
                WebDriver wd = we.getWrappedDriver();
                List<WebElement> wl = wd.findElements(By.name(we.getAttribute("name")));
                if (values[0].equals("index")){
                    return wl.get(Integer.parseInt(values[1])).isSelected();
                }else if (values[0].equals("value")){
                    for (WebElement w : wl){
                        if (w.getAttribute("value").equals(values[1])){
                            return w.isSelected();
                        }
                    }
                }
                return false;
            }else if (we.getAttribute("type").equals("checkbox")){
                return we.isSelected();
            }else{
                Exception e = new Exception("The element is not Radio or CheckBox Object");
                throw e;                
            }
        }else{
            Exception e = new Exception("The element is not INPUT Object");
            throw e;
        }        
    }

    ///
    ///把元素滚动到可视区
    ///
    public void scroll()
    {
        this.focus();
    }
    
    ///
    ///高亮元素
    ///
    public void highLight() throws InterruptedException
    {
        this.focus();
        JavascriptExecutor js = getJSE();
        String old_style = we.getAttribute("style");
        for (int i = 0; i < 3; i++) {              
            js.executeScript("arguments[0].setAttribute('style', arguments[1]);", this.we, "background-color: red; border: 2px solid red;" + old_style); 
            Thread.sleep(500);
            js.executeScript("arguments[0].setAttribute('style', arguments[1]);", this.we, old_style); 
            Thread.sleep(500);
        }
    }

    ///
    ///触发元素的特定事件
    ///
    public void fireEvent(String event){
        JavascriptExecutor js = getJSE();
        js.executeScript(String.format("arguments[0].%s()", event), this.we);
    }
    
    ///
    ///使元素获取焦点
    ///
    public void focus(){
//        this.we.sendKeys("");
        JavascriptExecutor js = getJSE();
        js.executeScript("arguments[0].focus();", this.we);
    }    
    
    ///
    ///对元素执行JavaScript操作；即执行元素的dom操作
    ///
    public void executeJS(String commands){
        JavascriptExecutor js = getJSE();    
        String[] comandArr = commands.split(";");
        commands = "";
        for (String comand : comandArr){
            if (!comand.trim().equals("")){
                commands += String.format("arguments[0].%s;", comand);
            }
        }
        if (!commands.equals("")){
            js.executeScript(commands, this.we);
        }
    }
    
    ///
    ///获取原始的RemoteWebElement对象
    ///
    public RemoteWebElement getNativeWebElement(){
        return this.we;
    }
        
    private JavascriptExecutor getJSE(){
        if (this.isExist()){
            if (this.jse == null){
                WebDriver wd = we.getWrappedDriver();
                this.jse = (JavascriptExecutor) wd;                
            }        
        }
        return jse;
    }
    
    private String setBackground(String color){
        JavascriptExecutor js = getJSE();
        String old_bg = we.getCssValue("background-color");
        js.executeScript("arguments[0].style.background = arguments[1];", this.we, color); 
        return old_bg;
    }        
    
}