Merge pull request #7089 from Shopify/fix-flaky-poll-app-logs

Fix flaky test in usePollAppLogs.test.tsx

Author: gonzaloriestra

packages/app/src/cli/services/app-logs/logs-command/ui/components/hooks/usePollAppLogs.test.tsx

@@ -404,6 +404,11 @@ describe('usePollAppLogs', () => {
     // needed to await the render
     await vi.advanceTimersByTimeAsync(0)
 
+    // Wait for the async polling function to execute
+    await waitForMockCalls(mockedPollAppLogs, 1)
+    // Flush React 19 batched state updates so hook.lastResult reflects the new state
+    await vi.advanceTimersByTimeAsync(0)
+
     // Initial invocation, 429 returned
     expect(mockedPollAppLogs).toHaveBeenCalledTimes(1)
 
@@ -447,6 +452,11 @@ describe('usePollAppLogs', () => {
     // needed to await the render
     await vi.advanceTimersByTimeAsync(0)
 
+    // Wait for the async polling function to execute
+    await waitForMockCalls(mockedPollAppLogs, 1)
+    // Flush React 19 batched state updates so hook.lastResult reflects the new state
+    await vi.advanceTimersByTimeAsync(0)
+
     // Initial invocation, 422 returned
     expect(mockedPollAppLogs).toHaveBeenCalledTimes(1)