feat: add "cleanup" function

Author: aleclarson

src/cleanup.js

@@ -0,0 +1,6 @@
+export default function cleanup() {
+  cleanup.queue.forEach(fn => fn());
+  cleanup.queue.clear();
+}
+
+cleanup.queue = new Set();

src/index.js

@@ -1,16 +1,18 @@
 // @flow
 import act from './act';
-import render from './render';
-import shallow from './shallow';
-import flushMicrotasksQueue from './flushMicrotasksQueue';
+import cleanup from './cleanup';
 import debug from './debug';
 import fireEvent from './fireEvent';
+import flushMicrotasksQueue from './flushMicrotasksQueue';
+import render from './render';
+import shallow from './shallow';
 import waitForElement from './waitForElement';
 
-export { render };
-export { shallow };
-export { flushMicrotasksQueue };
+export { act };
+export { cleanup };
 export { debug };
 export { fireEvent };
+export { flushMicrotasksQueue };
+export { render };
+export { shallow };
 export { waitForElement };
-export { act };

src/render.js

@@ -7,6 +7,7 @@ import { queryByAPI } from './helpers/queryByAPI';
 import a11yAPI from './helpers/a11yAPI';
 import debugShallow from './helpers/debugShallow';
 import debugDeep from './helpers/debugDeep';
+import cleanup from './cleanup';
 
 type Options = {
   wrapper?: React.ComponentType<any>,
@@ -34,6 +35,7 @@ export default function render<T>(
   const update = updateWithAct(renderer, wrap);
   const instance = renderer.root;
 
+  cleanup.queue.add(renderer.unmount);
   return {
     ...getByAPI(instance),
     ...queryByAPI(instance),

typings/index.d.ts

@@ -135,6 +135,7 @@ export declare const shallow: <P = {}>(
   instance: ReactTestInstance | React.ReactElement<P>
 ) => { output: React.ReactElement<P> };
 export declare const flushMicrotasksQueue: () => Promise<any>;
+export declare const cleanup: () => void;
 export declare const debug: DebugAPI;
 export declare const fireEvent: FireEventAPI;
 export declare const waitForElement: WaitForElementFunction;