From ecffbd4f77426d0f4cbd161e3fcbac984d87c938 Mon Sep 17 00:00:00 2001
From: Dale Fukami <dalefukami@users.noreply.github.com>
Date: Thu, 9 Nov 2023 10:01:43 -0700
Subject: [PATCH] Wrap the dropdown in a Portal (#153)

---
 .../DropdownMenu/DropdownMenu.test.tsx        |  7 +++-
 src/components/DropdownMenu/DropdownMenu.tsx  | 39 ++++++++++---------
 .../SelectInput/SelectInput.test.tsx          |  3 ++
 3 files changed, 29 insertions(+), 20 deletions(-)

diff --git a/src/components/DropdownMenu/DropdownMenu.test.tsx b/src/components/DropdownMenu/DropdownMenu.test.tsx
index 6a1ff83..539b0d5 100644
--- a/src/components/DropdownMenu/DropdownMenu.test.tsx
+++ b/src/components/DropdownMenu/DropdownMenu.test.tsx
@@ -55,6 +55,9 @@ jest.mock('@radix-ui/react-dropdown-menu', () => ({
   Arrow: (props: DropdownMenuArrowProps) => {
     mockRadixArrow(props);
     return <div data-testid="arrow">{props.children}</div>;
+  },
+  DropdownMenuPortal: (props: any) => {
+    return <div data-testid="portal">{props.children}</div>;
   }
 }));
 
@@ -70,11 +73,11 @@ describe('<DropdownMenu />', () => {
     const root = container.firstChild;
     const trigger = screen.getByTestId('trigger');
     const content = screen.getByTestId('content');
+    const portal = screen.getByTestId('portal');
 
-    // Check that trigger and content are children of root
     expect(root).toBe(screen.getByTestId('root'));
     expect(trigger.parentElement).toBe(root);
-    expect(content.parentElement).toBe(root);
+    expect(content.parentElement).toBe(portal);
 
     // Check that items are only rendered in the dropdown content
     const items = container.getElementsByClassName('mock-item');
diff --git a/src/components/DropdownMenu/DropdownMenu.tsx b/src/components/DropdownMenu/DropdownMenu.tsx
index 9cdec64..cc9dffc 100644
--- a/src/components/DropdownMenu/DropdownMenu.tsx
+++ b/src/components/DropdownMenu/DropdownMenu.tsx
@@ -7,7 +7,8 @@ import {
   Root as RadixUIDropdownMenuRoot,
   Trigger as RadixUIDropdownMenuTrigger,
   Content as RadixUIDropdownMenuContent,
-  Item as RadixUIDropdownMenuItem
+  Item as RadixUIDropdownMenuItem,
+  DropdownMenuPortal as RadixDropdownMenuPortal
 } from '@radix-ui/react-dropdown-menu';
 
 import './DropdownMenu.scss';
@@ -70,23 +71,25 @@ export const DropdownMenu = forwardRef<HTMLDivElement, DropdownMenuProps>(
         <RadixUIDropdownMenuTrigger className={classNames(className, 'zui-dropdown-trigger')}>
           {triggerElement}
         </RadixUIDropdownMenuTrigger>
-        <RadixUIDropdownMenuContent
-          className={classNames(menuClassName, 'zui-dropdown-menu')}
-          align={alignMenu}
-          side={side}
-          ref={ref}
-        >
-          {alignMenu === 'center' && showArrow && <RadixUIDropdownMenuArrow className="zui-dropdown-arrow" />}
-          {items.map(item => (
-            <RadixUIDropdownMenuItem
-              className={classNames('zui-dropdown-item', item.className)}
-              key={item.id}
-              onSelect={(event: Event) => item.onSelect(event)}
-            >
-              {item.label}
-            </RadixUIDropdownMenuItem>
-          ))}
-        </RadixUIDropdownMenuContent>
+        <RadixDropdownMenuPortal>
+          <RadixUIDropdownMenuContent
+            className={classNames(menuClassName, 'zui-dropdown-menu')}
+            align={alignMenu}
+            side={side}
+            ref={ref}
+          >
+            {alignMenu === 'center' && showArrow && <RadixUIDropdownMenuArrow className="zui-dropdown-arrow" />}
+            {items.map(item => (
+              <RadixUIDropdownMenuItem
+                className={classNames('zui-dropdown-item', item.className)}
+                key={item.id}
+                onSelect={(event: Event) => item.onSelect(event)}
+              >
+                {item.label}
+              </RadixUIDropdownMenuItem>
+            ))}
+          </RadixUIDropdownMenuContent>
+        </RadixDropdownMenuPortal>
       </RadixUIDropdownMenuRoot>
     );
   }
diff --git a/src/components/SelectInput/SelectInput.test.tsx b/src/components/SelectInput/SelectInput.test.tsx
index 92a84f7..f67c435 100644
--- a/src/components/SelectInput/SelectInput.test.tsx
+++ b/src/components/SelectInput/SelectInput.test.tsx
@@ -46,6 +46,9 @@ jest.mock('@radix-ui/react-dropdown-menu', () => ({
   Item: (props: DropdownMenuItemProps) => {
     mockRadixItem(props);
     return <div className={'mock-item ' + props.className}>{props.children}</div>;
+  },
+  DropdownMenuPortal: (props: any) => {
+    return <div data-testid="portal">{props.children}</div>;
   }
 }));