diff --git a/src/core_plugins/kibana/public/dashboard/panel/dashboard_panel.test.js b/src/core_plugins/kibana/public/dashboard/panel/dashboard_panel.test.js
index 35d1b1ab11384..322936be9a722 100644
--- a/src/core_plugins/kibana/public/dashboard/panel/dashboard_panel.test.js
+++ b/src/core_plugins/kibana/public/dashboard/panel/dashboard_panel.test.js
@@ -14,7 +14,7 @@ import { getEmbeddableFactoryMock } from '../__tests__/get_embeddable_factories_
 
 import {
   takeMountedSnapshot,
-} from 'ui_framework/src/test';
+} from 'ui_framework/test';
 
 function getProps(props = {}) {
   const defaultTestProps = {
diff --git a/src/core_plugins/kibana/public/dashboard/panel/panel_header/panel_header_container.test.js b/src/core_plugins/kibana/public/dashboard/panel/panel_header/panel_header_container.test.js
index 0ff43bcc65411..4b91d80dd9caa 100644
--- a/src/core_plugins/kibana/public/dashboard/panel/panel_header/panel_header_container.test.js
+++ b/src/core_plugins/kibana/public/dashboard/panel/panel_header/panel_header_container.test.js
@@ -16,7 +16,7 @@ import {
 import { getEmbeddableFactoryMock } from '../../__tests__/get_embeddable_factories_mock';
 import {
   TestSubjects,
-} from 'ui_framework/src/test';
+} from 'ui_framework/test';
 
 function getProps(props = {}) {
   const defaultTestProps = {
diff --git a/src/core_plugins/kibana/public/dashboard/top_nav/clone_modal.test.js b/src/core_plugins/kibana/public/dashboard/top_nav/clone_modal.test.js
index 80947ab8200c7..3638aa6eab185 100644
--- a/src/core_plugins/kibana/public/dashboard/top_nav/clone_modal.test.js
+++ b/src/core_plugins/kibana/public/dashboard/top_nav/clone_modal.test.js
@@ -1,6 +1,9 @@
 import React from 'react';
 import sinon from 'sinon';
 import { mount, render } from 'enzyme';
+import {
+  findTestSubject,
+} from 'ui_framework/test';
 
 import {
   DashboardCloneModal,
@@ -32,14 +35,14 @@ test('renders DashboardCloneModal', () => {
 
 test('onClone', () => {
   createComponent();
-  component.find('[data-test-subj="cloneConfirmButton"]').simulate('click');
+  findTestSubject(component, 'cloneConfirmButton', false).simulate('click');
   sinon.assert.calledWith(onClone, 'dash title');
   sinon.assert.notCalled(onClose);
 });
 
 test('onClose', () => {
   createComponent();
-  component.find('[data-test-subj="cloneCancelButton"]').simulate('click');
+  findTestSubject(component, 'cloneCancelButton', false).simulate('click');
   sinon.assert.calledOnce(onClose);
   sinon.assert.notCalled(onClone);
 });
@@ -48,6 +51,6 @@ test('title', () => {
   createComponent();
   const event = { target: { value: 'a' } };
   component.find('input').simulate('change', event);
-  component.find('[data-test-subj="cloneConfirmButton"]').simulate('click');
+  findTestSubject(component, 'cloneConfirmButton', false).simulate('click');
   sinon.assert.calledWith(onClone, 'a');
 });
diff --git a/src/dev/jest/config.json b/src/dev/jest/config.json
index f1eb6a82a0129..a30fe06e0c4f8 100644
--- a/src/dev/jest/config.json
+++ b/src/dev/jest/config.json
@@ -16,14 +16,14 @@
   "moduleNameMapper": {
     "^ui_framework/components": "<rootDir>/ui_framework/components",
     "^ui_framework/services": "<rootDir>/ui_framework/services",
-    "^ui_framework/src/test": "<rootDir>/ui_framework/src/test",
+    "^ui_framework/test": "<rootDir>/ui_framework/test",
     "^ui/(.*)": "<rootDir>/src/ui/public/$1",
     "\\.(jpg|jpeg|png|gif|eot|otf|webp|svg|ttf|woff|woff2|mp4|webm|wav|mp3|m4a|aac|oga)$": "<rootDir>/src/dev/jest/mocks/file_mock.js",
     "\\.(css|less|scss)$": "<rootDir>/src/dev/jest/mocks/style_mock.js"
   },
   "setupFiles": [
     "<rootDir>/src/dev/jest/setup/babel_polyfill.js",
-    "<rootDir>/src/dev/jest/setup/request_animation_frame_polyfill.js"
+    "<rootDir>/src/dev/jest/setup/request_animation_frame_polyfill.js",
     "<rootDir>/src/dev/jest/setup/enzyme.js"
   ],
   "coverageDirectory": "<rootDir>/target/jest-coverage",
diff --git a/src/jest/setup/enzyme.js b/src/jest/setup/enzyme.js
new file mode 100644
index 0000000000000..82edfc9e5adea
--- /dev/null
+++ b/src/jest/setup/enzyme.js
@@ -0,0 +1,4 @@
+import { configure } from 'enzyme';
+import Adapter from 'enzyme-adapter-react-16';
+
+configure({ adapter: new Adapter() });
diff --git a/ui_framework/src/components/code_editor/__snapshots__/code_editor.test.js.snap b/ui_framework/src/components/code_editor/__snapshots__/code_editor.test.js.snap
index 971c88b219d5f..1778b94580ad5 100644
--- a/ui_framework/src/components/code_editor/__snapshots__/code_editor.test.js.snap
+++ b/ui_framework/src/components/code_editor/__snapshots__/code_editor.test.js.snap
@@ -2,23 +2,21 @@
 
 exports[`KuiCodeEditor behavior hint element should be disabled when the ui ace box gains focus 1`] = `
 <div
-  className="kuiCodeEditorKeyboardHint kuiCodeEditorKeyboardHint-isInactive"
+  class="kuiCodeEditorKeyboardHint"
   data-test-subj="codeEditorHint"
-  id={42}
-  onClick={[Function]}
-  onKeyDown={[Function]}
+  id="42"
   role="button"
-  tabIndex="0"
+  tabindex="0"
 >
   <p
-    className="kuiText kuiVerticalRhythmSmall"
+    class="kuiText kuiVerticalRhythmSmall"
   >
     Press Enter to start 
     editing
     .
   </p>
   <p
-    className="kuiText kuiVerticalRhythmSmall"
+    class="kuiText kuiVerticalRhythmSmall"
   >
     When you’re done, press Escape to stop 
     editing
@@ -29,23 +27,46 @@ exports[`KuiCodeEditor behavior hint element should be disabled when the ui ace
 
 exports[`KuiCodeEditor behavior hint element should be enabled when the ui ace box loses focus 1`] = `
 <div
-  className="kuiCodeEditorKeyboardHint"
+  class="kuiCodeEditorKeyboardHint"
   data-test-subj="codeEditorHint"
-  id={42}
-  onClick={[Function]}
-  onKeyDown={[Function]}
+  id="42"
   role="button"
-  tabIndex="0"
+  tabindex="0"
 >
   <p
-    className="kuiText kuiVerticalRhythmSmall"
+    class="kuiText kuiVerticalRhythmSmall"
   >
     Press Enter to start 
     editing
     .
   </p>
   <p
-    className="kuiText kuiVerticalRhythmSmall"
+    class="kuiText kuiVerticalRhythmSmall"
+  >
+    When you’re done, press Escape to stop 
+    editing
+    .
+  </p>
+</div>
+`;
+
+exports[`KuiCodeEditor behavior hint element should be tabable 1`] = `
+<div
+  class="kuiCodeEditorKeyboardHint"
+  data-test-subj="codeEditorHint"
+  id="42"
+  role="button"
+  tabindex="0"
+>
+  <p
+    class="kuiText kuiVerticalRhythmSmall"
+  >
+    Press Enter to start 
+    editing
+    .
+  </p>
+  <p
+    class="kuiText kuiVerticalRhythmSmall"
   >
     When you’re done, press Escape to stop 
     editing
@@ -68,28 +89,12 @@ exports[`KuiCodeEditor is rendered 1`] = `
     <p
       class="kuiText kuiVerticalRhythmSmall"
     >
-      <!-- react-text: 4 -->
-      Press Enter to start 
-      <!-- /react-text -->
-      <!-- react-text: 5 -->
-      editing
-      <!-- /react-text -->
-      <!-- react-text: 6 -->
-      .
-      <!-- /react-text -->
+      Press Enter to start editing.
     </p>
     <p
       class="kuiText kuiVerticalRhythmSmall"
     >
-      <!-- react-text: 8 -->
-      When you’re done, press Escape to stop 
-      <!-- /react-text -->
-      <!-- react-text: 9 -->
-      editing
-      <!-- /react-text -->
-      <!-- react-text: 10 -->
-      .
-      <!-- /react-text -->
+      When you’re done, press Escape to stop editing.
     </p>
   </div>
   <div
@@ -197,28 +202,12 @@ exports[`KuiCodeEditor props isReadOnly renders alternate hint text 1`] = `
     <p
       class="kuiText kuiVerticalRhythmSmall"
     >
-      <!-- react-text: 4 -->
-      Press Enter to start 
-      <!-- /react-text -->
-      <!-- react-text: 5 -->
-      interacting with the code
-      <!-- /react-text -->
-      <!-- react-text: 6 -->
-      .
-      <!-- /react-text -->
+      Press Enter to start interacting with the code.
     </p>
     <p
       class="kuiText kuiVerticalRhythmSmall"
     >
-      <!-- react-text: 8 -->
-      When you’re done, press Escape to stop 
-      <!-- /react-text -->
-      <!-- react-text: 9 -->
-      interacting with the code
-      <!-- /react-text -->
-      <!-- react-text: 10 -->
-      .
-      <!-- /react-text -->
+      When you’re done, press Escape to stop interacting with the code.
     </p>
   </div>
   <div
diff --git a/ui_framework/src/components/code_editor/code_editor.test.js b/ui_framework/src/components/code_editor/code_editor.test.js
index e204f4bf02edb..f610d8a7d450a 100644
--- a/ui_framework/src/components/code_editor/code_editor.test.js
+++ b/ui_framework/src/components/code_editor/code_editor.test.js
@@ -4,6 +4,7 @@ import { mount } from 'enzyme';
 import { KuiCodeEditor } from './code_editor';
 import { keyCodes } from '../../services';
 import {
+  findTestSubject,
   requiredProps,
   takeMountedSnapshot,
 } from '../../test';
@@ -37,20 +38,21 @@ describe('KuiCodeEditor', () => {
 
     describe('hint element', () => {
       test('should be tabable', () => {
-        expect(component.find('[data-test-subj="codeEditorHint"]').prop('tabIndex')).toBe('0');
+        const hint = findTestSubject(component, 'codeEditorHint');
+        expect(hint).toMatchSnapshot();
       });
 
       test('should be disabled when the ui ace box gains focus', () => {
-        const hint = component.find('[data-test-subj="codeEditorHint"]');
-        hint.simulate('keydown', { keyCode: keyCodes.ENTER });
-        expect(hint).toMatchSnapshot();
+        const hint = findTestSubject(component, 'codeEditorHint', false);
+        hint.simulate('keyup', { keyCode: keyCodes.ENTER });
+        expect(findTestSubject(component, 'codeEditorHint')).toMatchSnapshot();
       });
 
       test('should be enabled when the ui ace box loses focus', () => {
-        const hint = component.find('[data-test-subj="codeEditorHint"]');
-        hint.simulate('keydown', { keyCode: keyCodes.ENTER });
+        const hint = findTestSubject(component, 'codeEditorHint', false);
+        hint.simulate('keyup', { keyCode: keyCodes.ENTER });
         component.instance().onBlurAce();
-        expect(hint).toMatchSnapshot();
+        expect(findTestSubject(component, 'codeEditorHint')).toMatchSnapshot();
       });
     });
 
@@ -66,11 +68,10 @@ describe('KuiCodeEditor', () => {
         component.instance().onKeydownAce({
           preventDefault: () => {},
           stopPropagation: () => {},
-          keyCode: keyCodes.ESCAPE
+          keyCode: keyCodes.ESCAPE,
         });
-        expect(
-          component.find('[data-test-subj="codeEditorHint"]').matchesElement(document.activeElement)
-        ).toBe(true);
+        const hint = findTestSubject(component, 'codeEditorHint');
+        expect(hint).toBe(document.activeElement);
       });
     });
   });
diff --git a/ui_framework/src/components/context_menu/__snapshots__/context_menu_item.test.js.snap b/ui_framework/src/components/context_menu/__snapshots__/context_menu_item.test.js.snap
index de10f1d71ac7e..a11599a5244f5 100644
--- a/ui_framework/src/components/context_menu/__snapshots__/context_menu_item.test.js.snap
+++ b/ui_framework/src/components/context_menu/__snapshots__/context_menu_item.test.js.snap
@@ -18,6 +18,21 @@ exports[`KuiContextMenuItem is rendered 1`] = `
 </button>
 `;
 
+exports[`KuiContextMenuItem props disabled is rendered 1`] = `
+<button
+  class="kuiContextMenuItem kuiContextMenuItem-disabled"
+  disabled=""
+>
+  <span
+    class="kuiContextMenu__itemLayout"
+  >
+    <span
+      class="kuiContextMenuItem__text"
+    />
+  </span>
+</button>
+`;
+
 exports[`KuiContextMenuItem props hasPanel is rendered 1`] = `
 <button
   class="kuiContextMenuItem"
diff --git a/ui_framework/src/components/context_menu/context_menu_item.test.js b/ui_framework/src/components/context_menu/context_menu_item.test.js
index fe4658d9d06fd..be6bbdeee0f88 100644
--- a/ui_framework/src/components/context_menu/context_menu_item.test.js
+++ b/ui_framework/src/components/context_menu/context_menu_item.test.js
@@ -29,6 +29,17 @@ describe('KuiContextMenuItem', () => {
       });
     });
 
+    describe('disabled', () => {
+      test('is rendered', () => {
+        const component = render(
+          <KuiContextMenuItem disabled />
+        );
+
+        expect(component)
+          .toMatchSnapshot();
+      });
+    });
+
     describe('onClick', () => {
       test(`isn't called upon instantiation`, () => {
         const onClickHandler = sinon.stub();
@@ -56,7 +67,7 @@ describe('KuiContextMenuItem', () => {
         const onClickHandler = sinon.stub();
 
         const component = mount(
-          <KuiContextMenuItem disabled={true} onClick={onClickHandler} />
+          <KuiContextMenuItem disabled onClick={onClickHandler} />
         );
 
         component.simulate('click');
diff --git a/ui_framework/src/components/context_menu/context_menu_panel.test.js b/ui_framework/src/components/context_menu/context_menu_panel.test.js
index 6555217ec5637..00115a173de1f 100644
--- a/ui_framework/src/components/context_menu/context_menu_panel.test.js
+++ b/ui_framework/src/components/context_menu/context_menu_panel.test.js
@@ -1,7 +1,7 @@
 import React from 'react';
-import { render, shallow, mount } from 'enzyme';
+import { render, mount } from 'enzyme';
 import sinon from 'sinon';
-import { requiredProps } from '../../test/required_props';
+import { findTestSubject, requiredProps } from '../../test';
 
 import {
   KuiContextMenuPanel,
@@ -75,7 +75,7 @@ describe('KuiContextMenuPanel', () => {
       test(`isn't called upon instantiation`, () => {
         const onCloseHandler = sinon.stub();
 
-        shallow(
+        mount(
           <KuiContextMenuPanel title="Title" onClose={onCloseHandler} />
         );
 
@@ -85,7 +85,7 @@ describe('KuiContextMenuPanel', () => {
       test('is called when the title is clicked', () => {
         const onCloseHandler = sinon.stub();
 
-        const component = shallow(
+        const component = mount(
           <KuiContextMenuPanel title="Title" onClose={onCloseHandler} />
         );
 
@@ -170,9 +170,7 @@ describe('KuiContextMenuPanel', () => {
           />
         );
 
-        expect(
-          component.find('[data-test-subj="itemB"]').matchesElement(document.activeElement)
-        ).toBe(true);
+        expect(findTestSubject(component, 'itemB')).toBe(document.activeElement);
       });
     });
 
@@ -278,9 +276,7 @@ describe('KuiContextMenuPanel', () => {
           </KuiContextMenuPanel>
         );
 
-        expect(
-          component.find('[data-test-subj="button"]').matchesElement(document.activeElement)
-        ).toBe(true);
+        expect(findTestSubject(component, 'button')).toBe(document.activeElement);
       });
 
       it('is not set on anything if hasFocus is false', () => {
@@ -290,9 +286,7 @@ describe('KuiContextMenuPanel', () => {
           </KuiContextMenuPanel>
         );
 
-        expect(
-          component.find('[data-test-subj="button"]').matchesElement(document.activeElement)
-        ).toBe(true);
+        expect(findTestSubject(component, 'button')).not.toBe(document.activeElement);
       });
     });
 
@@ -314,60 +308,48 @@ describe('KuiContextMenuPanel', () => {
         );
       });
 
-      it(`doesn't focus an item by default`, () => {
-        expect(component.find('[data-test-subj]').contains(document.activeElement)).toBe(false);
+      it(`focuses the panel by default`, () => {
+        expect(component.getDOMNode()).toBe(document.activeElement);
       });
 
       it('down arrow key focuses the first menu item', () => {
         component.simulate('keydown', { keyCode: keyCodes.DOWN });
 
-        expect(
-          component.find('[data-test-subj="itemA"]').matchesElement(document.activeElement)
-        ).toBe(true);
+        expect(findTestSubject(component, 'itemA')).toBe(document.activeElement);
       });
 
       it('subsequently, down arrow key focuses the next menu item', () => {
         component.simulate('keydown', { keyCode: keyCodes.DOWN });
         component.simulate('keydown', { keyCode: keyCodes.DOWN });
 
-        expect(
-          component.find('[data-test-subj="itemB"]').matchesElement(document.activeElement)
-        ).toBe(true);
+        expect(findTestSubject(component, 'itemB')).toBe(document.activeElement);
       });
 
       it('down arrow key wraps to first menu item', () => {
         component.simulate('keydown', { keyCode: keyCodes.UP });
         component.simulate('keydown', { keyCode: keyCodes.DOWN });
 
-        expect(
-          component.find('[data-test-subj="itemA"]').matchesElement(document.activeElement)
-        ).toBe(true);
+        expect(findTestSubject(component, 'itemA')).toBe(document.activeElement);
       });
 
       it('up arrow key focuses the last menu item', () => {
         component.simulate('keydown', { keyCode: keyCodes.UP });
 
-        expect(
-          component.find('[data-test-subj="itemC"]').matchesElement(document.activeElement)
-        ).toBe(true);
+        expect(findTestSubject(component, 'itemC')).toBe(document.activeElement);
       });
 
       it('subsequently, up arrow key focuses the previous menu item', () => {
         component.simulate('keydown', { keyCode: keyCodes.UP });
         component.simulate('keydown', { keyCode: keyCodes.UP });
 
-        expect(
-          component.find('[data-test-subj="itemB"]').matchesElement(document.activeElement)
-        ).toBe(true);
+        expect(findTestSubject(component, 'itemB')).toBe(document.activeElement);
       });
 
       it('up arrow key wraps to last menu item', () => {
         component.simulate('keydown', { keyCode: keyCodes.DOWN });
         component.simulate('keydown', { keyCode: keyCodes.UP });
 
-        expect(
-          component.find('[data-test-subj="itemC"]').matchesElement(document.activeElement)
-        ).toBe(true);
+        expect(findTestSubject(component, 'itemC')).toBe(document.activeElement);
       });
 
       it(`right arrow key shows next panel with focused item's index`, () => {
diff --git a/ui_framework/src/components/modal/confirm_modal.test.js b/ui_framework/src/components/modal/confirm_modal.test.js
index ef98bbe895a39..efd6c69b0c3df 100644
--- a/ui_framework/src/components/modal/confirm_modal.test.js
+++ b/ui_framework/src/components/modal/confirm_modal.test.js
@@ -2,7 +2,7 @@ import React from 'react';
 import sinon from 'sinon';
 import { mount, render } from 'enzyme';
 
-import { requiredProps } from '../../test/required_props';
+import { findTestSubject, requiredProps } from '../../test';
 import { keyCodes } from '../../services';
 
 import {
@@ -42,7 +42,8 @@ test('onConfirm', () => {
       confirmButtonText="Confirm Button Text"
     />
   );
-  component.find('[data-test-subj="confirmModalConfirmButton"]').simulate('click');
+
+  findTestSubject(component, 'confirmModalConfirmButton', false).simulate('click');
   sinon.assert.calledOnce(onConfirm);
   sinon.assert.notCalled(onCancel);
 });
@@ -57,7 +58,8 @@ describe('onCancel', () => {
         confirmButtonText="Confirm Button Text"
       />
     );
-    component.find('[data-test-subj="confirmModalCancelButton"]').simulate('click');
+
+    findTestSubject(component, 'confirmModalCancelButton', false).simulate('click');
     sinon.assert.notCalled(onConfirm);
     sinon.assert.calledOnce(onCancel);
   });
@@ -72,7 +74,8 @@ describe('onCancel', () => {
         data-test-subj="modal"
       />
     );
-    component.find('[data-test-subj="modal"]').simulate('keydown', { keyCode: keyCodes.ESCAPE });
+
+    findTestSubject(component, 'modal', false).simulate('keydown', { keyCode: keyCodes.ESCAPE });
     sinon.assert.notCalled(onConfirm);
     sinon.assert.calledOnce(onCancel);
   });
@@ -83,12 +86,14 @@ describe('defaultFocusedButton', () => {
     const component = mount(
       <KuiConfirmModal
         onCancel={onCancel}
-        defaultFocusedButton={CANCEL_BUTTON}
+        onConfirm={onConfirm}
         cancelButtonText="Cancel Button Text"
         confirmButtonText="Confirm Button Text"
+        defaultFocusedButton={CANCEL_BUTTON}
       />
     );
-    const button = component.find('[data-test-subj="confirmModalCancelButton"]').getDOMNode();
+
+    const button = findTestSubject(component, 'confirmModalCancelButton');
     expect(document.activeElement).toEqual(button);
   });
 
@@ -96,12 +101,14 @@ describe('defaultFocusedButton', () => {
     const component = mount(
       <KuiConfirmModal
         onCancel={onCancel}
-        defaultFocusedButton={CONFIRM_BUTTON}
+        onConfirm={onConfirm}
         cancelButtonText="Cancel Button Text"
         confirmButtonText="Confirm Button Text"
+        defaultFocusedButton={CONFIRM_BUTTON}
       />
     );
-    const button = component.find('[data-test-subj="confirmModalConfirmButton"]').getDOMNode();
+
+    const button = findTestSubject(component, 'confirmModalConfirmButton');
     expect(document.activeElement).toEqual(button);
   });
 
@@ -109,6 +116,7 @@ describe('defaultFocusedButton', () => {
     const component = mount(
       <KuiConfirmModal
         onCancel={onCancel}
+        onConfirm={onConfirm}
         cancelButtonText="Cancel Button Text"
         confirmButtonText="Confirm Button Text"
       />
diff --git a/ui_framework/src/components/pager/pager.test.js b/ui_framework/src/components/pager/pager.test.js
index ab05d73e92d7e..2904411735ce1 100644
--- a/ui_framework/src/components/pager/pager.test.js
+++ b/ui_framework/src/components/pager/pager.test.js
@@ -1,7 +1,7 @@
 import React from 'react';
 import sinon from 'sinon';
 import { render, mount } from 'enzyme';
-import { requiredProps } from '../../test/required_props';
+import { findTestSubject, requiredProps } from '../../test';
 
 import {
   KuiPager,
@@ -72,7 +72,7 @@ describe('property', () => {
         totalItems={20}
       />);
       const pager = mount(component);
-      pager.find('[data-test-subj="pagerPreviousButton"]').simulate('click');
+      findTestSubject(pager, 'pagerPreviousButton', false).simulate('click');
       sinon.assert.calledOnce(onPreviousPage);
       sinon.assert.notCalled(onNextPage);
     });
@@ -90,7 +90,7 @@ describe('property', () => {
         totalItems={20}
       />);
       const pager = mount(component);
-      pager.find('[data-test-subj="pagerNextButton"]').simulate('click');
+      findTestSubject(pager, 'pagerNextButton', false).simulate('click');
       sinon.assert.calledOnce(onNextPage);
       sinon.assert.notCalled(onPreviousPage);
     });
diff --git a/ui_framework/src/components/pager/pager_button_group.test.js b/ui_framework/src/components/pager/pager_button_group.test.js
index 272f0ac042964..0356aa348abbc 100644
--- a/ui_framework/src/components/pager/pager_button_group.test.js
+++ b/ui_framework/src/components/pager/pager_button_group.test.js
@@ -1,7 +1,7 @@
 import React from 'react';
 import sinon from 'sinon';
 import { render, mount } from 'enzyme';
-import { requiredProps } from '../../test/required_props';
+import { findTestSubject, requiredProps } from '../../test';
 
 import {
   KuiPagerButtonGroup,
@@ -27,13 +27,12 @@ test('renders KuiPagerButtonGroup', () => {
 });
 
 describe('property', () => {
-
   function findPreviousButton(pager) {
-    return pager.find('[data-test-subj="pagerPreviousButton"]');
+    return findTestSubject(pager, 'pagerPreviousButton', false);
   }
 
   function findNextButton(pager) {
-    return pager.find('[data-test-subj="pagerNextButton"]');
+    return findTestSubject(pager, 'pagerNextButton', false);
   }
 
   test('onNext', () => {
diff --git a/ui_framework/src/test/find_test_subject.js b/ui_framework/src/test/find_test_subject.js
new file mode 100644
index 0000000000000..7864dee0bb215
--- /dev/null
+++ b/ui_framework/src/test/find_test_subject.js
@@ -0,0 +1,10 @@
+// Extract the DOM node which matches a specific test subject selector.
+export const findTestSubject = (mountedComponent, testSubjectSelector, isDOMNode = true) => {
+  const testSubject = mountedComponent.find(`[data-test-subj="${testSubjectSelector}"]`);
+
+  if (isDOMNode) {
+    return testSubject.hostNodes().getDOMNode();
+  }
+
+  return testSubject.hostNodes();
+};
diff --git a/ui_framework/src/test/index.js b/ui_framework/src/test/index.js
index b6884ec88ed6d..94e5bd41564b2 100644
--- a/ui_framework/src/test/index.js
+++ b/ui_framework/src/test/index.js
@@ -1,5 +1,6 @@
 export { requiredProps } from './required_props';
 export { takeMountedSnapshot } from './take_mounted_snapshot';
+export { findTestSubject } from './find_test_subject';
 import * as TestSubjects from './test_subjects';
 export { TestSubjects };
 
diff --git a/ui_framework/test/index.js b/ui_framework/test/index.js
new file mode 100644
index 0000000000000..6821eadd764fe
--- /dev/null
+++ b/ui_framework/test/index.js
@@ -0,0 +1 @@
+export * from '../src/test';