fix(dialog): add waitFor to enable focus

Author: DobroTora

src/components/dialog/dialog.interactions.stories.tsx

@@ -1,6 +1,6 @@
 import React, { useState } from "react";
 import { StoryObj } from "@storybook/react";
-import { userEvent, within, expect } from "@storybook/test";
+import { userEvent, within, expect, waitFor } from "@storybook/test";
 
 import Dialog from ".";
 import Button from "../button";
@@ -65,11 +65,12 @@ export const FocusManagement: Story = {
     await userEvent.click(openButton);
 
     const dialog = await portal.findByRole("dialog");
+
+    await waitFor(() => expect(dialog).toBeVisible());
+
     const closeButton = within(dialog).getByRole("button", { name: /close/i });
     const firstInput = within(dialog).getByLabelText(/name/i);
 
-    await expect(dialog).toBeVisible();
-
     await userEvent.tab();
     await userEvent.tab();