From 830dfc4d9919b1862cc2b60b5ab734714f02b293 Mon Sep 17 00:00:00 2001
From: Tom Avalexing <nickmyalex@gmail.com>
Date: Wed, 18 Oct 2023 19:42:48 +0300
Subject: [PATCH] add clicking on dropdown menu test (#2103)

* add clicking on dropdown menu test

* add play function
---
 .../__stories__/DropdownMenu.stories.tsx      | 94 ++++++++++++++-----
 1 file changed, 68 insertions(+), 26 deletions(-)

diff --git a/front/src/modules/ui/layout/dropdown/components/__stories__/DropdownMenu.stories.tsx b/front/src/modules/ui/layout/dropdown/components/__stories__/DropdownMenu.stories.tsx
index 43fb5cf66..40c67e7ef 100644
--- a/front/src/modules/ui/layout/dropdown/components/__stories__/DropdownMenu.stories.tsx
+++ b/front/src/modules/ui/layout/dropdown/components/__stories__/DropdownMenu.stories.tsx
@@ -1,7 +1,10 @@
 import { useState } from 'react';
 import styled from '@emotion/styled';
 import { Decorator, Meta, StoryObj } from '@storybook/react';
+import { userEvent, within } from '@storybook/testing-library';
+import { PlayFunction } from '@storybook/types';
 
+import { Button } from '@/ui/input/button/components/Button';
 import { DropdownMenuSkeletonItem } from '@/ui/input/relation-picker/components/skeletons/DropdownMenuSkeletonItem';
 import { MenuItem } from '@/ui/navigation/menu-item/components/MenuItem';
 import { MenuItemMultiSelectAvatar } from '@/ui/navigation/menu-item/components/MenuItemMultiSelectAvatar';
@@ -9,6 +12,8 @@ import { MenuItemSelectAvatar } from '@/ui/navigation/menu-item/components/MenuI
 import { Avatar } from '@/users/components/Avatar';
 import { ComponentDecorator } from '~/testing/decorators/ComponentDecorator';
 
+import { DropdownScope } from '../../scopes/DropdownScope';
+import { DropdownMenu } from '../DropdownMenu';
 import { DropdownMenuHeader } from '../DropdownMenuHeader';
 import { DropdownMenuInput } from '../DropdownMenuInput';
 import { DropdownMenuInputContainer } from '../DropdownMenuInputContainer';
@@ -18,20 +23,33 @@ import { StyledDropdownMenu } from '../StyledDropdownMenu';
 import { StyledDropdownMenuSeparator } from '../StyledDropdownMenuSeparator';
 import { StyledDropdownMenuSubheader } from '../StyledDropdownMenuSubheader';
 
-const meta: Meta<typeof StyledDropdownMenu> = {
+const meta: Meta<typeof DropdownMenu> = {
   title: 'UI/Layout/Dropdown/DropdownMenu',
-  component: StyledDropdownMenu,
-  decorators: [ComponentDecorator],
+  component: DropdownMenu,
+
+  decorators: [
+    ComponentDecorator,
+    (Story) => (
+      <DropdownScope dropdownScopeId="testDropdownMenu">
+        <Story />
+      </DropdownScope>
+    ),
+  ],
+  args: {
+    clickableComponent: <Button title="Open Dropdown" />,
+    dropdownHotkeyScope: { scope: 'testDropdownMenu' },
+    dropdownOffset: { x: 0, y: -8 },
+  },
   argTypes: {
-    as: { table: { disable: true } },
-    children: { control: false },
-    theme: { table: { disable: true } },
-    width: { type: 'number', defaultValue: undefined },
+    clickableComponent: { control: false },
+    dropdownHotkeyScope: { control: false },
+    dropdownOffset: { control: false },
+    dropdownComponents: { control: false },
   },
 };
 
 export default meta;
-type Story = StoryObj<typeof StyledDropdownMenu>;
+type Story = StoryObj<typeof DropdownMenu>;
 
 const FakeContentBelow = () => (
   <div style={{ position: 'absolute' }}>
@@ -167,15 +185,30 @@ const WithContentBelowDecorator: Decorator = (Story) => (
   </StyledFakeBelowContainer>
 );
 
+const playInteraction: PlayFunction<any, any> = async ({ canvasElement }) => {
+  const canvas = within(canvasElement);
+
+  const button = await canvas.findByRole('button');
+
+  userEvent.click(button);
+};
+
 export const Empty: Story = {
-  args: { children: <StyledFakeMenuContent /> },
+  args: {
+    dropdownComponents: (
+      <StyledDropdownMenu>
+        <StyledFakeMenuContent />
+      </StyledDropdownMenu>
+    ),
+  },
+  play: playInteraction,
 };
 
 export const WithHeaders: Story = {
   decorators: [WithContentBelowDecorator],
   args: {
-    children: (
-      <>
+    dropdownComponents: (
+      <StyledDropdownMenu>
         <DropdownMenuHeader>Header</DropdownMenuHeader>
         <StyledDropdownMenuSeparator />
         <StyledDropdownMenuSubheader>Subheader 1</StyledDropdownMenuSubheader>
@@ -191,31 +224,33 @@ export const WithHeaders: Story = {
             <MenuItem text={name} />
           ))}
         </DropdownMenuItemsContainer>
-      </>
+      </StyledDropdownMenu>
     ),
   },
+  play: playInteraction,
 };
 
 export const SearchWithLoadingMenu: Story = {
   decorators: [WithContentBelowDecorator],
   args: {
-    children: (
-      <>
+    dropdownComponents: (
+      <StyledDropdownMenu>
         <DropdownMenuSearchInput value="query" autoFocus />
         <StyledDropdownMenuSeparator />
         <DropdownMenuItemsContainer hasMaxHeight>
           <DropdownMenuSkeletonItem />
         </DropdownMenuItemsContainer>
-      </>
+      </StyledDropdownMenu>
     ),
   },
+  play: playInteraction,
 };
 
 export const WithInput: Story = {
   decorators: [WithContentBelowDecorator],
   args: {
-    children: (
-      <>
+    dropdownComponents: (
+      <StyledDropdownMenu>
         <DropdownMenuInputContainer>
           <DropdownMenuInput defaultValue="Lorem ipsum" autoFocus />
         </DropdownMenuInputContainer>
@@ -225,29 +260,36 @@ export const WithInput: Story = {
             <MenuItem text={name} />
           ))}
         </DropdownMenuItemsContainer>
-      </>
+      </StyledDropdownMenu>
     ),
   },
+  play: playInteraction,
 };
 
 export const SelectableMenuItemWithAvatar: Story = {
   decorators: [WithContentBelowDecorator],
   args: {
-    children: (
-      <DropdownMenuItemsContainer hasMaxHeight>
-        <FakeSelectableMenuItemList hasAvatar />
-      </DropdownMenuItemsContainer>
+    dropdownComponents: (
+      <StyledDropdownMenu>
+        <DropdownMenuItemsContainer hasMaxHeight>
+          <FakeSelectableMenuItemList hasAvatar />
+        </DropdownMenuItemsContainer>
+      </StyledDropdownMenu>
     ),
   },
+  play: playInteraction,
 };
 
 export const CheckableMenuItemWithAvatar: Story = {
   decorators: [WithContentBelowDecorator],
   args: {
-    children: (
-      <DropdownMenuItemsContainer hasMaxHeight>
-        <FakeCheckableMenuItemList hasAvatar />
-      </DropdownMenuItemsContainer>
+    dropdownComponents: (
+      <StyledDropdownMenu>
+        <DropdownMenuItemsContainer hasMaxHeight>
+          <FakeCheckableMenuItemList hasAvatar />
+        </DropdownMenuItemsContainer>
+      </StyledDropdownMenu>
     ),
   },
+  play: playInteraction,
 };