[Enterprise Search] Move http out of React Context and to Kea Logic; prefer directly mounting

x-pack/plugins/enterprise_search/public/applications/__mocks__/flash_messages_logic.mock.ts

@@ -0,0 +1,17 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License;
+ * you may not use this file except in compliance with the Elastic License.
+ */
+
+export const mockFlashMessagesValues = {
+  messages: [],
+  queuedMessages: [],
+};
+
+export const mockFlashMessagesActions = {
+  setFlashMessages: jest.fn(),
+  clearFlashMessages: jest.fn(),
+  setQueuedMessages: jest.fn(),
+  clearQueuedMessages: jest.fn(),
+};

x-pack/plugins/enterprise_search/public/applications/__mocks__/http_logic.mock.ts

@@ -0,0 +1,13 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License;
+ * you may not use this file except in compliance with the Elastic License.
+ */
+
+import { httpServiceMock } from 'src/core/public/mocks';
+
+export const mockHttpValues = {
+  http: httpServiceMock.createSetupContract(),
+  errorConnecting: false,
+  readOnlyMode: false,
+};

x-pack/plugins/enterprise_search/public/applications/__mocks__/index.ts

@@ -7,6 +7,10 @@
 export { mockHistory, mockLocation } from './react_router_history.mock';
 export { mockKibanaContext } from './kibana_context.mock';
 export { mockLicenseContext } from './license_context.mock';
+export { mockHttpValues } from './http_logic.mock';
+export { mockFlashMessagesValues, mockFlashMessagesActions } from './flash_messages_logic.mock';
+export { mockAllValues, mockAllActions, setMockValues } from './kea.mock';
+
 export {
   mountWithContext,
   mountWithKibanaContext,

x-pack/plugins/enterprise_search/public/applications/__mocks__/kea.mock.ts

@@ -4,21 +4,46 @@
  * you may not use this file except in compliance with the Elastic License.
  */
 
+/**
+ * NOTE: These variable names MUST start with 'mock*' in order for
+ * Jest to accept its use within a jest.mock()
+ */
+import { mockHttpValues } from './http_logic.mock';
+import { mockFlashMessagesValues, mockFlashMessagesActions } from './flash_messages_logic.mock';
+
+export const mockAllValues = {
+  ...mockHttpValues,
+  ...mockFlashMessagesValues,
+};
+export const mockAllActions = {
+  ...mockFlashMessagesActions,
+};
+
+/**
+ * Import this file directly to mock useValues with a set of default values for all shared logic files.
+ * Example usage:
+ *
+ * import '../../../__mocks__/kea'; // Must come before kea's import, adjust relative path as needed
+ */
 jest.mock('kea', () => ({
   ...(jest.requireActual('kea') as object),
-  useValues: jest.fn(() => ({})),
-  useActions: jest.fn(() => ({})),
+  useValues: jest.fn(() => ({ ...mockAllValues })),
+  useActions: jest.fn(() => ({ ...mockAllActions })),
 }));
 
 /**
+ * Call this function to override a specific set of Kea values while retaining all other defaults
  * Example usage within a component test:
  *
- * import '../../../__mocks__/kea'; // Must come before kea's import, adjust relative path as needed
- *
- * import { useActions, useValues } from 'kea';
+ * import '../../../__mocks__/kea';
+ * import { setMockValues } from ''../../../__mocks__';
  *
  * it('some test', () => {
- *   (useValues as jest.Mock).mockImplementationOnce(() => ({ someValue: 'hello' }));
- *   (useActions as jest.Mock).mockImplementationOnce(() => ({ someAction: () => 'world' }));
+ *   setMockValues({ someValue: 'hello' });
  * });
  */
+import { useValues } from 'kea';
+
+export const setMockValues = (values: object) => {
+  (useValues as jest.Mock).mockImplementation(() => ({ ...mockAllValues, ...values }));
+};

x-pack/plugins/enterprise_search/public/applications/__mocks__/kibana_context.mock.ts

@@ -4,15 +4,13 @@
  * you may not use this file except in compliance with the Elastic License.
  */
 
-import { httpServiceMock } from 'src/core/public/mocks';
 import { ExternalUrl } from '../shared/enterprise_search_url';
 
 /**
  * A set of default Kibana context values to use across component tests.
  * @see enterprise_search/public/index.tsx for the KibanaContext definition/import
  */
 export const mockKibanaContext = {
-  http: httpServiceMock.createSetupContract(),
   navigateToUrl: jest.fn(),
   setBreadcrumbs: jest.fn(),
   setDocTitle: jest.fn(),

x-pack/plugins/enterprise_search/public/applications/__mocks__/mount_with_context.mock.tsx

@@ -67,7 +67,7 @@ export const mountWithKibanaContext = (children: React.ReactNode, context?: obje
  */
 export const mountWithAsyncContext = async (
   children: React.ReactNode,
-  context: object
+  context?: object
 ): Promise<ReactWrapper> => {
   let wrapper: ReactWrapper | undefined;
 

x-pack/plugins/enterprise_search/public/applications/__mocks__/react_router_history.mock.ts

@@ -14,6 +14,7 @@ export const mockHistory = {
   location: {
     pathname: '/current-path',
   },
+  listen: jest.fn(() => jest.fn()),
 };
 export const mockLocation = {
   key: 'someKey',

x-pack/plugins/enterprise_search/public/applications/app_search/components/engine_overview/components/empty_state.test.tsx

@@ -4,6 +4,7 @@
  * you may not use this file except in compliance with the Elastic License.
  */
 
+import '../../../../__mocks__/kea.mock';
 import '../../../../__mocks__/shallow_usecontext.mock';
 
 import React from 'react';

x-pack/plugins/enterprise_search/public/applications/app_search/components/engine_overview/components/empty_state.tsx

@@ -5,10 +5,12 @@
  */
 
 import React, { useContext } from 'react';
+import { useValues } from 'kea';
 import { EuiPageContent, EuiEmptyPrompt, EuiButton } from '@elastic/eui';
 import { FormattedMessage } from '@kbn/i18n/react';
 
 import { sendTelemetry } from '../../../../shared/telemetry';
+import { HttpLogic } from '../../../../shared/http';
 import { SetAppSearchChrome as SetPageChrome } from '../../../../shared/kibana_chrome';
 import { KibanaContext, IKibanaContext } from '../../../../index';
 import { CREATE_ENGINES_PATH } from '../../../routes';
@@ -18,9 +20,9 @@ import { EngineOverviewHeader } from './header';
 import './empty_state.scss';
 
 export const EmptyState: React.FC = () => {
+  const { http } = useValues(HttpLogic);
   const {
     externalUrl: { getAppSearchUrl },
-    http,
   } = useContext(KibanaContext) as IKibanaContext;
 
   const buttonProps = {

x-pack/plugins/enterprise_search/public/applications/app_search/components/engine_overview/components/header.test.tsx

@@ -4,6 +4,7 @@
  * you may not use this file except in compliance with the Elastic License.
  */
 
+import '../../../../__mocks__/kea.mock';
 import '../../../../__mocks__/shallow_usecontext.mock';
 
 import React from 'react';

x-pack/plugins/enterprise_search/public/applications/app_search/components/engine_overview/components/header.tsx

@@ -5,6 +5,7 @@
  */
 
 import React, { useContext } from 'react';
+import { useValues } from 'kea';
 import {
   EuiPageHeader,
   EuiPageHeaderSection,
@@ -16,12 +17,13 @@ import {
 import { FormattedMessage } from '@kbn/i18n/react';
 
 import { sendTelemetry } from '../../../../shared/telemetry';
+import { HttpLogic } from '../../../../shared/http';
 import { KibanaContext, IKibanaContext } from '../../../../index';
 
 export const EngineOverviewHeader: React.FC = () => {
+  const { http } = useValues(HttpLogic);
   const {
     externalUrl: { getAppSearchUrl },
-    http,
   } = useContext(KibanaContext) as IKibanaContext;
 
   const buttonProps = {

x-pack/plugins/enterprise_search/public/applications/app_search/components/engine_overview/engine_overview.test.tsx

@@ -4,22 +4,21 @@
  * you may not use this file except in compliance with the Elastic License.
  */
 
+import '../../../__mocks__/kea.mock';
 import '../../../__mocks__/react_router_history.mock';
 
 import React from 'react';
 import { act } from 'react-dom/test-utils';
 import { shallow, ReactWrapper } from 'enzyme';
 
-import { mountWithAsyncContext, mockKibanaContext } from '../../../__mocks__';
+import { mountWithAsyncContext, mockHttpValues, setMockValues } from '../../../__mocks__';
 
 import { LoadingState, EmptyState } from './components';
 import { EngineTable } from './engine_table';
 
 import { EngineOverview } from './';
 
 describe('EngineOverview', () => {
-  const mockHttp = mockKibanaContext.http;
-
   describe('non-happy-path states', () => {
     it('isLoading', () => {
       const wrapper = shallow(<EngineOverview />);
@@ -28,15 +27,16 @@ describe('EngineOverview', () => {
     });
 
     it('isEmpty', async () => {
-      const wrapper = await mountWithAsyncContext(<EngineOverview />, {
+      setMockValues({
         http: {
-          ...mockHttp,
+          ...mockHttpValues.http,
           get: () => ({
             results: [],
             meta: { page: { total_results: 0 } },
           }),
         },
       });
+      const wrapper = await mountWithAsyncContext(<EngineOverview />);
 
       expect(wrapper.find(EmptyState)).toHaveLength(1);
     });
@@ -65,12 +65,11 @@ describe('EngineOverview', () => {
 
     beforeEach(() => {
       jest.clearAllMocks();
+      setMockValues({ http: { ...mockHttpValues.http, get: mockApi } });
     });
 
     it('renders and calls the engines API', async () => {
-      const wrapper = await mountWithAsyncContext(<EngineOverview />, {
-        http: { ...mockHttp, get: mockApi },
-      });
+      const wrapper = await mountWithAsyncContext(<EngineOverview />);
 
       expect(wrapper.find(EngineTable)).toHaveLength(1);
       expect(mockApi).toHaveBeenNthCalledWith(1, '/api/app_search/engines', {
@@ -84,7 +83,6 @@ describe('EngineOverview', () => {
     describe('when on a platinum license', () => {
       it('renders a 2nd meta engines table & makes a 2nd meta engines API call', async () => {
         const wrapper = await mountWithAsyncContext(<EngineOverview />, {
-          http: { ...mockHttp, get: mockApi },
           license: { type: 'platinum', isActive: true },
         });
 
@@ -103,19 +101,15 @@ describe('EngineOverview', () => {
         wrapper.find(EngineTable).prop('pagination');
 
       it('passes down page data from the API', async () => {
-        const wrapper = await mountWithAsyncContext(<EngineOverview />, {
-          http: { ...mockHttp, get: mockApi },
-        });
+        const wrapper = await mountWithAsyncContext(<EngineOverview />);
         const pagination = getTablePagination(wrapper);
 
         expect(pagination.totalEngines).toEqual(100);
         expect(pagination.pageIndex).toEqual(0);
       });
 
       it('re-polls the API on page change', async () => {
-        const wrapper = await mountWithAsyncContext(<EngineOverview />, {
-          http: { ...mockHttp, get: mockApi },
-        });
+        const wrapper = await mountWithAsyncContext(<EngineOverview />);
         await act(async () => getTablePagination(wrapper).onPaginate(5));
         wrapper.update();
 

x-pack/plugins/enterprise_search/public/applications/app_search/components/engine_overview/engine_overview.tsx

@@ -5,20 +5,21 @@
  */
 
 import React, { useContext, useEffect, useState } from 'react';
+import { useValues } from 'kea';
 import {
   EuiPageContent,
   EuiPageContentHeader,
   EuiPageContentBody,
   EuiTitle,
   EuiSpacer,
 } from '@elastic/eui';
-import { FormattedMessage } from '@kbn/i18n/react';
+import { i18n } from '@kbn/i18n';
 
 import { SetAppSearchChrome as SetPageChrome } from '../../../shared/kibana_chrome';
 import { SendAppSearchTelemetry as SendTelemetry } from '../../../shared/telemetry';
 import { FlashMessages } from '../../../shared/flash_messages';
+import { HttpLogic } from '../../../shared/http';
 import { LicenseContext, ILicenseContext, hasPlatinumLicense } from '../../../shared/licensing';
-import { KibanaContext, IKibanaContext } from '../../../index';
 
 import { EngineIcon } from './assets/engine_icon';
 import { MetaEngineIcon } from './assets/meta_engine_icon';
@@ -38,7 +39,7 @@ interface ISetEnginesCallbacks {
 }
 
 export const EngineOverview: React.FC = () => {
-  const { http } = useContext(KibanaContext) as IKibanaContext;
+  const { http } = useValues(HttpLogic);
   const { license } = useContext(LicenseContext) as ILicenseContext;
 
   const [isLoading, setIsLoading] = useState(true);
@@ -94,10 +95,9 @@ export const EngineOverview: React.FC = () => {
           <EuiTitle size="s">
             <h2>
               <EngineIcon />
-              <FormattedMessage
-                id="xpack.enterpriseSearch.appSearch.enginesOverview.engines"
-                defaultMessage="Engines"
-              />
+              {i18n.translate('xpack.enterpriseSearch.appSearch.enginesOverview.engines', {
+                defaultMessage: 'Engines',
+              })}
             </h2>
           </EuiTitle>
         </EuiPageContentHeader>
@@ -119,10 +119,9 @@ export const EngineOverview: React.FC = () => {
               <EuiTitle size="s">
                 <h2>
                   <MetaEngineIcon />
-                  <FormattedMessage
-                    id="xpack.enterpriseSearch.appSearch.enginesOverview.metaEngines"
-                    defaultMessage="Meta Engines"
-                  />
+                  {i18n.translate('xpack.enterpriseSearch.appSearch.enginesOverview.metaEngines', {
+                    defaultMessage: 'Meta Engines',
+                  })}
                 </h2>
               </EuiTitle>
             </EuiPageContentHeader>