object list test fix

Evolveum · Jun 3, 2020 · 1feb37d · 1feb37d
1 parent 742702a
commit 1feb37d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...veum/midpoint/schrodinger/component/assignmentholder/AssignmentHolderObjectListTable.java b/...veum/midpoint/schrodinger/component/assignmentholder/AssignmentHolderObjectListTable.java
@@ -79,7 +79,8 @@ public SelenideElement getToolbarButton(String iconCssClass){
         ElementsCollection toolbarButtonsList = buttonToolbar
                 .findAll(By.tagName("button"));
         for (SelenideElement button : toolbarButtonsList) {
-            if (button.$(Schrodinger.byElementAttributeValue("i", "class", iconCssClass)).exists()) {
+            SelenideElement iconElement = button.$(By.tagName("i")).waitUntil(Condition.visible, MidPoint.TIMEOUT_DEFAULT_2_S);
+            if (iconElement != null && iconElement.exists() && iconElement.getAttribute("class") != null && iconElement.getAttribute("class").contains(iconCssClass)) {
                 buttonElement = button;
             }
         }