Selenium Webdriver——Table類封裝

標籤：

WebTable.java

import java.util.List;import org.openqa.selenium.By;import org.openqa.selenium.WebElement;public class WebTable {        private WebElement webTable;    public WebTable(WebElement webElement){        this.webTable = webElement;    }        //等到表格的行數    public int getRowCount(){        List<WebElement>rowCounts = webTable.findElements(By.tagName("tr"));        return rowCounts.size();    }        //得到指定行的列數    public int getColCount(int rowId){        List<WebElement>rowCounts = webTable.findElements(By.tagName("tr"));        //取得當前的tr        WebElement rowNum = rowCounts.get(rowId);        //計算當前的td數        List<WebElement>colCounts =rowNum.findElements(By.tagName("td"));        return colCounts.size();    }        // 得到指定儲存格的內容    public  String getCellText(int rowIdx, int colIdx) {                String text = "";                try{            List<WebElement> rowCounts = webTable.findElements(By.tagName("tr"));            WebElement currentRow = rowCounts.get(rowIdx);            List<WebElement> td = currentRow.findElements(By.tagName("td"));            WebElement cell = td.get(colIdx);             text = cell.getText();               }catch(IndexOutOfBoundsException e){            System.out.println("超出table邊界值");        }                return text;     }}

 

對webTable類進行測試

import org.openqa.selenium.By;import org.openqa.selenium.WebDriver;import org.openqa.selenium.firefox.FirefoxDriver;import org.testng.Assert;import org.testng.annotations.Test;public class testTable {      @Test  public void testTableWebElement() {            WebDriver driver = new FirefoxDriver();      driver.manage().window().maximize();      driver.get("F:\\table.html");      WebTable webtable = new WebTable(driver.findElement(By.tagName("table")));      Assert.assertEquals(4, webtable.getRowCount());      Assert.assertEquals(5, webtable.getColCount(1));      System.out.println( webtable.getCellText(1, 1));      driver.quit();  }}

 

Selenium Webdriver——Table類封裝