[Fleet] Fix add agent in the package policy table (#104749) (#105690)

Co-authored-by: Nicolas Chaulet <nicolas.chaulet@elastic.co>
elastic · Jul 14, 2021 · 1696afe · 1696afe
1 parent 7f46afe
commit 1696afe
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/x-pack/plugins/fleet/public/components/package_policy_actions_menu.tsx b/x-pack/plugins/fleet/public/components/package_policy_actions_menu.tsx
@@ -28,6 +28,7 @@ export const PackagePolicyActionsMenu: React.FunctionComponent<{
   const { getHref } = useLink();
   const hasWriteCapabilities = useCapabilities().write;
   const refreshAgentPolicy = useAgentPolicyRefresh();
+  const [isActionsMenuOpen, setIsActionsMenuOpen] = useState(false);
 
   const onEnrollmentFlyoutClose = useMemo(() => {
     return () => setIsEnrollmentFlyoutOpen(false);
@@ -48,7 +49,10 @@ export const PackagePolicyActionsMenu: React.FunctionComponent<{
     // </EuiContextMenuItem>,
     <EuiContextMenuItem
       icon="plusInCircle"
-      onClick={() => setIsEnrollmentFlyoutOpen(true)}
+      onClick={() => {
+        setIsActionsMenuOpen(false);
+        setIsEnrollmentFlyoutOpen(true);
+      }}
       key="addAgent"
     >
       <FormattedMessage
@@ -112,7 +116,11 @@ export const PackagePolicyActionsMenu: React.FunctionComponent<{
           />
         </EuiPortal>
       )}
-      <ContextMenuActions items={menuItems} />
+      <ContextMenuActions
+        isOpen={isActionsMenuOpen}
+        items={menuItems}
+        onChange={(isOpen) => setIsActionsMenuOpen(isOpen)}
+      />
     </>
   );
 };