Add enzyme rerender test helper

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

@@ -0,0 +1,16 @@
+/*
+ * 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 { ShallowWrapper } from 'enzyme';
+
+/**
+ * Quick and easy helper for re-rendering a React component in Enzyme
+ * after (e.g.) updating Kea values
+ */
+export const rerender = (wrapper: ShallowWrapper) => {
+  wrapper.setProps({}); // Re-renders
+  wrapper.update(); // Just in case
+};

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

@@ -15,4 +15,5 @@ export { mockAllValues, mockAllActions, setMockValues, setMockActions } from './
 export { mountAsync } from './mount_async.mock';
 export { mountWithIntl } from './mount_with_i18n.mock';
 export { shallowWithIntl } from './shallow_with_i18n.mock';
+export { rerender } from './enzyme_rerender.mock';
 // Note: shallow_useeffect must be imported directly as a file

x-pack/plugins/enterprise_search/public/applications/app_search/components/credentials/credentials_flyout/form_components/key_engine_access.test.tsx

@@ -5,6 +5,7 @@
  */
 
 import { setMockValues, setMockActions } from '../../../../../__mocks__/kea.mock';
+import { rerender } from '../../../../../__mocks__';
 
 import React from 'react';
 import { shallow } from 'enzyme';
@@ -60,7 +61,7 @@ describe('FormKeyEngineAccess', () => {
       ...values,
       fullEngineAccessChecked: false,
     });
-    wrapper.setProps({}); // Re-render
+    rerender(wrapper);
 
     expect(wrapper.find('#all_engines').prop('checked')).toEqual(false);
     expect(wrapper.find('#all_engines').prop('value')).toEqual('false');

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

@@ -5,6 +5,7 @@
  */
 
 import { setMockValues } from '../../../__mocks__/kea.mock';
+import { rerender } from '../../../__mocks__';
 
 import React from 'react';
 import { shallow } from 'enzyme';
@@ -40,11 +41,11 @@ describe('EngineNav', () => {
     expect(wrapper.find(EuiBadge)).toHaveLength(0);
 
     setMockValues({ ...values, isSampleEngine: true });
-    wrapper.setProps({}); // Re-render
+    rerender(wrapper);
     expect(wrapper.find(EuiBadge).prop('children')).toEqual('SAMPLE ENGINE');
 
     setMockValues({ ...values, isMetaEngine: true });
-    wrapper.setProps({}); // Re-render
+    rerender(wrapper);
     expect(wrapper.find(EuiBadge).prop('children')).toEqual('META ENGINE');
   });
 

x-pack/plugins/enterprise_search/public/applications/enterprise_search/index.test.tsx

@@ -8,6 +8,7 @@ import React from 'react';
 import { shallow } from 'enzyme';
 
 import { setMockValues } from '../__mocks__/kea.mock';
+import { rerender } from '../__mocks__';
 
 import { EnterpriseSearch } from './';
 import { SetupGuide } from './components/setup_guide';
@@ -40,7 +41,7 @@ describe('EnterpriseSearch', () => {
       errorConnecting: true,
       config: { host: '' },
     });
-    wrapper.setProps({}); // Re-render
+    rerender(wrapper);
 
     expect(wrapper.find(ErrorConnecting)).toHaveLength(0);
     expect(wrapper.find(ProductSelector)).toHaveLength(1);