selenium webdriver 杂记 - 自定义Table操作类(代码展示)

前言：WebDriver中没有关于Table的相关操作类，下面用代码直接展示如何自定义Table操作类
定义两个类，Point.java Table.java
Point.java
中定义了表格中获取某个单元格时行列坐标，以此来唯一标识指定的单元格
Point中定义两个属性{rowNo,colNo}，分别对应指定单元格的{行，列}，索引起始均为0 ，如{0,0}表示第0行，第0列的单元格；
Table.java
该类中方法包含{取指定行元素(private)，取所有行元素(private)，取指定行中所有Cell，取指定行和列Cell，取指定列中所有Cell}
取指定行元素(private) --> private WebElement getRow(Point point)
该方法用于获取 指定行对应的WebElement实例，point中属性{rowNo,colNo}初始化rowNo即可，将colNo置为null , or anyelse
取所有行元素(private) --> private List<WebElement> getAllRows()
该方法用于获取表格中所有行对应的WebElement实例集合
取指定行中所有Cell --> public List<WebElement> getCellsOfRow(Point point)
该方法用于获取表格中指定行中所有单元格对应的WebElement实例集合，point中属性{rowNo,colNo}初始化rowNo即可
取指定行和列Cell --> public WebElement getCellOfRow(Point point)
该方法用于获取表格中指定行、列的单元格
取指定列中所有Cell --> public List<WebElement> getCellsOfCol(Point point)
该方法用于获取表格中指定所在列的所有单元格的集合，注：当某行不存在该列时该方法会将该单元格元素置为null添加到List数组中
将上一个方法(public List<WebElement> getCellsOfCol(Point point))得到的List数组中为null的元素移除，public List<WebElement> TrimCellsOfCol

代码如下：
package demo.table;

public class Point {
private Integer rowNo;
private Integer colNo;
public Point(){}
public Point(Integer rowNo, Integer colNo) {
super();
this.rowNo = rowNo;
this.colNo = colNo;
}
public int getRowNo() {
return rowNo;
}
public void setRowNo(Integer rowNo) {
this.rowNo = rowNo;
}
public int getColNo() {
return colNo;
}
public void setColNo(Integer colNo) {
this.colNo = colNo;
}
@Override
public String toString() {
return "Point [rowNo=" + rowNo + ", colNo=" + colNo + "]";
}

}

package demo.table;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Iterator;
import java.util.List;
import org.openqa.selenium.By;
import org.openqa.selenium.WebElement;
import com.jd.acceptance.page.Page;
import com.jd.acceptance.vo.Point;
/**
 * @author xuleilei
 * Table元素处理工具：
 * {取指定行元素(private)，取所有行元素(private)，取指定行中所有Cell，取指定行和列Cell，取指定列中所有Cell}
 */
public class Table {
private By by;
private WebElement table;
public Table(By by) {
super();
this.by = by;
this.table = Page.getElement(this.by);
}

/**
* 取指定行元素
* @param point
* point对象中指定行(rowNo)属性
* @return
* 指定行所对应的WebElement对象
*/
private WebElement getRow(Point point){
WebElement target = null;
List<WebElement> allRows = this.getAllRows();
try{
target = allRows.get(point.getRowNo());
}catch(IndexOutOfBoundsException e){
throw new MyException(">>>rowNo out of range : \n"
+ "rowNo should be [rowNo >= 0 || rowNo < "+allRows.size()+"]");
}
return target;
}    
/**
* 取所有行元素
* @return
* List<WebElement>返回表格中所有行元素
*/
private List<WebElement> getAllRows(){
List<WebElement> allRows = null;
allRows = this.table.findElements(By.tagName("tr"));
return allRows;
}
/**
* 取指定行中所有Cell
* @param point
* point对象中指定行(rowNo)属性
* @return
* 返回指定行中所有单元格元素
*/
public List<WebElement> getCellsOfRow(Point point){
List<WebElement> cells = null;
WebElement row = this.getRow(point);
cells = row.findElements(By.tagName("th"));
if(cells.size()>0){
return cells;
}
cells = row.findElements(By.tagName("td"));
if(cells.size()>0){
return cells;
}
return null;
}
/**
* 取指定行和列Cell
* @param point
* point对象中指定行(rowNo)和列(colNo)
* @return
* 返回指定行和列对象的WebElement对象
*/
public WebElement getCellOfRow(Point point){
List<WebElement> cells = null;  
WebElement Row = this.getRow(point);
    WebElement target = null;  
    int cell = point.getColNo();
    //分开处理列里面"<th>"、"<td>"两种标签
    cells = Row.findElements(By.tagName("th"));
    if(cells.size() == 0){
   
cells = Row.findElements(By.tagName("td"));
    }
    if(cells.size() == 0){
   
return null;
    }
    try{
target = cells.get(cell);  
    }catch(IndexOutOfBoundsException e){
   
throw new MyException(">>>colNo out of range : \n"
+ "colNo should be [colNo >= 0 || colNo < "+cells.size()+"]");
    }
return target; 

}
/**
* 取指定列中所有Cell
* @param point
* point对象中指定列(colNo)
* @return
*
返回指定列中所有单元格元素，该方法会将不存在的列所对应的单元格元素置为null,添加到List结果中

*/

public List<WebElement> getCellsOfCol(Point point){
int colNo = point.getColNo();
if(colNo >= this.maxLength()){
throw new MyException(">>>colNo should be less than the max number("+this.maxLength()+")"
+ " of any row's column amount in this table");
}
List<WebElement> cells = new ArrayList<WebElement>();
List<WebElement> tmpCells = null;
Iterator<WebElement> allRows = this.getAllRows().iterator();
WebElement row = null;
while(allRows.hasNext()){
row = allRows.next();
try{
tmpCells = row.findElements(By.tagName("th"));
if(tmpCells.size() == 0){
tmpCells = row.findElements(By.tagName("td"));
}
if(tmpCells.size()>0){
if(colNo < tmpCells.size()){
cells.add(tmpCells.get(colNo));
}else{
cells.add(null);
}
}else    //当tr中不存在td 和 th 时，添加null
cells.add(null);

}catch(IndexOutOfBoundsException e){
   
throw new MyException(">>>colNo out of range : \n"
+ "colNo should be [colNo >= 0 || colNo < "+tmpCells.size()+"]");
}

}
return cells;
}
/**
* 获取表格中行的最大宽度，即行的最大单元格个数
* @return
*/
private int maxLength(){
int max = 0;
int arrLength = this.getAllRows().size();
Iterator<WebElement> allRows = this.getAllRows().iterator();
int maxArr[] = new int[arrLength];
WebElement row = null;
List<WebElement> tmpCells = null;
int i = 0;
while(allRows.hasNext()){
row = allRows.next();
if(i == 0){
tmpCells = row.findElements(By.tagName("th"));
if(tmpCells.size() == 0){
tmpCells = row.findElements(By.tagName("td"));
}
}else{
tmpCells = row.findElements(By.tagName("td"));
}
maxArr[i] = tmpCells.size();
i++;
}
Arrays.sort(maxArr);
max = maxArr[arrLength-1];
return max;
}
/**
* 将指定列中所有单元格
* @param cells
* @return
*/
@SuppressWarnings("null")
public List<WebElement> TrimCellsOfCol(List<WebElement> cells){
List<WebElement> tmp = null;
Iterator<WebElement> cellIter = cells.iterator();
while(cellIter.hasNext()){
WebElement e = cellIter.next();
if(e != null){
tmp.add(e);
}
}
return tmp;
}

}