From 438c3afdc927aa16345dde44f2ba559892dfb9ea Mon Sep 17 00:00:00 2001
From: anovazzi1 <otavio2204@gmail.com>
Date: Thu, 28 Nov 2024 17:01:58 -0300
Subject: [PATCH] Improve test reliability and performance in
 toggleComponent.spec.ts (#4927)

refactor: Improve test reliability and performance in toggleComponent.spec.ts
---
 src/frontend/tests/core/unit/toggleComponent.spec.ts | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/frontend/tests/core/unit/toggleComponent.spec.ts b/src/frontend/tests/core/unit/toggleComponent.spec.ts
index 8fd41f65e..e11d3642b 100644
--- a/src/frontend/tests/core/unit/toggleComponent.spec.ts
+++ b/src/frontend/tests/core/unit/toggleComponent.spec.ts
@@ -25,7 +25,9 @@ test(
 
     while (modalCount === 0) {
       await page.getByText("New Flow", { exact: true }).click();
-      await page.waitForTimeout(3000);
+      await page.waitForSelector('[data-testid="modal-title"]', {
+        timeout: 3000,
+      });
       modalCount = await page.getByTestId("modal-title")?.count();
     }
     await page.waitForSelector('[data-testid="blank-flow"]', {
@@ -36,7 +38,9 @@ test(
     await page.getByTestId("sidebar-search-input").click();
     await page.getByTestId("sidebar-search-input").fill("directory");
 
-    await page.waitForTimeout(1000);
+    await page.waitForSelector('[data-testid="dataDirectory"]', {
+      timeout: 30000,
+    });
     await page
       .getByTestId("dataDirectory")
       .dragTo(page.locator('//*[@id="react-flow-id"]'));