diff options
Diffstat (limited to 'testbench-api/src')
| -rw-r--r-- | testbench-api/src/main/java/com/vaadin/testbench/elements/TreeGridElement.java | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/testbench-api/src/main/java/com/vaadin/testbench/elements/TreeGridElement.java b/testbench-api/src/main/java/com/vaadin/testbench/elements/TreeGridElement.java index 13dfbbf565..3eb8b30b5e 100644 --- a/testbench-api/src/main/java/com/vaadin/testbench/elements/TreeGridElement.java +++ b/testbench-api/src/main/java/com/vaadin/testbench/elements/TreeGridElement.java @@ -119,6 +119,24 @@ public class TreeGridElement extends GridElement { } /** + * Check whether the given indices correspond to a cell that contains a + * visible hierarchy toggle element. + * + * @param rowIndex + * 0-based row index + * @param hierarchyColumnIndex + * 0-based index of the hierarchy column + * @return {@code true} if this cell has the expand toggle visible + */ + public boolean hasExpandToggle(int rowIndex, int hierarchyColumnIndex) { + WebElement expandElement = getExpandElement(rowIndex, + hierarchyColumnIndex); + List<String> classes = Arrays + .asList(expandElement.getAttribute("class").split(" ")); + return classes.contains("expanded") || classes.contains("collapsed"); + } + + /** * Gets the expand/collapse element for the given row. * * @param rowIndex |