feat(cleanup): add after each cleanup helper

Author: bcarroll22

cleanup-after-each.js

@@ -0,0 +1 @@
+afterEach(require('./dist').cleanup);

other/cheat-sheet.pdf

@@ -2,7 +2,9 @@ import React from 'react';
 import 'jest-native/extend-expect';
 import { Button } from 'react-native';
 
-import { render, fireEvent } from '../';
+import { render, fireEvent, cleanup } from '../';
+
+afterEach(cleanup);
 
 test('render calls useEffect immediately', () => {
   const effectCb = jest.fn();

src/__tests__/act.js

@@ -1,7 +1,9 @@
 import React from 'react';
 import { Text, View } from 'react-native';
 
-import { render, queryAllByProp } from '../';
+import { render, queryAllByProp, cleanup } from '../';
+
+afterEach(cleanup);
 
 // This is to ensure custom queries can be passed to render. In most cases, you
 // wouldn't/shouldn't need to do this, but we do allow it so we'll test to

src/__tests__/bugs.js

@@ -1,13 +1,14 @@
 import React from 'react';
 import { Text } from 'react-native';
 
-import { render } from '../';
+import { cleanup, render } from '../';
 
 beforeEach(() => {
   jest.spyOn(console, 'log').mockImplementation(() => {});
 });
 
 afterEach(() => {
+  cleanup();
   console.log.mockRestore();
 });
 

src/__tests__/debug.js

@@ -2,7 +2,9 @@ import React from 'react';
 import 'jest-native/extend-expect';
 import { Text } from 'react-native';
 
-import { render, wait } from '../';
+import { cleanup, render, wait } from '../';
+
+afterEach(cleanup);
 
 const fetchAMessage = () =>
   new Promise(resolve => {

src/__tests__/end-to-end.js

@@ -2,7 +2,9 @@ import React from 'react';
 import 'jest-native/extend-expect';
 import { Button, Image, Text, TextInput, TouchableHighlight } from 'react-native';
 
-import { render, fireEvent, eventMap, NativeTestEvent, getEventHandlerName, wait } from '../';
+import { render, fireEvent, eventMap, getEventHandlerName, wait, cleanup } from '../';
+
+afterEach(cleanup);
 
 Object.keys(eventMap).forEach(key => {
   describe(`${key} events`, () => {

src/__tests__/events.js

@@ -2,7 +2,9 @@ import React from 'react';
 import 'jest-native/extend-expect';
 import { TouchableOpacity, Text, View } from 'react-native';
 
-import { render, fireEvent, wait } from '../';
+import { render, fireEvent, wait, cleanup } from '../';
+
+afterEach(cleanup);
 
 global.fetch = require('jest-fetch-mock');
 

src/__tests__/fetch.js

@@ -1,6 +1,8 @@
 import React from 'react';
 import { Button, TextInput, View } from 'react-native';
-import { render, fireEvent } from '../';
+import { render, fireEvent, cleanup } from '../';
+
+afterEach(cleanup);
 
 function Login({ onSubmit, user }) {
   return (

src/__tests__/forms.js

@@ -2,7 +2,9 @@ import React from 'react';
 import { Button, Picker, Text, View } from 'react-native';
 import { toMatchDiffSnapshot } from 'snapshot-diff';
 
-import { fireEvent, render } from '../';
+import { cleanup, fireEvent, render } from '../';
+
+afterEach(cleanup);
 
 test('<Picker /> works', () => {
   function Wrapper() {

src/__tests__/misc.js

@@ -1,7 +1,9 @@
 import React from 'react';
 import { Text, SafeAreaView, View } from 'react-native';
 
-import { render, toJSON } from '../';
+import { cleanup, render } from '../';
+
+afterEach(cleanup);
 
 test('renders View', () => {
   const { container } = render(<View />);

src/__tests__/render.js

@@ -2,7 +2,9 @@ import React from 'react';
 import 'jest-native/extend-expect';
 import { Text } from 'react-native';
 
-import { render } from '../';
+import { cleanup, render } from '../';
+
+afterEach(cleanup);
 
 test('rerender will re-render the element', () => {
   const Greeting = props => <Text>{props.message}</Text>;

src/__tests__/rerender.js

@@ -1,7 +1,9 @@
 import React from 'react';
 import { Button, Text, View } from 'react-native';
 
-import { render, fireEvent, prettyPrint } from '../';
+import { render, fireEvent, prettyPrint, cleanup } from '../';
+
+afterEach(cleanup);
 
 class StopWatch extends React.Component {
   state = { lapse: 0, running: false };

src/__tests__/stopwatch.js

@@ -8,10 +8,12 @@ import {
   getQueriesForElement,
   NativeTestEvent,
   prettyPrint,
-  proxyElement as proxy,
+  proxyElement,
 } from './lib';
 import act from './act-compat';
 
+const renderers = new Set();
+
 function render(ui, { options = {}, wrapper: WrapperComponent, queries } = {}) {
   const wrapUiIfNeeded = innerElement =>
     WrapperComponent ? (
@@ -28,7 +30,9 @@ function render(ui, { options = {}, wrapper: WrapperComponent, queries } = {}) {
     testRenderer = TR.create(wrapUiIfNeeded(ui), options);
   });
 
-  const wrappers = proxy(testRenderer.root).findAll(n => n.getProp('pointerEvents') === 'box-none');
+  renderers.add(testRenderer);
+
+  const wrappers = proxyElement(testRenderer.root).findAll(n => n.type === 'View');
   const baseElement = wrappers[0]; // Includes YellowBox and your render
   const container = wrappers[1]; // Includes only your render
 
@@ -49,6 +53,15 @@ function render(ui, { options = {}, wrapper: WrapperComponent, queries } = {}) {
   };
 }
 
+function cleanup() {
+  renderers.forEach(cleanupRenderer);
+}
+
+function cleanupRenderer(renderer) {
+  renderer.unmount();
+  renderers.delete(renderer);
+}
+
 function fireEvent(...args) {
   let returnValue;
   act(() => {
@@ -68,4 +81,4 @@ Object.keys(rntlFireEvent).forEach(typeArg => {
 });
 
 export * from './lib';
-export { act, fireEvent, render, NativeTestEvent };
+export { act, cleanup, fireEvent, render, NativeTestEvent };

src/index.js

@@ -2,7 +2,9 @@ import React from 'react';
 import { Button, Text, TextInput, View } from 'react-native';
 import cases from 'jest-in-case';
 
-import { render } from '../../';
+import { cleanup, render } from '../../';
+
+afterEach(cleanup);
 
 cases(
   'getBy* queries throw an error when there are multiple elements returned',

src/lib/__tests__/get-by-errors.js

@@ -1,7 +1,9 @@
 import React from 'react';
-import { Text, View } from 'react-native';
+import { View } from 'react-native';
 
-import { render, queryByProp, queryByTestId } from '../../';
+import { render, queryByProp, queryByTestId, cleanup } from '../../';
+
+afterEach(cleanup);
 
 test('queryByProp', () => {
   const { container } = render(

src/lib/__tests__/misc.js

@@ -1,7 +1,9 @@
 import React from 'react';
 import { Text, View } from 'react-native';
 
-import { render, prettyPrint } from '../../';
+import { render, prettyPrint, cleanup } from '../../';
+
+afterEach(cleanup);
 
 test('it prints correctly with no children', () => {
   const { container } = render(<View />);

src/lib/__tests__/pretty-print.js

@@ -1,7 +1,9 @@
 import React from 'react';
 import { Button, Image, Text, TextInput, View } from 'react-native';
 
-import { render } from '../../';
+import { cleanup, render } from '../../';
+
+afterEach(cleanup);
 
 test('find asynchronously finds elements', async () => {
   const {

src/lib/__tests__/queries.find.js

@@ -2,7 +2,9 @@ import React from 'react';
 import cases from 'jest-in-case';
 import { Button, Image, Text, TextInput, TouchableOpacity } from 'react-native';
 
-import { getDefaultNormalizer, render } from '../../';
+import { cleanup, getDefaultNormalizer, render } from '../../';
+
+afterEach(cleanup);
 
 cases(
   'matches find case-sensitive full strings by default',

src/lib/__tests__/text-matchers.js

@@ -1,7 +1,9 @@
 import React from 'react';
 import { Text, View } from 'react-native';
 
-import { prettyPrint, render, toJSON } from '../../';
+import { cleanup, prettyPrint, render, toJSON } from '../../';
+
+afterEach(cleanup);
 
 test('it converts to json', () => {
   function ParentComponent({ children }) {

src/lib/__tests__/to-json.js

@@ -1,7 +1,9 @@
 import React from 'react';
 import { View } from 'react-native';
 
-import { render, waitForElementToBeRemoved } from '../../';
+import { cleanup, render, waitForElementToBeRemoved } from '../../';
+
+afterEach(cleanup);
 
 jest.useFakeTimers();
 

src/lib/__tests__/wait-for-element-to-be-removed.fake-timers.js

@@ -1,7 +1,9 @@
 import React from 'react';
 import { View } from 'react-native';
 
-import { render, waitForElementToBeRemoved } from '../../';
+import { cleanup, render, waitForElementToBeRemoved } from '../../';
+
+afterEach(cleanup);
 
 test('resolves only when the element is removed', async () => {
   class MutatedElement extends React.Component {

src/lib/__tests__/wait-for-element-to-be-removed.js

@@ -1,7 +1,9 @@
 import React from 'react';
 import { Text, View } from 'react-native';
 
-import { render, waitForElement } from '../../';
+import { cleanup, render, waitForElement } from '../../';
+
+afterEach(cleanup);
 
 test('waits for element to appear', async () => {
   const { rerender, getByTestId } = render(<View />);

src/lib/__tests__/wait-for-element.js

@@ -1,7 +1,9 @@
 import React from 'react';
 import { Button, View } from 'react-native';
 
-import { render, within } from '../../';
+import { cleanup, render, within } from '../../';
+
+afterEach(cleanup);
 
 test('it works when scoping to a smaller set of elements', () => {
   const { getByTestId } = render(

src/lib/__tests__/within.js

@@ -1,6 +1,10 @@
 import { asyncAct } from '../act-compat';
+import { NativeTestEvent } from '../lib/events';
 import { configure as configureNTL } from '../lib';
 
+// Make this global for convenience, just like browser events
+global.NativeTestEvent = NativeTestEvent;
+
 configureNTL({
   asyncWrapper: async cb => {
     let result;