Refactor: refactor wait times and selectors in filterEdge-shard-1.spe…

…c.ts (#5049) * refactor: Update wait times and selectors in filterEdge-shard-1.spec.ts
langflow-ai · Dec 5, 2024 · 8c73d5d · 8c73d5d
1 parent 18c423b
commit 8c73d5d
Showing 1 changed file with 16 additions and 12 deletions.
diff --git a/src/frontend/tests/extended/features/filterEdge-shard-1.spec.ts b/src/frontend/tests/extended/features/filterEdge-shard-1.spec.ts
@@ -25,31 +25,36 @@ 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.waitForTimeout(1000);
-
-    await page.getByTestId(
-      "input-list-plus-btn-edit_metadata_indexing_include-2",
-    );
 
     await page.getByTestId("blank-flow").click();
 
-    await page.waitForTimeout(1000);
+    await page.waitForSelector('[data-testid="sidebar-options-trigger"]', {
+      timeout: 3000,
+    });
 
     await page.getByTestId("sidebar-options-trigger").click();
-    await page
-      .getByTestId("sidebar-legacy-switch")
-      .isVisible({ timeout: 5000 });
+
+    await expect(page.getByTestId("sidebar-legacy-switch")).toBeVisible({
+      timeout: 5000,
+    });
     await page.getByTestId("sidebar-legacy-switch").click();
     await expect(page.getByTestId("sidebar-legacy-switch")).toBeChecked();
     await page.getByTestId("sidebar-options-trigger").click();
 
     await page.getByTestId("sidebar-search-input").click();
     await page.getByTestId("sidebar-search-input").fill("retrievalqa");
 
-    await page.waitForTimeout(1000);
+    await page.waitForSelector(
+      '[data-testid="langchain_utilitiesRetrieval QA"]',
+      {
+        timeout: 3000,
+      },
+    );
     await page
       .getByTestId("langchain_utilitiesRetrieval QA")
       .dragTo(page.locator('//*[@id="react-flow-id"]'));
@@ -59,7 +64,6 @@ test(
     await page.getByTestId("zoom_out").click();
     await page.getByTestId("zoom_out").click();
     await page.getByTestId("zoom_out").click();
-    await page.waitForTimeout(500);
 
     let visibleElementHandle;