自定义table表格

package com.gloryroad.testcase;



 import java.util.List;

 import java.util.NoSuchElementException;



 import org.openqa.selenium.By;

 import org.openqa.selenium.WebElement;



 public class Table {

private WebElement _table;

public Table(WebElement table){

set_table(table);

}

public WebElement get_table() {

return _table;

}

public void set_table(WebElement _table) {

this._table = _table;

}


public int getRowCount(){

List<WebElement> tableRows=_table.findElements(By.tagName("tr"));

return tableRows.size();



}

/**

* 获取列数

* @return

*/

public int getColumnCount(){


List<WebElement> tableRows=_table.findElements(By.tagName("tr"));

return tableRows.get(0).findElements(By.tagName("td")).size();

}

/**

* 获取表格中的某个单元格

* @param row

* @param cols

* @return

*/

public WebElement getCell(int row,int cols){

try {

List<WebElement> tableRows=_table.findElements(By.tagName("tr"));

System.out.println("行总数:"+tableRows.size());

System.out.println("行号:"+row);

WebElement currentRow=tableRows.get(row-1);

List<WebElement> tableCols=currentRow.findElements(By.tagName("td"));

System.out.println("列总数:"+tableCols.size());

WebElement cell=tableCols.get(cols-1);

System.out.println("列号:"+cols);

return cell;

} catch (Exception e) {

// TODO: handle exception

throw new NoSuchElementException("么有找到相关元素");


}



}


public WebElement getWebElementInCell(int row, int cols,By by){


WebElement cell = null;

try {

List<WebElement> tableRows=_table.findElements(By.tagName("tr"));

System.out.println("行总数:"+tableRows.size());

System.out.println("行号:"+row);

WebElement currentRow=tableRows.get(row-1);

List<WebElement> tableCols=currentRow.findElements(By.tagName("td"));

System.out.println("列总数:"+tableCols.size());

cell=tableCols.get(cols-1);

System.out.println("列号:"+cols);



} catch (Exception e) {

// TODO: handle exception

e.printStackTrace();

}


return cell.findElement(by);

}

 }