Add tests for interstitial states.

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

@@ -0,0 +1,50 @@
+/*
+ * 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, TestBed, TestBedConfig } from '@kbn/test/jest';
+
+import { AppWithRouter } from '../../../public/application/app';
+import { WithAppDependencies } from '../helpers';
+
+const testBedConfig: TestBedConfig = {
+  memoryRouter: {
+    initialEntries: [`/overview`],
+    componentRoutePath: '/overview',
+  },
+  doMountAsync: true,
+};
+
+export type AppTestBed = TestBed & {
+  actions: ReturnType<typeof createActions>;
+};
+
+const createActions = (testBed: TestBed) => {
+  const clickDeprecationToggle = async () => {
+    const { find, component } = testBed;
+
+    await act(async () => {
+      find('deprecationLoggingToggle').simulate('click');
+    });
+
+    component.update();
+  };
+
+  return {
+    clickDeprecationToggle,
+  };
+};
+
+export const setupAppPage = async (overrides?: Record<string, unknown>): Promise<AppTestBed> => {
+  const initTestBed = registerTestBed(WithAppDependencies(AppWithRouter, overrides), testBedConfig);
+  const testBed = await initTestBed();
+
+  return {
+    ...testBed,
+    actions: createActions(testBed),
+  };
+};

x-pack/plugins/upgrade_assistant/__jest__/client_integration/app/cluster_upgrade.test.tsx

@@ -0,0 +1,84 @@
+/*
+ * 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 { setupEnvironment } from '../helpers';
+import { AppTestBed, setupAppPage } from './app.helpers';
+
+describe('Cluster upgrade', () => {
+  let testBed: AppTestBed;
+  let server: ReturnType<typeof setupEnvironment>['server'];
+  let httpRequestsMockHelpers: ReturnType<typeof setupEnvironment>['httpRequestsMockHelpers'];
+
+  beforeEach(() => {
+    ({ server, httpRequestsMockHelpers } = setupEnvironment());
+  });
+
+  afterEach(() => {
+    server.restore();
+  });
+
+  describe('when user is still preparing for upgrade', () => {
+    beforeEach(async () => {
+      testBed = await setupAppPage();
+    });
+
+    test('renders overview', () => {
+      const { exists, find } = testBed;
+      expect(exists('overview')).toBe(true);
+      expect(exists('isUpgradingMessage')).toBe(false);
+      expect(exists('isUpgradeCompleteMessage')).toBe(false);
+    });
+  });
+
+  describe('when cluster is in the process of a rolling upgrade', () => {
+    beforeEach(async () => {
+      httpRequestsMockHelpers.setLoadDeprecationLoggingResponse(null, {
+        statusCode: 426,
+        attributes: {
+          allNodesUpgraded: false,
+        },
+      });
+
+      await act(async () => {
+        testBed = await setupAppPage();
+      });
+    });
+
+    test('renders rolling upgrade message', async () => {
+      const { component, exists } = testBed;
+      component.update();
+      expect(exists('overview')).toBe(false);
+      expect(exists('isUpgradingMessage')).toBe(true);
+      expect(exists('isUpgradeCompleteMessage')).toBe(false);
+    });
+  });
+
+  describe('when cluster has been upgraded', () => {
+    beforeEach(async () => {
+      httpRequestsMockHelpers.setLoadDeprecationLoggingResponse(null, {
+        statusCode: 426,
+        attributes: {
+          allNodesUpgraded: true,
+        },
+      });
+
+      await act(async () => {
+        testBed = await setupAppPage();
+      });
+    });
+
+    test('renders upgrade complete message', () => {
+      const { component, exists } = testBed;
+      component.update();
+      expect(exists('overview')).toBe(false);
+      expect(exists('isUpgradingMessage')).toBe(false);
+      expect(exists('isUpgradeCompleteMessage')).toBe(true);
+    });
+  });
+});

x-pack/plugins/upgrade_assistant/public/application/app.tsx

@@ -67,6 +67,7 @@ const App: React.FunctionComponent = () => {
         paddingSize="none"
         verticalPosition="center"
         horizontalPosition="center"
+        data-test-subj="isUpgradingMessage"
       >
         <EuiEmptyPrompt
           iconType="logoElasticsearch"
@@ -100,6 +101,7 @@ const App: React.FunctionComponent = () => {
         paddingSize="none"
         verticalPosition="center"
         horizontalPosition="center"
+        data-test-subj="isUpgradeCompleteMessage"
       >
         <EuiEmptyPrompt
           iconType="logoElasticsearch"