Introducing the Table Class:
public class CustomTable {
private WebDriver driver;
private By baseLocator;
private By headCellLocator;
private By bodyRowLocator;
private By bodyCellLocator;
private String[] headColumns;
private List<WebElement> foundRows;
public CustomTable(WebDriver driver, By baseLocator) {
this.driver = driver;
this.baseLocator = baseLocator;
}
public CustomTable initialize(By headCellLocator, By bodyRowLocator, By bodyCellLocator) {
this.headCellLocator = headCellLocator;
this.bodyRowLocator = bodyRowLocator;
this.bodyCellLocator = bodyCellLocator;
this.extractHeadColumns();
return this;
}
private void extractHeadColumns() {
List<WebElement> headCells = this.driver
.findElement(this.baseLocator)
.findElements(this.headCellLocator);
for(WebElement headCell : headCells) {
ArrayUtils.add(this.headColumns, headCell.getText().trim());
}
}
public CustomTable getRows() {
this.foundRows = this.driver
.findElement(this.baseLocator)
.findElements(this.bodyRowLocator);
return this;
}
public CustomTable filterRowsByColumnValue(String columnName, String columnValue) {
int columnIndex = ArrayUtils.indexOf(this.headColumns, columnName);
getRows();
List<WebElement> filteredRows = new ArrayList<WebElement>();
for (WebElement row : foundRows) {
String cellText = row.findElements(this.bodyCellLocator)
.get(columnIndex)
.getText()
.trim();
if (cellText.equals(columnValue)) {
filteredRows.add(row);
}
}
this.foundRows = filteredRows;
return this;
}
public String[] getColumnValues(String columnName) {
int columnIndex = ArrayUtils.indexOf(this.headColumns, columnName);
String[] columnValues = null;
String cellText;
for (WebElement row : foundRows) {
cellText = row.findElements(this.bodyCellLocator)
.get(columnIndex)
.getText()
.trim();
ArrayUtils.add(columnValues, cellText);
}
return columnValues;
}
}
CustomTable customTable = new CustomTable(By.cssSelector('body .customtable'));
customTable.initialize(By.cssSelector('thead th'), By.cssSelector('tbody tr'), By.cssSelector('td'));
customTable.getRows().getColumnValues('Product Name');
// or
customTable.filterRowsByColumnValue('Category', 'Electronics').getColumnValues('Price');