[Security Solution][Endpoint] Add additional tests for Policy Response

x-pack/plugins/fleet/public/applications/fleet/sections/agents/agent_details_page/components/agent_details/agent_details_integrations.tsx

@@ -25,6 +25,8 @@ import { i18n } from '@kbn/i18n';
 import { FormattedMessage } from '@kbn/i18n-react';
 import styled from 'styled-components';
 
+import type { EuiAccordionProps } from '@elastic/eui/src/components/accordion';
+
 import type { Agent, AgentPolicy, PackagePolicy } from '../../../../../types';
 import type { FleetServerAgentComponentUnit } from '../../../../../../../../common/types/models/agent';
 import { useLink, useUIExtension } from '../../../../../hooks';
@@ -69,18 +71,29 @@ const StyledEuiLink = styled(EuiLink)`
   font-size: ${(props) => props.theme.eui.euiFontSizeS};
 `;
 
-const CollapsablePanel: React.FC<{ id: string; title: React.ReactNode }> = ({
-  id,
-  title,
-  children,
-}) => {
+const CollapsablePanel: React.FC<{
+  id: string;
+  title: React.ReactNode;
+  'data-test-subj'?: string;
+}> = ({ id, title, children, 'data-test-subj': dataTestSubj }) => {
+  const arrowProps = useMemo<EuiAccordionProps['arrowProps']>(() => {
+    if (dataTestSubj) {
+      return {
+        'data-test-subj': `${dataTestSubj}-openCloseToggle`,
+      };
+    }
+    return undefined;
+  }, [dataTestSubj]);
+
   return (
     <EuiPanel paddingSize="none">
       <StyledEuiAccordion
         id={id}
         arrowDisplay="left"
         buttonClassName="ingest-integration-title-button"
         buttonContent={title}
+        arrowProps={arrowProps}
+        data-test-subj={dataTestSubj}
       >
         {children}
       </StyledEuiAccordion>
@@ -92,7 +105,8 @@ export const AgentDetailsIntegration: React.FunctionComponent<{
   agent: Agent;
   agentPolicy: AgentPolicy;
   packagePolicy: PackagePolicy;
-}> = memo(({ agent, agentPolicy, packagePolicy }) => {
+  'data-test-subj'?: string;
+}> = memo(({ agent, agentPolicy, packagePolicy, 'data-test-subj': dataTestSubj }) => {
   const { getHref } = useLink();
   const theme = useEuiTheme();
 
@@ -204,6 +218,7 @@ export const AgentDetailsIntegration: React.FunctionComponent<{
   return (
     <CollapsablePanel
       id={packagePolicy.id}
+      data-test-subj={dataTestSubj}
       title={
         <EuiTitle size="xs">
           <h3>
@@ -234,7 +249,12 @@ export const AgentDetailsIntegration: React.FunctionComponent<{
               </EuiFlexItem>
               {showNeedsAttentionBadge && (
                 <EuiFlexItem grow={false}>
-                  <EuiBadge color={theme.euiTheme.colors.danger} iconType="warning" iconSide="left">
+                  <EuiBadge
+                    color={theme.euiTheme.colors.danger}
+                    iconType="warning"
+                    iconSide="left"
+                    data-test-subj={dataTestSubj ? `${dataTestSubj}-needsAttention` : undefined}
+                  >
                     <FormattedMessage
                       id="xpack.fleet.agentDetailsIntegrations.needsAttention.label"
                       defaultMessage="Needs attention"
@@ -270,13 +290,16 @@ export const AgentDetailsIntegrationsSection: React.FunctionComponent<{
 
   return (
     <EuiFlexGroup direction="column" gutterSize="m">
-      {(agentPolicy.package_policies as PackagePolicy[]).map((packagePolicy) => {
+      {(agentPolicy.package_policies as PackagePolicy[]).map((packagePolicy, index) => {
+        const testSubj = (packagePolicy.package?.name ?? 'packagePolicy') + '-' + index;
+
         return (
-          <EuiFlexItem grow={false} key={packagePolicy.id}>
+          <EuiFlexItem grow={false} key={packagePolicy.id} data-test-subj={testSubj}>
             <AgentDetailsIntegration
               agent={agent}
               agentPolicy={agentPolicy}
               packagePolicy={packagePolicy}
+              data-test-subj={`${testSubj}-accordion`}
             />
           </EuiFlexItem>
         );

x-pack/plugins/security_solution/common/endpoint/data_generators/endpoint_metadata_generator.ts

@@ -5,18 +5,102 @@
  * 2.0.
  */
 
+/* eslint-disable max-classes-per-file */
+
 import type { DeepPartial } from 'utility-types';
-import { merge } from 'lodash';
+import { merge, set } from 'lodash';
 import { gte } from 'semver';
 import type { EndpointCapabilities } from '../service/response_actions/constants';
 import { BaseDataGenerator } from './base_data_generator';
 import type { HostMetadataInterface, OSFields } from '../types';
 import { EndpointStatus, HostPolicyResponseActionStatus } from '../types';
 
+export interface GetCustomEndpointMetadataGeneratorOptions {
+  /** Version for agent/endpoint. Defaults to the stack version */
+  version: string;
+  /** OS type for the generated endpoint hosts */
+  os: 'macOS' | 'windows' | 'linux';
+}
+
 /**
  * Metadata generator for docs that are sent by the Endpoint running on hosts
  */
 export class EndpointMetadataGenerator extends BaseDataGenerator {
+  /**
+   * Returns a Custom `EndpointMetadataGenerator` subclass that will generate specific
+   * documents based on input arguments
+   */
+  static custom({
+    version,
+    os,
+  }: Partial<GetCustomEndpointMetadataGeneratorOptions> = {}): typeof EndpointMetadataGenerator {
+    return class extends EndpointMetadataGenerator {
+      generate(overrides: DeepPartial<HostMetadataInterface> = {}): HostMetadataInterface {
+        if (version) {
+          set(overrides, 'agent.version', version);
+        }
+
+        if (os) {
+          switch (os) {
+            case 'linux':
+              set(overrides, 'host.os', EndpointMetadataGenerator.linuxOSFields);
+              break;
+
+            case 'macOS':
+              set(overrides, 'host.os', EndpointMetadataGenerator.macOSFields);
+              break;
+
+            default:
+              set(overrides, 'host.os', EndpointMetadataGenerator.windowsOSFields);
+          }
+        }
+
+        return super.generate(overrides);
+      }
+    };
+  }
+
+  public static get windowsOSFields(): OSFields {
+    return {
+      name: 'Windows',
+      full: 'Windows 10',
+      version: '10.0',
+      platform: 'Windows',
+      family: 'windows',
+      Ext: {
+        variant: 'Windows Pro',
+      },
+    };
+  }
+
+  public static get linuxOSFields(): OSFields {
+    return {
+      Ext: {
+        variant: 'Debian',
+      },
+      kernel: '4.19.0-21-cloud-amd64 #1 SMP Debian 4.19.249-2 (2022-06-30)',
+      name: 'Linux',
+      family: 'debian',
+      type: 'linux',
+      version: '10.12',
+      platform: 'debian',
+      full: 'Debian 10.12',
+    };
+  }
+
+  public static get macOSFields(): OSFields {
+    return {
+      name: 'macOS',
+      full: 'macOS Monterey',
+      version: '12.6.1',
+      platform: 'macOS',
+      family: 'Darwin',
+      Ext: {
+        variant: 'Darwin',
+      },
+    };
+  }
+
   /** Generate an Endpoint host metadata document */
   generate(overrides: DeepPartial<HostMetadataInterface> = {}): HostMetadataInterface {
     const ts = overrides['@timestamp'] ?? new Date().getTime();
@@ -102,16 +186,7 @@ export class EndpointMetadataGenerator extends BaseDataGenerator {
 
   protected randomOsFields(): OSFields {
     return this.randomChoice([
-      {
-        name: 'Windows',
-        full: 'Windows 10',
-        version: '10.0',
-        platform: 'Windows',
-        family: 'windows',
-        Ext: {
-          variant: 'Windows Pro',
-        },
-      },
+      EndpointMetadataGenerator.windowsOSFields,
       {
         name: 'Windows',
         full: 'Windows Server 2016',
@@ -142,18 +217,8 @@ export class EndpointMetadataGenerator extends BaseDataGenerator {
           variant: 'Windows Server Release 2',
         },
       },
-      {
-        Ext: {
-          variant: 'Debian',
-        },
-        kernel: '4.19.0-21-cloud-amd64 #1 SMP Debian 4.19.249-2 (2022-06-30)',
-        name: 'Linux',
-        family: 'debian',
-        type: 'linux',
-        version: '10.12',
-        platform: 'debian',
-        full: 'Debian 10.12',
-      },
+      EndpointMetadataGenerator.linuxOSFields,
+      EndpointMetadataGenerator.macOSFields,
     ]);
   }
 }