add await for act

Signed-off-by: Viktor Tsvetkov <142901247+vtsvetkov-splunk@users.noreply.github.com>
splunk · Jan 28, 2025 · 604b4f4 · 604b4f4
1 parent a76f613
commit 604b4f4
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/ui/src/components/EntityModal/EntityModal.test.tsx b/ui/src/components/EntityModal/EntityModal.test.tsx
@@ -491,12 +491,12 @@ describe('Oauth - separated endpoint authorization', () => {
 
         // triggering manually external oauth window behaviour after success authorization
         const code = '200';
-        window.getMessage({ code, state: stateCodeFromUrl, error: undefined });
-
-        await waitFor(async () => {
-            expect(requestHandler).toHaveBeenCalledTimes(1);
+        await act(async () => {
+            window.getMessage({ code, state: stateCodeFromUrl, error: undefined });
         });
 
+        expect(requestHandler).toHaveBeenCalledTimes(1);
+
         const receivedRequest: Request = requestHandler.mock.calls[0][0];
         const receivedBody = await receivedRequest.text();
 
@@ -534,7 +534,7 @@ describe('Oauth - separated endpoint authorization', () => {
         // triggering manually external oauth window behaviour after success authorization
         const code = '200';
         const passedState = `tests${stateCodeFromUrl}`;
-        act(() => {
+        await act(async () => {
             window.getMessage({ code, state: passedState, error: undefined });
         });