[PR feedback] Address failing EuiHeaderLinks desktop play()

cee-chen · cee-chen · commit 9df1ada7bd99 · 2024-09-06T10:26:04.000-07:00
diff --git a/packages/eui/src/components/header/header_links/header_links.stories.tsx b/packages/eui/src/components/header/header_links/header_links.stories.tsx
@@ -46,11 +46,18 @@ export const Playground: Story = {
       </EuiHeaderSection>
     </EuiHeader>
   ),
-  play: async ({ canvasElement }) => {
+  play: async ({ canvasElement, step }) => {
     const canvas = within(canvasElement);
-    await waitFor(async () => {
-      await fireEvent.click(canvas.getByLabelText('Open menu'));
+    await waitFor(() => canvas.getByLabelText('App menu')); // Wait for rendering to finish
+
+    step('Open mobile popover', async () => {
+      await waitFor(async () => {
+        const mobilePopover = canvas.queryByLabelText('Open menu');
+        if (mobilePopover) {
+          await fireEvent.click(mobilePopover);
+          await canvas.waitForEuiPopoverVisible();
+        }
+      });
     });
-    await canvas.waitForEuiPopoverVisible();
   },
 };
