test(browser): add waitForElementToBeRemoved at overlay test

toss · Dec 29, 2024 · 1b07d64 · 1b07d64
1 parent 6cb845b
commit 1b07d64
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/packages/browser/test/index.test.tsx b/packages/browser/test/index.test.tsx
@@ -1,4 +1,4 @@
-import { cleanup, render, screen } from '@testing-library/react';
+import { cleanup, render, screen, waitForElementToBeRemoved } from '@testing-library/react';
 import { userEvent } from '@testing-library/user-event';
 import { afterEach, describe, expect, test } from 'vitest';
 import { useFunnel } from '../src/index.js';
@@ -110,6 +110,8 @@ describe('Test useFunnel browser router', () => {
     // close overlay
     await user.click(screen.getByText('Close Overlay'));
 
+    await waitForElementToBeRemoved(() => screen.queryAllByText('overlay: vitest'));
+
     expect(screen.queryByText('Go A')).toBeNull();
     expect(screen.queryByText('Go B')).not.toBeNull();
     expect(screen.queryByText('overlay: vitest')).toBeNull();