feature: screen

Author: mdjastrzebski

src/__tests__/screen.test.tsx

@@ -0,0 +1,39 @@
+import * as React from 'react';
+import { Text } from 'react-native';
+import { render, screen } from '..';
+
+test('screen has the same queries as render result', () => {
+  const result = render(<Text>Mt. Everest</Text>);
+  expect(screen).toBe(result);
+
+  expect(screen.getByText('Mt. Everest')).toBeTruthy();
+  expect(screen.queryByText('Mt. Everest')).toBeTruthy();
+  expect(screen.getAllByText('Mt. Everest')).toHaveLength(1);
+  expect(screen.queryAllByText('Mt. Everest')).toHaveLength(1);
+});
+
+test('screen holds last render result', () => {
+  render(<Text>Mt. Everest</Text>);
+  render(<Text>Mt. Blanc</Text>);
+  const finalResult = render(<Text>Śnieżka</Text>);
+  expect(screen).toBe(finalResult);
+
+  expect(screen.getByText('Śnieżka')).toBeTruthy();
+  expect(screen.queryByText('Mt. Everest')).toBeFalsy();
+  expect(screen.queryByText('Mt. Blanc')).toBeFalsy();
+});
+
+test('screen throws without render', () => {
+  expect(() => screen.container).toThrowError(
+    '`render` method has not been called'
+  );
+  expect(() => screen.debug()).toThrowError(
+    '`render` method has not been called'
+  );
+  expect(() => screen.debug.shallow()).toThrowError(
+    '`render` method has not been called'
+  );
+  expect(() => screen.getByText('Mt. Everest')).toThrowError(
+    '`render` method has not been called'
+  );
+});

src/cleanup.ts

@@ -1,9 +1,11 @@
 import * as React from 'react';
+import { clearRenderResult } from './screen';
 
 type CleanUpFunction = (nextElement?: React.ReactElement<any>) => void;
 let cleanupQueue = new Set<CleanUpFunction>();
 
 export default function cleanup() {
+  clearRenderResult();
   cleanupQueue.forEach((fn) => fn());
   cleanupQueue.clear();
 }

src/pure.ts

@@ -1,20 +1,22 @@
 import act from './act';
 import cleanup from './cleanup';
 import fireEvent from './fireEvent';
-import render from './render';
+import render, { RenderResult } from './render';
 import waitFor from './waitFor';
 import waitForElementToBeRemoved from './waitForElementToBeRemoved';
 import { within, getQueriesForElement } from './within';
 import { getDefaultNormalizer } from './matches';
 import { renderHook } from './renderHook';
+import { screen } from './screen';
 
 export { act };
 export { cleanup };
 export { fireEvent };
-export { render };
+export { render, RenderResult };
 export { waitFor };
 export { waitForElementToBeRemoved };
 export { within, getQueriesForElement };
 export { getDefaultNormalizer };
 export { renderHook };
+export { screen };
 export type RenderAPI = ReturnType<typeof render>;

src/render.tsx

@@ -6,6 +6,7 @@ import { addToCleanupQueue } from './cleanup';
 import debugShallow from './helpers/debugShallow';
 import debugDeep from './helpers/debugDeep';
 import { getQueriesForElement } from './within';
+import { setRenderResult } from './screen';
 
 type Options = {
   wrapper?: React.ComponentType<any>;
@@ -15,6 +16,8 @@ type TestRendererOptions = {
   createNodeMock: (element: React.ReactElement) => any;
 };
 
+export type RenderResult = ReturnType<typeof render>;
+
 /**
  * Renders test component deeply using react-test-renderer and exposes helpers
  * to assert on the output.
@@ -40,7 +43,7 @@ export default function render<T>(
 
   addToCleanupQueue(unmount);
 
-  return {
+  const result = {
     ...getQueriesForElement(instance),
     update,
     unmount,
@@ -49,6 +52,9 @@ export default function render<T>(
     toJSON: renderer.toJSON,
     debug: debug(instance, renderer),
   };
+
+  setRenderResult(result);
+  return result;
 }
 
 function renderWithAct(

src/screen.ts

@@ -0,0 +1,108 @@
+import { ReactTestInstance } from 'react-test-renderer';
+import { RenderResult } from './render';
+
+const SCREEN_ERROR = '`render` method has not been called';
+
+const notImplemented = () => {
+  throw new Error(SCREEN_ERROR);
+};
+
+const notImplementedDebug = () => {
+  throw new Error(SCREEN_ERROR);
+};
+notImplementedDebug.shallow = notImplemented;
+
+const defaultScreen: RenderResult = {
+  get container(): ReactTestInstance {
+    throw new Error(SCREEN_ERROR);
+  },
+  debug: notImplementedDebug,
+  update: notImplemented,
+  unmount: notImplemented,
+  rerender: notImplemented,
+  toJSON: notImplemented,
+  getByLabelText: notImplemented,
+  getAllByLabelText: notImplemented,
+  queryByLabelText: notImplemented,
+  queryAllByLabelText: notImplemented,
+  findByLabelText: notImplemented,
+  findAllByLabelText: notImplemented,
+  getByA11yHint: notImplemented,
+  getByHintText: notImplemented,
+  getAllByA11yHint: notImplemented,
+  getAllByHintText: notImplemented,
+  queryByA11yHint: notImplemented,
+  queryByHintText: notImplemented,
+  queryAllByA11yHint: notImplemented,
+  queryAllByHintText: notImplemented,
+  findByA11yHint: notImplemented,
+  findByHintText: notImplemented,
+  findAllByA11yHint: notImplemented,
+  findAllByHintText: notImplemented,
+  getByRole: notImplemented,
+  getAllByRole: notImplemented,
+  queryByRole: notImplemented,
+  queryAllByRole: notImplemented,
+  findByRole: notImplemented,
+  findAllByRole: notImplemented,
+  getByA11yStates: notImplemented,
+  getAllByA11yStates: notImplemented,
+  queryByA11yStates: notImplemented,
+  queryAllByA11yStates: notImplemented,
+  findByA11yStates: notImplemented,
+  findAllByA11yStates: notImplemented,
+  getByA11yState: notImplemented,
+  getAllByA11yState: notImplemented,
+  queryByA11yState: notImplemented,
+  queryAllByA11yState: notImplemented,
+  findByA11yState: notImplemented,
+  findAllByA11yState: notImplemented,
+  getByA11yValue: notImplemented,
+  getAllByA11yValue: notImplemented,
+  queryByA11yValue: notImplemented,
+  queryAllByA11yValue: notImplemented,
+  findByA11yValue: notImplemented,
+  findAllByA11yValue: notImplemented,
+  UNSAFE_getByProps: notImplemented,
+  UNSAFE_getAllByProps: notImplemented,
+  UNSAFE_queryByProps: notImplemented,
+  UNSAFE_queryAllByProps: notImplemented,
+  UNSAFE_getByType: notImplemented,
+  UNSAFE_getAllByType: notImplemented,
+  UNSAFE_queryByType: notImplemented,
+  UNSAFE_queryAllByType: notImplemented,
+  getByPlaceholderText: notImplemented,
+  getAllByPlaceholderText: notImplemented,
+  queryByPlaceholderText: notImplemented,
+  queryAllByPlaceholderText: notImplemented,
+  findByPlaceholderText: notImplemented,
+  findAllByPlaceholderText: notImplemented,
+  getByDisplayValue: notImplemented,
+  getAllByDisplayValue: notImplemented,
+  queryByDisplayValue: notImplemented,
+  queryAllByDisplayValue: notImplemented,
+  findByDisplayValue: notImplemented,
+  findAllByDisplayValue: notImplemented,
+  getByTestId: notImplemented,
+  getAllByTestId: notImplemented,
+  queryByTestId: notImplemented,
+  queryAllByTestId: notImplemented,
+  findByTestId: notImplemented,
+  findAllByTestId: notImplemented,
+  getByText: notImplemented,
+  getAllByText: notImplemented,
+  queryByText: notImplemented,
+  queryAllByText: notImplemented,
+  findByText: notImplemented,
+  findAllByText: notImplemented,
+};
+
+export let screen: RenderResult = defaultScreen;
+
+export function setRenderResult(output: RenderResult) {
+  screen = output;
+}
+
+export function clearRenderResult() {
+  screen = defaultScreen;
+}