Selenium 2 & WebDriver &多線程 並發

 

     我用的是Selenium2，至於它的背景和歷史就不贅述了。Selenium2也叫WebDriver。下面講個例子，用WebDriver+java來寫個自動化測試的程序。（如果能用firefox去測試的話，我就直接用Selenium IDE錄腳本了。。。）有個前提：就是我們只用IE瀏覽器（WebDriver支持多種瀏覽器）。需求：打開網易163郵箱，並且登錄，然后打開收件箱；以上哪一步異常就截屏。這是我的程序要做的自動化測試的功能。

     首先需要去官網下載WebDriver和IE的驅動IEDriverServer.exe（根據你的系統，64位系統必須用64位的驅動，32位運行不了；32位的系統同理）

     官網下載地址：http://docs.seleniumhq.org/download/

     下面是下載位置的截圖：

    

      由於我用到了多線程，所以我將代碼放在了一個線程里面，多線程的事情后面再說，先看打開一個瀏覽器完成需求功能的代碼。（單元測試工具我這里用的是testNG）

import java.io.File;
import java.net.URL;
import java.text.SimpleDateFormat;
import java.util.Date;

import org.openqa.selenium.By;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.Platform;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.ie.InternetExplorerDriver;
import org.openqa.selenium.remote.DesiredCapabilities;
import org.openqa.selenium.remote.RemoteWebDriver;
import org.openqa.selenium.support.ui.ExpectedCondition;
import org.openqa.selenium.support.ui.WebDriverWait;
import org.testng.annotations.Test;



public class TestLogin_testNG implements Runnable{
    public static final SimpleDateFormat sf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss.SSSS");

    @Test
    public void run() {
        System.out.println(Thread.currentThread().getId()+"-----------------------"+sf.format(new Date()));//打印線程啟動時間
        
        /*
         * 當采用 InterenetExplorerDriver時，可能會遇到一個安全問題提示："Protected Mode must be
         * set to the same value (enabled or disabled) for all
         * zones"。想要解決這一問題，需要設置特定的功能
         */
        DesiredCapabilities capability = DesiredCapabilities.internetExplorer();
        capability.setCapability(InternetExplorerDriver.INTRODUCE_FLAKINESS_BY_IGNORING_SECURITY_DOMAINS,true);
        
        //最好設置，用來匹配node中要使用的瀏覽器
        capability.setBrowserName("internet explorer");
        capability.setVersion("9");
        capability.setPlatform(Platform.WINDOWS);
        
        WebDriver driver = null;
        
        //設置本地驅動，如果你實例化Driver的時候是"WebDriver driver = new InternetExplorerDriver(capability)"這種方式，就必須設置
        System.setProperty("webdriver.ie.driver","D:\\IEDriverServer.exe");
        
        try{
            driver = new InternetExplorerDriver(capability);//本地測試，非遠程方式
//            driver = new RemoteWebDriver(new URL("http://localhost:4444/wd/hub"),capability);//遠程方式，想用多線程需要這么寫
            System.out.println(Thread.currentThread().getId()+"訪問163郵箱網頁開始時間："+sf.format(new Date()));
            
            driver.get("http://mail.163.com/");//打開網頁
            
            try {
                //等待頁面打開，超時設置3秒
                WebElement username = new WebDriverWait(driver, 3).until(new ExpectedCondition<WebElement>() {
                    public WebElement apply(WebDriver d) {
                        return d.findElement(By.id("idInput"));
                    }
                });
                if(null==username){
                    System.out.println(Thread.currentThread().getId()+" Visit http://mail.163.com/ Timeout !!!");
                    this.CaptureScreenshot(Thread.currentThread().getId()+"visit", driver);
                    driver.quit();
                    Thread.currentThread().interrupt();
                }else{
                    System.out.println(Thread.currentThread().getId()+"訪問163郵箱網頁結束時間："+sf.format(new Date()));
                    
                    username.clear();//清除username的value
                    username.sendKeys("testDemobyxy");//設置username的value
                    
                    WebElement pwd = driver.findElement(By.id("pwdInput"));
                    pwd.clear();
                    pwd.sendKeys("testDemobyxy123");
                    
                    WebElement submit = driver.findElement(By.id("loginBtn"));
                    
                    System.out.println(Thread.currentThread().getId()+"登錄開始時間："+sf.format(new Date()));
                    submit.click();//提交
                    
                    try {
                        //等待登錄成功，超時設置3秒
                        WebElement inbox = new WebDriverWait(driver, 3).until(new ExpectedCondition<WebElement>() {
                            public WebElement apply(WebDriver d) {
                                return  d.findElement(By.id("_mail_component_2_2"));
                            }
                        });
                        if(null==inbox){
                            System.out.println(Thread.currentThread().getId()+" Loign email Timeout !!!");
                            this.CaptureScreenshot(Thread.currentThread().getId()+"login", driver);
                            driver.quit();
                            Thread.currentThread().interrupt();
                        }else{
                            System.out.println(Thread.currentThread().getId()+"登錄成功時間："+sf.format(new Date()));
                            
                            System.out.println(Thread.currentThread().getId()+"點擊收件箱時間："+sf.format(new Date()));
                            inbox.click();
                            
                            try {
                                //等待打開收件箱頁面，超時設置3秒
                                Boolean flag = new WebDriverWait(driver, 3).until(new ExpectedCondition<Boolean>() {
                                    public Boolean apply(WebDriver d) {
                                        return  d.findElement(By.id("_mail_button_4_114")) != null;
                                    }
                                });
                                if(!flag){
                                    System.out.println(Thread.currentThread().getId()+" Open inBox page Timeout !!!");
                                    this.CaptureScreenshot(Thread.currentThread().getId()+"openInboxPage", driver);
                                    driver.quit();
                                    Thread.currentThread().interrupt();
                                }else{
                                    System.out.println(Thread.currentThread().getId()+"打開收件箱時間："+sf.format(new Date()));
                                    driver.quit();
                                }
                            } catch (Exception e) {
                                System.out.println(Thread.currentThread().getId()+" Open inBox page Error !!!");
                                e.printStackTrace();
                                this.CaptureScreenshot(Thread.currentThread().getId()+"openInboxPage", driver);
                                driver.quit();
                                Thread.currentThread().interrupt();
                            }
                        }
                    } catch (Exception e) {
                        System.out.println(Thread.currentThread().getId()+" Loign email Error !!!");
                        e.printStackTrace();
                        this.CaptureScreenshot(Thread.currentThread().getId()+"login", driver);
                        driver.quit();
                        Thread.currentThread().interrupt();
                    }
                }
            } catch (Exception e) {
                System.out.println(Thread.currentThread().getId()+" Visit http://mail.163.com/ Error !!!");
                e.printStackTrace();
                this.CaptureScreenshot(Thread.currentThread().getId()+"visit", driver);
                driver.quit();
                Thread.currentThread().interrupt();
            }
        }catch (Exception e) {
            e.printStackTrace();
            this.CaptureScreenshot(Thread.currentThread().getId()+"visit", driver);
            driver.quit();
        }finally{
            if(null!=driver){
                driver.quit();
            }
        }
    }

    /**
     * 截屏方法
     * @param fileName
     * @param driver
     */
    public void CaptureScreenshot(String fileName, WebDriver driver) {
        String dirName = "d:/screenshot";
        if (!(new File(dirName).isDirectory())) {
            new File(dirName).mkdir();
        }
        SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMdd-HHmmss");
        String time = sdf.format(new Date());
        TakesScreenshot tsDriver = (TakesScreenshot) driver;
        File image = new File(dirName + File.separator + time + "_" + fileName==null?"":fileName + ".png");
        tsDriver.getScreenshotAs(OutputType.FILE).renameTo(image);
    }
}

　　代碼比較簡單，下面主要說下多線程的問題。可以看到我上面的代碼是一個線程。由於需求上的要求要有一定數量的並發，所以我寫了一個測試類，里面只有一個main方法，將上面的代碼new出了很多實例並且啟動他們。結果我發現，我的電腦最多運行10個線程，如果運行10個以上，程序就會出錯，跑不起來了。所以又去看了官網和網上的一些博客。

　　要想多線程並發的運行WebDriver，必須同時滿足2個條件，首先你的測試程序是多線程，其次需要用到Selenium Server。下載位置如下圖：

　　下載下來后是一個jar包，需要在命令行中運行。里面有2個東西：hub和node。可以這么理解：hub是運行在一個服務器上的線程池，負責收集測試case提交的請求，將請求分配給匹配的node；而node就是工作線程，具體干活的。node可以有多個，運行在不同的系統上。hub和node也可以不在一個機器上。

　　多線程並發運行WebDriver的步驟：1.運行hub 2.運行node 3.運行test case 。下面說下具體實現方法。

　　1.運行hub。在命令行中輸入：java -jar selenium-server-standalone-2.37.0.jar -role hub -maxSession 40 -port 4444

參數中必須指明-role hub 才是運行hub。默認端口是4444，如果端口被占用就需要指定其他。-maxSession是最大處理的會話請求，我這里設置為40。如果不指定的話，默認是1（即單線程模式了）。

　　2.運行node。（先說下運行一個node情況）在命令行中輸入（下面的命令是一行敲完）：

java -Dwebdriver.ie.driver=D:\IEDriverServer.exe -jar selenium-server-standalone-2.37.0.jar -role node -hub http://127.0.0.1:4444/grid/register -maxSession 20 -browser "browserName=internet explorer,version=9,platform=WINDOWS,maxInstances=20" -port 5555

由於node是可以運行在不通系統上的，所以指定驅動位置-Dwebdriver.ie.driver=D:\IEDriverServer.exe。參數中必須指明-role node才是運行node。參數-hub 后面是第一步中hub的IP和端口：http：//hub的IP：端口/grid/register  。node默認的maxSession的值就是5（最多並發5個瀏覽器），即啟動一個node會默認有5個firefox、1個chrome、1個IE的實例。如果用IE瀏覽器的話，就算你的測試case是多線程，最終也會是一個一個的執行。但是如果在后面的-browser的參數中指明maxInstances=5，那么就會同時運行5個瀏覽器。-browser參數是指明node可以用的瀏覽器信息。注意，如果node的maxSession和maxInstances設置的有問題，那么hub的命令窗口中會給出警告。通過這里能夠知道你的node是否設置成功。運行node后，窗口中也會顯示該node的信息。-port是端口號，默認端口是5555，如果端口被占用就需要指定其他。如果你啟動第二個node的話，端口就必須指定了，不能是5555。

　　我設置的node是只運行IE，並且並發數是20，最多有20個IE瀏覽器在運行。node中的maxSession的值不能超過hub中的。如果想多線程並發要在hub和node的參數中同時指明maxSession值。node中如果用IE瀏覽器，指明maxSession后還需要指明同樣大小的maxInstances值。我的例子最終會同時運行20個IE瀏覽器。maxSession是說node可以有幾個瀏覽器同時運行，而maxInstances是說某個瀏覽器可以有幾個同時運行。由於我的電腦運行20個IE已經有些卡了，那么可以再另外一個電腦上再運行一個20Session大小的node。個人測試結果：運行一個20Session大小的node和運行2個10Session大小的node沒什么差別。運行多個node主要還是為了能夠分布式的測試，不至於一個電腦打開太多瀏覽器。

　　3.運行test case。首先將上面代碼中的44和47行注釋掉，將48行注釋打開。我們需要用遠程的方式將請求提交給hub（后面的/wd/hub是固定的）。

WebDriver driver = new RemoteWebDriver(new URL("http://hub的IP:端口/wd/hub"),capability);

由於是遠程的方式，所以44行的設置就沒什么用了。下面你可以運行你的程序了，你會發現同時啟動20個線程，就會有20個IE瀏覽器同時在運行。