[ILM] Add searchable snapshot field

x-pack/plugins/index_lifecycle_management/__jest__/client_integration/app/app.helpers.tsx

@@ -7,16 +7,17 @@
 import React from 'react';
 import { act } from 'react-dom/test-utils';
 import { registerTestBed, TestBed, TestBedConfig } from '@kbn/test/jest';
+import { KibanaContextProvider } from '../../../../../../src/plugins/kibana_react/public/context';
+import { createBreadcrumbsMock } from '../../../public/application/services/breadcrumbs.mock';
+import { licensingMock } from '../../../../licensing/public/mocks';
 import { App } from '../../../public/application/app';
 import { TestSubjects } from '../helpers';
-import { createBreadcrumbsMock } from '../../../public/application/services/breadcrumbs.mock';
-import { KibanaContextProvider } from '../../../../../../src/plugins/kibana_react/public/context';
 
 const breadcrumbService = createBreadcrumbsMock();
 
 const AppWithContext = (props: any) => {
   return (
-    <KibanaContextProvider services={{ breadcrumbService }}>
+    <KibanaContextProvider services={{ breadcrumbService, license: licensingMock.createLicense() }}>
       <App {...props} />
     </KibanaContextProvider>
   );

x-pack/plugins/index_lifecycle_management/__jest__/client_integration/edit_policy/edit_policy.helpers.tsx

@@ -9,12 +9,15 @@ import { act } from 'react-dom/test-utils';
 
 import { registerTestBed, TestBedConfig } from '@kbn/test/jest';
 
+import { licensingMock } from '../../../../licensing/public/mocks';
+
 import { EditPolicy } from '../../../public/application/sections/edit_policy';
 import { DataTierAllocationType } from '../../../public/application/sections/edit_policy/types';
 
 import { Phases as PolicyPhases } from '../../../common/types';
 
 import { KibanaContextProvider } from '../../../public/shared_imports';
+import { AppServicesContext } from '../../../public/types';
 import { createBreadcrumbsMock } from '../../../public/application/services/breadcrumbs.mock';
 
 type Phases = keyof PolicyPhases;
@@ -53,10 +56,16 @@ const testBedConfig: TestBedConfig = {
 
 const breadcrumbService = createBreadcrumbsMock();
 
-const MyComponent = (props: any) => {
+const MyComponent = ({ appServicesContext, ...rest }: any) => {
   return (
-    <KibanaContextProvider services={{ breadcrumbService }}>
-      <EditPolicy {...props} />
+    <KibanaContextProvider
+      services={{
+        breadcrumbService,
+        license: licensingMock.createLicense({ license: { type: 'enterprise' } }),
+        ...appServicesContext,
+      }}
+    >
+      <EditPolicy {...rest} />
     </KibanaContextProvider>
   );
 };
@@ -67,10 +76,10 @@ type SetupReturn = ReturnType<typeof setup>;
 
 export type EditPolicyTestBed = SetupReturn extends Promise<infer U> ? U : SetupReturn;
 
-export const setup = async () => {
-  const testBed = await initTestBed();
+export const setup = async (arg?: { appServicesContext: Partial<AppServicesContext> }) => {
+  const testBed = await initTestBed(arg);
 
-  const { find, component, form } = testBed;
+  const { find, component, form, exists } = testBed;
 
   const createFormToggleAction = (dataTestSubject: string) => async (checked: boolean) => {
     await act(async () => {
@@ -128,12 +137,15 @@ export const setup = async () => {
     component.update();
   };
 
-  const toggleForceMerge = (phase: Phases) => createFormToggleAction(`${phase}-forceMergeSwitch`);
-
-  const setForcemergeSegmentsCount = (phase: Phases) =>
-    createFormSetValueAction(`${phase}-selectedForceMergeSegments`);
-
-  const setBestCompression = (phase: Phases) => createFormToggleAction(`${phase}-bestCompression`);
+  const createForceMergeActions = (phase: Phases) => {
+    const toggleSelector = `${phase}-forceMergeSwitch`;
+    return {
+      forceMergeFieldExists: () => exists(toggleSelector),
+      toggleForceMerge: createFormToggleAction(toggleSelector),
+      setForcemergeSegmentsCount: createFormSetValueAction(`${phase}-selectedForceMergeSegments`),
+      setBestCompression: createFormToggleAction(`${phase}-bestCompression`),
+    };
+  };
 
   const setIndexPriority = (phase: Phases) =>
     createFormSetValueAction(`${phase}-phaseIndexPriority`);
@@ -180,7 +192,35 @@ export const setup = async () => {
     await createFormSetValueAction('warm-selectedPrimaryShardCount')(value);
   };
 
+  const shrinkExists = () => exists('shrinkSwitch');
+
   const setFreeze = createFormToggleAction('freezeSwitch');
+  const freezeExists = () => exists('freezeSwitch');
+
+  const createSearchableSnapshotActions = (phase: Phases) => {
+    const fieldSelector = `searchableSnapshotField-${phase}`;
+    const licenseCalloutSelector = `${fieldSelector}.searchableSnapshotDisabledDueToLicense`;
+    const toggleSelector = `${fieldSelector}.searchableSnapshotToggle`;
+
+    const toggleSearchableSnapshot = createFormToggleAction(toggleSelector);
+    return {
+      searchableSnapshotDisabled: () => exists(licenseCalloutSelector),
+      searchableSnapshotsExists: () => exists(fieldSelector),
+      findSearchableSnapshotToggle: () => find(toggleSelector),
+      searchableSnapshotDisabledDueToLicense: () =>
+        exists(`${fieldSelector}.searchableSnapshotDisabledDueToLicense`),
+      toggleSearchableSnapshot,
+      setSearchableSnapshot: async (value: string) => {
+        await toggleSearchableSnapshot(true);
+        act(() => {
+          find(`searchableSnapshotField-${phase}.searchableSnapshotCombobox`).simulate('change', [
+            { label: value },
+          ]);
+        });
+        component.update();
+      },
+    };
+  };
 
   return {
     ...testBed,
@@ -192,10 +232,9 @@ export const setup = async () => {
         setMaxDocs,
         setMaxAge,
         toggleRollover,
-        toggleForceMerge: toggleForceMerge('hot'),
-        setForcemergeSegments: setForcemergeSegmentsCount('hot'),
-        setBestCompression: setBestCompression('hot'),
+        ...createForceMergeActions('hot'),
         setIndexPriority: setIndexPriority('hot'),
+        ...createSearchableSnapshotActions('hot'),
       },
       warm: {
         enable: enable('warm'),
@@ -206,9 +245,8 @@ export const setup = async () => {
         setSelectedNodeAttribute: setSelectedNodeAttribute('warm'),
         setReplicas: setReplicas('warm'),
         setShrink,
-        toggleForceMerge: toggleForceMerge('warm'),
-        setForcemergeSegments: setForcemergeSegmentsCount('warm'),
-        setBestCompression: setBestCompression('warm'),
+        shrinkExists,
+        ...createForceMergeActions('warm'),
         setIndexPriority: setIndexPriority('warm'),
       },
       cold: {
@@ -219,7 +257,9 @@ export const setup = async () => {
         setSelectedNodeAttribute: setSelectedNodeAttribute('cold'),
         setReplicas: setReplicas('cold'),
         setFreeze,
+        freezeExists,
         setIndexPriority: setIndexPriority('cold'),
+        ...createSearchableSnapshotActions('cold'),
       },
       delete: {
         enable: enable('delete'),