[Snapshot restore] Add support for feature_states

x-pack/plugins/snapshot_restore/__jest__/client_integration/helpers/constant.ts

@@ -15,4 +15,8 @@ export const POLICY_NAME = 'my-test-policy';
 
 export const SNAPSHOT_NAME = 'my-test-snapshot';
 
-export const POLICY_EDIT = getPolicy({ name: POLICY_NAME, retention: { minCount: 1 } });
+export const POLICY_EDIT = getPolicy({
+  name: POLICY_NAME,
+  retention: { minCount: 1 },
+  config: { includeGlobalState: true, featureStates: ['kibana'] },
+});

x-pack/plugins/snapshot_restore/__jest__/client_integration/helpers/http_requests.ts

@@ -90,6 +90,11 @@ const registerHttpRequestMockHelpers = (
     error?: ResponseError
   ) => mockResponse('GET', `${API_BASE_PATH}policies/indices`, response, error);
 
+  const setLoadPoliciesResponse = (
+    response: HttpResponse = { indices: [] },
+    error?: ResponseError
+  ) => mockResponse('GET', `${API_BASE_PATH}policies`, response, error);
+
   const setAddPolicyResponse = (response?: HttpResponse, error?: ResponseError) =>
     mockResponse('POST', `${API_BASE_PATH}policies`, response, error);
 
@@ -119,6 +124,9 @@ const registerHttpRequestMockHelpers = (
       error
     );
 
+  const setLoadFeaturesResponse = (response: HttpResponse = [], error?: ResponseError) =>
+    mockResponse('GET', `${API_BASE_PATH}policies/features`, response, error);
+
   return {
     setLoadRepositoriesResponse,
     setLoadRepositoryTypesResponse,
@@ -131,6 +139,8 @@ const registerHttpRequestMockHelpers = (
     setGetPolicyResponse,
     setCleanupRepositoryResponse,
     setRestoreSnapshotResponse,
+    setLoadFeaturesResponse,
+    setLoadPoliciesResponse,
   };
 };
 

x-pack/plugins/snapshot_restore/__jest__/client_integration/helpers/policy_form.helpers.ts

@@ -40,6 +40,11 @@ export const formSetup = async (
 export type PolicyFormTestSubjects =
   | 'advancedCronInput'
   | 'allIndicesToggle'
+  | 'globalStateToggle'
+  | 'featureStatesDropdown'
+  | 'toggleIncludeNone'
+  | 'noFeatureStatesCallout'
+  | 'featureStatesToggle'
   | 'backButton'
   | 'deselectIndicesLink'
   | 'allDataStreamsToggle'

x-pack/plugins/snapshot_restore/__jest__/client_integration/helpers/policy_list.helpers.ts

@@ -0,0 +1,59 @@
+/*
+ * 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 { act } from 'react-dom/test-utils';
+import {
+  registerTestBed,
+  AsyncTestBedConfig,
+  TestBed,
+  findTestSubject,
+} from '@kbn/test-jest-helpers';
+import { HttpSetup } from '@kbn/core/public';
+import { PolicyList } from '../../../public/application/sections/home/policy_list';
+import { WithAppDependencies } from './setup_environment';
+
+const testBedConfig: AsyncTestBedConfig = {
+  memoryRouter: {
+    initialEntries: ['/policies'],
+    componentRoutePath: '/policies/:policyName?',
+  },
+  doMountAsync: true,
+};
+
+const createActions = (testBed: TestBed) => {
+  const clickPolicyAt = async (index: number) => {
+    const { component, table, router } = testBed;
+    const { rows } = table.getMetaData('policyTable');
+    const repositoryLink = findTestSubject(rows[index].reactWrapper, 'policyLink');
+
+    await act(async () => {
+      const { href } = repositoryLink.props();
+      router.navigateTo(href!);
+    });
+
+    component.update();
+  };
+
+  return {
+    clickPolicyAt,
+  };
+};
+
+export type PoliciesListTestBed = TestBed & {
+  actions: ReturnType<typeof createActions>;
+};
+
+export const setupPoliciesListPage = async (httpSetup: HttpSetup) => {
+  const initTestBed = registerTestBed(WithAppDependencies(PolicyList, httpSetup), testBedConfig);
+
+  const testBed = await initTestBed();
+
+  return {
+    ...testBed,
+    actions: createActions(testBed),
+  };
+};

x-pack/plugins/snapshot_restore/__jest__/client_integration/helpers/restore_snapshot.helpers.ts

@@ -51,6 +51,14 @@ const setupActions = (testBed: TestBed<RestoreSnapshotFormTestSubject>) => {
       component.update();
     },
 
+    async toggleFeatureState() {
+      await act(async () => {
+        form.toggleEuiSwitch('includeFeatureStatesSwitch');
+      });
+
+      component.update();
+    },
+
     toggleIncludeAliases() {
       act(() => {
         form.toggleEuiSwitch('includeAliasesSwitch');
@@ -99,9 +107,13 @@ export type RestoreSnapshotFormTestSubject =
   | 'snapshotRestoreStepLogistics'
   | 'includeGlobalStateSwitch'
   | 'includeAliasesSwitch'
+  | 'featureStatesDropdown'
+  | 'includeFeatureStatesSwitch'
+  | 'toggleIncludeNone'
   | 'nextButton'
   | 'restoreButton'
   | 'systemIndicesInfoCallOut'
+  | 'noFeatureStatesCallout'
   | 'dataStreamWarningCallOut'
   | 'restoreSnapshotsForm.backButton'
   | 'restoreSnapshotsForm.nextButton'

x-pack/plugins/snapshot_restore/__jest__/client_integration/helpers/setup_environment.tsx

@@ -7,6 +7,7 @@
 
 import React from 'react';
 import { i18n } from '@kbn/i18n';
+import { merge } from 'lodash';
 import { LocationDescriptorObject } from 'history';
 
 import { HttpSetup } from '@kbn/core/public';
@@ -16,16 +17,30 @@ import {
   breadcrumbService,
   docTitleService,
 } from '../../../public/application/services/navigation';
+import {
+  AuthorizationContext,
+  Authorization,
+  Privileges,
+  GlobalFlyout,
+} from '../../../public/shared_imports';
 import { AppContextProvider } from '../../../public/application/app_context';
 import { textService } from '../../../public/application/services/text';
 import { init as initHttpRequests } from './http_requests';
 import { UiMetricService } from '../../../public/application/services';
 
+const { GlobalFlyoutProvider } = GlobalFlyout;
 const history = scopedHistoryMock.create();
 history.createHref.mockImplementation((location: LocationDescriptorObject) => {
   return `${location.pathname}?${location.search}`;
 });
 
+const createAuthorizationContextValue = (privileges: Privileges) => {
+  return {
+    isLoading: false,
+    privileges: privileges ?? { hasAllPrivileges: false, missingPrivileges: {} },
+  } as Authorization;
+};
+
 export const services = {
   uiMetricService: new UiMetricService('snapshot_restore'),
   httpService,
@@ -60,16 +75,24 @@ export const setupEnvironment = () => {
   this.terminate = () => {};
 };
 
-export const WithAppDependencies = (Comp: any, httpSetup?: HttpSetup) => (props: any) => {
-  // We need to optionally setup the httpService since some cit helpers (such as snapshot_list.helpers)
-  // use jest mocks to stub the fetch hooks instead of mocking api responses.
-  if (httpSetup) {
-    httpService.setup(httpSetup);
-  }
+export const WithAppDependencies =
+  (Comp: any, httpSetup?: HttpSetup, { privileges, ...overrides }: Record<string, unknown> = {}) =>
+  (props: any) => {
+    // We need to optionally setup the httpService since some cit helpers (such as snapshot_list.helpers)
+    // use jest mocks to stub the fetch hooks instead of mocking api responses.
+    if (httpSetup) {
+      httpService.setup(httpSetup);
+    }
 
-  return (
-    <AppContextProvider value={appDependencies as any}>
-      <Comp {...props} />
-    </AppContextProvider>
-  );
-};
+    return (
+      <AuthorizationContext.Provider
+        value={createAuthorizationContextValue(privileges as Privileges)}
+      >
+        <AppContextProvider value={merge(appDependencies, overrides) as any}>
+          <GlobalFlyoutProvider>
+            <Comp {...props} />
+          </GlobalFlyoutProvider>
+        </AppContextProvider>
+      </AuthorizationContext.Provider>
+    );
+  };

x-pack/plugins/snapshot_restore/__jest__/client_integration/home.test.ts

@@ -497,10 +497,12 @@ describe('<SnapshotRestoreHome />', () => {
       const snapshot1 = fixtures.getSnapshot({
         repository: REPOSITORY_NAME,
         snapshot: `a${getRandomString()}`,
+        featureStates: ['kibana'],
       });
       const snapshot2 = fixtures.getSnapshot({
         repository: REPOSITORY_NAME,
         snapshot: `b${getRandomString()}`,
+        includeGlobalState: false,
       });
       const snapshots = [snapshot1, snapshot2];
 
@@ -709,6 +711,30 @@ describe('<SnapshotRestoreHome />', () => {
             expect(exists('snapshotDetail')).toBe(false);
           });
 
+          test('should show feature states if include global state is enabled', async () => {
+            const { find } = testBed;
+
+            // Assert against first snapshot shown in the table, which should have includeGlobalState and a featureState
+            expect(find('includeGlobalState.value').text()).toEqual('Yes');
+            expect(find('snapshotFeatureStatesSummary.featureStatesList').text()).toEqual('kibana');
+
+            // Close the flyout
+            find('snapshotDetail.closeButton').simulate('click');
+
+            // Replace the get snapshot details api call with the payload of the second snapshot which we're about to click
+            httpRequestsMockHelpers.setGetSnapshotResponse(
+              snapshot2.repository,
+              snapshot2.snapshot,
+              snapshot2
+            );
+
+            // Now we will assert against the second result of the table which shouldnt have includeGlobalState or a featureState
+            await testBed.actions.clickSnapshotAt(1);
+
+            expect(find('includeGlobalState.value').text()).toEqual('No');
+            expect(find('snapshotFeatureStatesSummary.value').text()).toEqual('No');
+          });
+
           describe('tabs', () => {
             test('should have 2 tabs', () => {
               const { find } = testBed;
@@ -738,7 +764,10 @@ describe('<SnapshotRestoreHome />', () => {
                 );
                 expect(find('snapshotDetail.uuid.value').text()).toBe(uuid);
                 expect(find('snapshotDetail.state.value').text()).toBe('Snapshot complete');
-                expect(find('snapshotDetail.includeGlobalState.value').text()).toBe('Yes');
+                expect(find('snapshotDetail.includeGlobalState.value').text()).toEqual('Yes');
+                expect(
+                  find('snapshotDetail.snapshotFeatureStatesSummary.featureStatesList').text()
+                ).toEqual('kibana');
                 expect(find('snapshotDetail.indices.title').text()).toBe(
                   `Indices (${indices.length})`
                 );