wait for overlay to be opacity-0

yext · Oct 25, 2023 · 9c9435b · 9c9435b
1 parent d8a1bef
commit 9c9435b
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/e2e-tests/tests/infra/StudioPlaywrightPage.ts b/e2e-tests/tests/infra/StudioPlaywrightPage.ts
@@ -134,8 +134,14 @@ export default class StudioPlaywrightPage {
    */
   async waitForLoadState() {
     await this.page.waitForLoadState();
-    await expect(
-      this.page.getByRole("button", { name: "Open Add Element Menu" })
-    ).toBeVisible();
+    const overlayDomEl = this.page.getByTestId("loading-overlay")
+    await expect.poll(() => overlayDomEl.evaluate((e: HTMLElement) => {
+       console.log(e.className)
+      return e.className.includes("opacity-0")
+    }), {
+      message: "Waiting for LoadingOverlay to finish.",
+      timeout: 5000,
+    }).toBeTruthy()
+    console.log("TRUTHED!")
   }
 }
diff --git a/packages/studio-ui/src/components/LoadingOverlay.tsx b/packages/studio-ui/src/components/LoadingOverlay.tsx
@@ -28,7 +28,7 @@ export default function LoadingOverlay(
 
   return (
     <>
-      <div className={overlayClassname}>{overlay}</div>
+      <div className={overlayClassname} data-testid="loading-overlay">{overlay}</div>
       <div className={childrenWrapperClassname}>{props.children}</div>
     </>
   );