Refactor waitForLoadingToFinish function to include a timeout

luismarques-io · Jan 21, 2024 · c98c1d8 · c98c1d8
1 parent d455f83
commit c98c1d8
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/src/test/test-utils.tsx b/src/test/test-utils.tsx
@@ -39,12 +39,15 @@ export const createComment = async (properties?: any) => {
 export const waitForLoadingToFinish = async () => {
   const tmpLoadingElements = [...screen.queryAllByTestId(/loading/i), ...screen.queryAllByText(/loading/i)];
   if (tmpLoadingElements.length > 0) {
-    await waitFor(async () => {
-      await waitForElementToBeRemoved(() => [
-        ...screen.queryAllByTestId(/loading/i),
-        ...screen.queryAllByText(/loading/i),
-      ]);
-    });
+    await waitFor(
+      async () => {
+        await waitForElementToBeRemoved(() => [
+          ...screen.queryAllByTestId(/loading/i),
+          ...screen.queryAllByText(/loading/i),
+        ]);
+      },
+      { timeout: 10000 }
+    );
   }
 };