Adding wait for button to exist

Signed-off-by: Lukas Grossmann <lgrossma@redhat.com>
redhat-developer · Jul 12, 2024 · e2e6002 · e2e6002
1 parent a172f1d
commit e2e6002
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/test/ui/common/ui/webview/newComponentWebViewEditor.ts b/test/ui/common/ui/webview/newComponentWebViewEditor.ts
@@ -208,7 +208,7 @@ export class LocalCodeBasePage extends Page {
 
     public async clickCreateComponent(): Promise<void> {
         await this.enterWebView(async (webView) => {
-            const button = await this.getCreateComponentButton(webView);
+            const button = await this.createButtonExists(webView);
             await button.click();
         });
     }
@@ -221,6 +221,19 @@ export class LocalCodeBasePage extends Page {
         return await webView.findWebElement(By.xpath('//button[contains(text(), "Select Folder")]'));
     }
 
+    private async createButtonExists(webView: WebView, timeout = 60_000): Promise<WebElement> {
+        return webView.getDriver().wait(async () => {
+            try {
+                const button = await this.getCreateComponentButton(webView);
+                if (button) {
+                    return button;
+                }
+            } catch (err) {
+                return null;
+            }
+        }, timeout);
+    }
+
     private async getCreateComponentButton(webView: WebView): Promise<WebElement> {
         return await webView.findWebElement(By.xpath('//span[contains(text(), "Create Component")]'));
     }