[Onboarding][Test] Added UI Unit Test for new Onboarding fields (#142276) (#142691)

* added ui unit test for onboarding

* added 1 more unit test

* [CI] Auto-commit changed files from 'node scripts/precommit_hook.js --ref HEAD~1..HEAD --fix'

* check fail fix

* [CI] Auto-commit changed files from 'node scripts/precommit_hook.js --ref HEAD~1..HEAD --fix'

* failed check fix again

* forgot to add 1 import type

* [CI] Auto-commit changed files from 'node scripts/precommit_hook.js --ref HEAD~1..HEAD --fix'

* PR comments

* PR Comments, added more test

Co-authored-by: kibanamachine <42973632+kibanamachine@users.noreply.github.com>
(cherry picked from commit 98f019a3ac)

Co-authored-by: Rickyanto Ang <rickyangwyn@gmail.com>

x-pack/plugins/security_solution/public/management/pages/policy/view/ingest_manager_integration/endpoint_policy_create_extension/endpoint_policy_create_extension.test.tsx

@@ -0,0 +1,168 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License
+ * 2.0; you may not use this file except in compliance with the Elastic License
+ * 2.0.
+ */
+
+import React from 'react';
+import userEvent from '@testing-library/user-event';
+import { screen } from '@testing-library/react';
+import { EndpointPolicyCreateExtension } from './endpoint_policy_create_extension';
+import type {
+  NewPackagePolicy,
+  NewPackagePolicyInput,
+  NewPackagePolicyInputStream,
+} from '@kbn/fleet-plugin/common';
+import type { AppContextTestRender } from '../../../../../../common/mock/endpoint';
+import { createAppRootMockRenderer } from '../../../../../../common/mock/endpoint';
+import { licenseService } from '../../../../../../common/hooks/use_license';
+
+jest.mock('../../../../../../common/lib/kibana');
+jest.mock('../../../../../../common/hooks/use_license', () => {
+  const licenseServiceInstance = {
+    isPlatinumPlus: jest.fn(),
+    isGoldPlus: jest.fn(),
+    isEnterprise: jest.fn(() => true),
+  };
+  return {
+    licenseService: licenseServiceInstance,
+    useLicense: () => {
+      return licenseServiceInstance;
+    },
+  };
+});
+
+const getMockNewPackage = (): NewPackagePolicy => {
+  const mockNewPackagePolicyInputStream: NewPackagePolicyInputStream = {
+    enabled: true,
+    data_stream: {
+      dataset: 'someDataset',
+      type: 'someType',
+    },
+  };
+
+  const mockNewPackagePolicyInput: NewPackagePolicyInput = {
+    type: 'someType',
+    enabled: true,
+    streams: [mockNewPackagePolicyInputStream],
+  };
+
+  const mockNewPackage: NewPackagePolicy = {
+    id: 'someid',
+    inputs: [mockNewPackagePolicyInput],
+    name: 'someName',
+    namespace: 'someNamespace',
+    enabled: true,
+    policy_id: 'somePolicyid',
+  };
+  return mockNewPackage;
+};
+
+describe('Onboarding Component new section', () => {
+  let render: () => ReturnType<AppContextTestRender['render']>;
+  let renderResult: ReturnType<typeof render>;
+  let mockedContext: AppContextTestRender;
+
+  beforeEach(() => {
+    mockedContext = createAppRootMockRenderer();
+  });
+
+  describe('When EndpointPolicyCreateExtension is mounted', () => {
+    it('renders EndpointPolicyCreateExtension options correctly (Default to Endpoint)', async () => {
+      renderResult = mockedContext.render(
+        <EndpointPolicyCreateExtension newPolicy={getMockNewPackage()} onChange={jest.fn()} />
+      );
+      expect(renderResult.getByTestId('selectIntegrationTypeId')).toBeVisible();
+      expect(renderResult.queryByText('Next-Generation Antivirus (NGAV)')).toBeVisible();
+      expect(
+        renderResult.queryByText('Essential EDR (Endpoint Detection & Response)')
+      ).toBeVisible();
+      expect(
+        renderResult.queryByText('Complete EDR (Endpoint Detection & Response)')
+      ).toBeVisible();
+    });
+
+    it('renders EndpointPolicyCreateExtension options correctly (set to Cloud)', async () => {
+      renderResult = mockedContext.render(
+        <EndpointPolicyCreateExtension newPolicy={getMockNewPackage()} onChange={jest.fn()} />
+      );
+      userEvent.selectOptions(screen.getByTestId('selectIntegrationTypeId'), ['cloud']);
+      expect(renderResult.getByText('Interactive only')).toBeVisible();
+      expect(renderResult.getByText('All events')).toBeVisible();
+    });
+
+    it('make sure onChange is called when user change environment', async () => {
+      const mockedOnChange = jest.fn();
+      renderResult = mockedContext.render(
+        <EndpointPolicyCreateExtension newPolicy={getMockNewPackage()} onChange={mockedOnChange} />
+      );
+      expect(mockedOnChange).toHaveBeenCalledTimes(1);
+      userEvent.selectOptions(screen.getByTestId('selectIntegrationTypeId'), ['cloud']);
+      expect(mockedOnChange).toHaveBeenCalledTimes(2);
+    });
+
+    it('make sure NGAV is the default value for endpoint environment', async () => {
+      renderResult = mockedContext.render(
+        <EndpointPolicyCreateExtension newPolicy={getMockNewPackage()} onChange={jest.fn()} />
+      );
+      expect(renderResult.getByDisplayValue('NGAV')).toBeChecked();
+      expect(renderResult.getByDisplayValue('EDREssential')).not.toBeChecked();
+      expect(renderResult.getByDisplayValue('EDRComplete')).not.toBeChecked();
+    });
+
+    it('make sure interactive only is the default value for cloud environment', async () => {
+      renderResult = mockedContext.render(
+        <EndpointPolicyCreateExtension newPolicy={getMockNewPackage()} onChange={jest.fn()} />
+      );
+      userEvent.selectOptions(screen.getByTestId('selectIntegrationTypeId'), ['cloud']);
+      expect(renderResult.getByDisplayValue('ALL_EVENTS')).not.toBeChecked();
+      expect(renderResult.getByDisplayValue('INTERACTIVE_ONLY')).toBeChecked();
+    });
+
+    it('all license should be able to see EDR Licese', async () => {
+      const licenseServiceMock = licenseService as jest.Mocked<typeof licenseService>;
+
+      licenseServiceMock.isEnterprise.mockReturnValue(false);
+      renderResult = mockedContext.render(
+        <EndpointPolicyCreateExtension newPolicy={getMockNewPackage()} onChange={jest.fn()} />
+      );
+      userEvent.click(screen.getByDisplayValue('EDREssential'));
+      expect(renderResult.getByDisplayValue('EDREssential')).toBeChecked();
+      expect(
+        renderResult.getByText(
+          'Note: advanced protections require a platinum license, and full response capabilities require an enterprise license.',
+          { exact: false }
+        )
+      ).toBeInTheDocument();
+    });
+
+    it('gold license below users will be able to see NGAV notes', async () => {
+      const licenseServiceMock = licenseService as jest.Mocked<typeof licenseService>;
+
+      licenseServiceMock.isGoldPlus.mockReturnValue(true);
+      renderResult = mockedContext.render(
+        <EndpointPolicyCreateExtension newPolicy={getMockNewPackage()} onChange={jest.fn()} />
+      );
+      expect(
+        renderResult.getByText('Note: advanced protections require a platinum license level.', {
+          exact: false,
+        })
+      ).toBeInTheDocument();
+    });
+
+    it('platinum license users will not be able to see NGAV notes', async () => {
+      const licenseServiceMock = licenseService as jest.Mocked<typeof licenseService>;
+
+      licenseServiceMock.isPlatinumPlus.mockReturnValue(true);
+      renderResult = mockedContext.render(
+        <EndpointPolicyCreateExtension newPolicy={getMockNewPackage()} onChange={jest.fn()} />
+      );
+      expect(
+        renderResult.queryByText('Note: advanced protections require a platinum license level.', {
+          exact: false,
+        })
+      ).not.toBeInTheDocument();
+    });
+  });
+});

x-pack/plugins/security_solution/public/management/pages/policy/view/ingest_manager_integration/endpoint_policy_create_extension/endpoint_policy_create_extension.tsx

@@ -240,6 +240,7 @@ export const EndpointPolicyCreateExtension = memo<PackagePolicyCreateExtensionCo
         >
           <EuiSelect
             id="selectIntegrationTypeId"
+            data-test-subj="selectIntegrationTypeId"
             options={environmentOptions}
             value={selectedEnvironment}
             onChange={onChangeEnvironment}