chore: Expanded mode basic test scenario

Author: georgylobko

src/app-layout/__tests__/runtime-drawers.test.tsx

@@ -1385,51 +1385,64 @@ describe('toolbar mode only features', () => {
     });
 
     describe('expanded mode for global drawers', () => {
-      test('should set a drawer to expanded mode by clicking on "expanded mode" button', async () => {
-        const drawerId = 'global-drawer';
-        awsuiPlugins.appLayout.registerDrawer({
-          ...drawerDefaults,
-          ariaLabels: {
-            expandedModeButton: 'Expanded mode button',
-          },
-          id: drawerId,
-          type: 'global',
-          isExpandable: true,
-        });
-        const { wrapper, globalDrawersWrapper } = await renderComponent(<AppLayout />);
+      describe.each(['global', 'global-ai'] as const)('drawer type = %s', type => {
+        const findDrawerTriggerById = (id: string, renderProps: Awaited<ReturnType<typeof renderComponent>>) => {
+          if (type === 'global') {
+            return renderProps.wrapper.findDrawerTriggerById(id);
+          } else {
+            return renderProps.globalDrawersWrapper.findAiDrawerTrigger();
+          }
+        };
 
-        wrapper.findDrawerTriggerById(drawerId)!.click();
-        expect(globalDrawersWrapper.findExpandedModeButtonByActiveDrawerId(drawerId)!.getElement()).toBeInTheDocument();
-        expect(globalDrawersWrapper.findDrawerById(drawerId)!.isDrawerInExpandedMode()).toBe(false);
-        globalDrawersWrapper.findExpandedModeButtonByActiveDrawerId(drawerId)!.click();
-        expect(globalDrawersWrapper.findDrawerById(drawerId)!.isDrawerInExpandedMode()).toBe(true);
-        expect(
-          getGeneratedAnalyticsMetadata(
-            globalDrawersWrapper.findExpandedModeButtonByActiveDrawerId(drawerId)!.getElement()
-          )
-        ).toEqual(
-          expect.objectContaining({
-            action: 'expand',
-            detail: {
-              label: 'Expanded mode button',
+        test('should set a drawer to expanded mode by clicking on "expanded mode" button', async () => {
+          const drawerId = 'global-drawer';
+          awsuiPlugins.appLayout.registerDrawer({
+            ...drawerDefaults,
+            ariaLabels: {
+              expandedModeButton: 'Expanded mode button',
             },
-          })
-        );
+            id: drawerId,
+            type,
+            isExpandable: true,
+          });
+          const renderProps = await renderComponent(<AppLayout />);
+          const { globalDrawersWrapper } = renderProps;
 
-        globalDrawersWrapper.findExpandedModeButtonByActiveDrawerId(drawerId)!.click();
-        expect(globalDrawersWrapper.findDrawerById(drawerId)!.isDrawerInExpandedMode()).toBe(false);
-        expect(
-          getGeneratedAnalyticsMetadata(
+          findDrawerTriggerById(drawerId, renderProps)!.click();
+          expect(
             globalDrawersWrapper.findExpandedModeButtonByActiveDrawerId(drawerId)!.getElement()
-          )
-        ).toEqual(
-          expect.objectContaining({
-            action: 'collapse',
-            detail: {
-              label: 'Expanded mode button',
-            },
-          })
-        );
+          ).toBeInTheDocument();
+          expect(globalDrawersWrapper.findDrawerById(drawerId)!.isDrawerInExpandedMode()).toBe(false);
+          globalDrawersWrapper.findExpandedModeButtonByActiveDrawerId(drawerId)!.click();
+          expect(globalDrawersWrapper.findDrawerById(drawerId)!.isDrawerInExpandedMode()).toBe(true);
+          expect(
+            getGeneratedAnalyticsMetadata(
+              globalDrawersWrapper.findExpandedModeButtonByActiveDrawerId(drawerId)!.getElement()
+            )
+          ).toEqual(
+            expect.objectContaining({
+              action: 'expand',
+              detail: {
+                label: 'Expanded mode button',
+              },
+            })
+          );
+
+          globalDrawersWrapper.findExpandedModeButtonByActiveDrawerId(drawerId)!.click();
+          expect(globalDrawersWrapper.findDrawerById(drawerId)!.isDrawerInExpandedMode()).toBe(false);
+          expect(
+            getGeneratedAnalyticsMetadata(
+              globalDrawersWrapper.findExpandedModeButtonByActiveDrawerId(drawerId)!.getElement()
+            )
+          ).toEqual(
+            expect.objectContaining({
+              action: 'collapse',
+              detail: {
+                label: 'Expanded mode button',
+              },
+            })
+          );
+        });
       });
 
       test('only one drawer could be in expanded mode. all other panels should be closed', async () => {