Java Selenium封装--RemoteWebDriver

 package com.selenium.driver;
 import java.io.File;
 import java.io.IOException;
 import java.net.URL;
 import java.util.HashMap;
 import java.util.Map;
 import java.util.Set;
 import java.util.regex.Matcher;
 import java.util.regex.Pattern;
 import org.apache.commons.io.FileUtils;
 import org.openqa.selenium.Alert;
 import org.openqa.selenium.Capabilities;
 import org.openqa.selenium.Cookie;
 import org.openqa.selenium.JavascriptExecutor;
 import org.openqa.selenium.NoSuchElementException;
 import org.openqa.selenium.OutputType;
 import org.openqa.selenium.TakesScreenshot;
 import org.openqa.selenium.WebDriver;
 import org.openqa.selenium.WebElement;
 import org.openqa.selenium.remote.Augmenter;
 import org.openqa.selenium.remote.RemoteWebDriver;
 import org.openqa.selenium.remote.RemoteWebElement;
 public class JSWebDriver{
     private RemoteWebDriver wd = null;
     private JavascriptExecutor jse = null;

     public JSWebDriver(URL remoteAddress, Capabilities desiredCapabilities) {
         wd = new RemoteWebDriver(remoteAddress, desiredCapabilities);
     }

     ///
     ///浏览器url导航
     ///
     public void goTo(String url){
         wd.get(url);
     }    

     ///
     ///浏览器退出
     ///
     public void quit(){
         wd.quit();
     }

     ///
     ///浏览器后退
     ///
     public void back(){
         wd.navigate().back();
     }

     ///
     ///浏览器前进
     ///
     public void forward(){
         wd.navigate().forward();
     }

     ///
     ///浏览器刷新
     ///
     public void refresh(){
         wd.navigate().refresh();
     }

     ///
     ///切换到新浏览器窗口；按照title、url、index；支持正则匹配
     ///
     public void switchToWindow(String by, String value, String...match) throws Exception{
         String currenthandle = wd.getWindowHandle();
         Set<String> handles = wd.getWindowHandles();
         int currentIndex = -1;
         String searchString = "";
         for(String handle : handles){
             currentIndex += 1;
             if(handle.equals(currenthandle)){
                 continue;
             }else{
                 wd.switchTo().window(handle);
                 if (match.length == 1 && match[0].equals("regex")){
                     if (by.equals("title")){
                         searchString = wd.getTitle();
                     }else if (by.equals("url")){
                         searchString = wd.getCurrentUrl();
                     }
                     Pattern pattern = Pattern.compile(value);
                     Matcher matcher = pattern.matcher(searchString);
                     if(matcher.find()){
                         return;
                     }
                 }else{
                     if (by.equals("title")){
                         searchString = wd.getTitle();
                     }else if (by.equals("url")){
                         searchString = wd.getCurrentUrl();
                     }else if (by.equals("index")){
                         searchString = Integer.toString(currentIndex);
                     }
                     if(searchString.equals(value)){
                         return;
                     }
                 }
             }
         }
         Exception e = new Exception("Swtich Window Failed, Please Make Sure The Locator Was Right.");
         throw e;
     }

     ///
     ///JS弹框确认
     ///
     public void clickAlertSure(){
         Alert alert = wd.switchTo().alert();
         alert.accept();
     }

     ///
     ///JS弹框取消
     ///
     public void clickAlertDismiss()
     {
         Alert alert = wd.switchTo().alert();
         alert.dismiss();
     }

     ///
     ///设置prompt弹框内容
     ///
     public void setPromptMessage(String parameter){
         Alert alert = wd.switchTo().alert();
         alert.sendKeys(parameter);
     }

     ///
     ///获取JS弹框内容
     ///
     public String getPromptMessage(){
         Alert alert = wd.switchTo().alert();
         return alert.getText();
     }    

     ///
     ///切换到Frame窗口；先定位到iframe元素
     ///
     public void switchToFrame(JSWebElement jselement){
         wd.switchTo().frame(jselement.getNativeWebElement());
     }

     ///
     ///执行JS脚本
     ///
     public void executeScript(String parameter){
         JavascriptExecutor js = getJSE();
         js.executeScript(parameter);
     }

     ///
     ///获取指定cookie
     ///
     public String getCookie(String name){
         Cookie cookie=wd.manage().getCookieNamed(name);
         if (cookie == null){ return "null"; }
         return cookie.getValue();
     }

     ///
     ///获取所有cookie
     ///
     public Map<String, String> getCookies(){
         Map<String, String> newCookies = new HashMap<String, String>();
         Set<Cookie> cookies= wd.manage().getCookies();
         for (Cookie cookie : cookies){
             newCookies.put(cookie.getName(), cookie.getValue());
         }
         return newCookies;
     }

     ///
     ///截取屏幕
     ///
     public void getScreen(String filepath){
         WebDriver augmentedDriver = new Augmenter().augment(this.wd);
         TakesScreenshot ts = (TakesScreenshot) augmentedDriver;
         File screenShotFile = ts.getScreenshotAs(OutputType.FILE);
         try {
             FileUtils.copyFile (screenShotFile, new File(filepath));
         }catch (IOException e){
             e.printStackTrace();
         }
     }

     ///
     ///获取title
     ///
     public String getTitle(){
         return wd.getTitle();
     }    

     ///
     ///获取url
     ///
     public String getUrl(){
         return wd.getCurrentUrl();
     }

     ///
     ///获取HTML源码
     ///
     public String getSource(){
         try {
             Thread.sleep(500);
         } catch (InterruptedException e) {
             e.printStackTrace();
         }
         return wd.getPageSource();
     }

     ///
     ///滚动页面到指定位置
     ///
     public void scroll(String x, String y){
         if (x.equals("left")){
             x = "0";
         }else if (x.equals("right")){
             x = "document.body.scrollWidth";
         }else if (x.equals("middle")){
             x = "document.body.scrollWidth/2";
         }
         if (y.equals("top")){
             y = "0";
         }else if (y.equals("buttom")){
             y = "document.body.scrollHeight";
         }else if (y.equals("middle")){
             y = "document.body.scrollHeight/2";
         }
         this.executeScript(String.format("scroll(%s,%s);", x, y));
     }

     ///
     ///最大化浏览器
     ///
     public void maximize(){
         wd.manage().window().maximize();
     }

     public JSWebElement findElementById(String using) {
         try {
             return new JSWebElement((RemoteWebElement)wd.findElementById(using));
         }catch (NoSuchElementException e){
             return new JSWebElement();
         }
     }

     public JSWebElement findElementByCssSelector(String using) {
         try {
             return new JSWebElement((RemoteWebElement)wd.findElementByCssSelector(using));
         }catch (NoSuchElementException e){
             return new JSWebElement();
         }
     }

     public JSWebElement findElementByXPath(String using) {
         try {
             return new JSWebElement((RemoteWebElement)wd.findElementByXPath(using));
         }catch (NoSuchElementException e){
             return new JSWebElement();
         }
     }

     public JSWebElement findElementByLinkText(String using) {
         try {
             return new JSWebElement((RemoteWebElement)wd.findElementByLinkText(using));
         }catch (NoSuchElementException e){
             return new JSWebElement();
         }
     }

     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();
         }
     }

     ///
     ///获取原生的RemoteWebdriver对象
     ///
     public RemoteWebDriver getNativeWebDriver(){
         return this.wd;
     }

     private JavascriptExecutor getJSE(){
         if (this.jse == null){
             this.jse = (JavascriptExecutor) this.wd;
         }
         return jse;
     }
 }