fix tests

elastic · Aug 29, 2024 · 1fec585 · 1fec585
1 parent 8451059
commit 1fec585
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 7 deletions.
diff --git a/...ck/plugins/fleet/public/applications/fleet/sections/agents/agent_list_page/index.test.tsx b/...ck/plugins/fleet/public/applications/fleet/sections/agents/agent_list_page/index.test.tsx
@@ -47,6 +47,14 @@ jest.mock('../../../hooks', () => ({
   },
   useFleetStatus: jest.fn().mockReturnValue({}),
   sendGetAgentStatus: jest.fn(),
+  sendBulkGetAgentPolicies: jest.fn().mockResolvedValue({
+    data: {
+      items: [
+        { id: 'policy1', is_managed: false },
+        { id: 'managed_policy', is_managed: true },
+      ],
+    },
+  }),
   sendGetAgentPolicies: jest.fn().mockResolvedValue({ data: { items: [] } }),
   sendGetAgentTags: jest.fn().mockReturnValue({ data: { items: ['tag1', 'tag2'] } }),
   useAuthz: jest

diff --git a/x-pack/plugins/fleet/public/components/agent_enrollment_flyout/index.tsx b/x-pack/plugins/fleet/public/components/agent_enrollment_flyout/index.tsx
@@ -31,7 +31,7 @@ import {
   useAuthz,
 } from '../../hooks';
 import { FLEET_SERVER_PACKAGE, MAX_FLYOUT_WIDTH } from '../../constants';
-import type { PackagePolicy, AgentPolicy } from '../../types';
+import type { PackagePolicy } from '../../types';
 
 import { Loading } from '..';
 
@@ -57,11 +57,6 @@ export const AgentEnrollmentFlyout: React.FunctionComponent<FlyOutProps> = ({
   isIntegrationFlow,
   installedPackagePolicy,
 }) => {
-  const findPolicyById = (policies: AgentPolicy[], id: string | undefined) => {
-    if (!id) return undefined;
-    return policies.find((p) => p.id === id);
-  };
-
   const authz = useAuthz();
 
   const fleetStatus = useFleetStatus();
@@ -87,7 +82,7 @@ export const AgentEnrollmentFlyout: React.FunctionComponent<FlyOutProps> = ({
 
   const selectedPolicy = agentPolicyWithPackagePolicies
     ? agentPolicyWithPackagePolicies
-    : findPolicyById(agentPolicies, selectedPolicyId);
+    : undefined;
 
   const hasNoFleetServerHost = fleetStatus.isReady && !fleetServerHost;