fixup! Added cleanup method

Author: apostolidhs

README.md

@@ -126,12 +126,12 @@ npm install --save-dev react-test-renderer@^x.y.z
 Both of these dependecies must be installed as at least version `16.8.0` to be compatible with
 `react-hooks-testing-library`.
 
-### `cleanup()`
+### `unmountAll()`
 
 Unmounts any of the Hooks that were mounted with `renderHook`.
 
-Optionally, it is possible to import `cleanup` in a global test file. Using that way, it isn't
-necessary to run `afterEach(cleanup)` on every test script.
+Optionally, it is possible to import `unmountAll` in a global test file. Using that way, it isn't
+necessary to run `afterEach(unmountAll)` on every test script.
 
 ```js
 import 'react-hooks-testing-library/cleanup-after-each'

cleanup-after-each.js

@@ -1 +1 @@
-afterEach(require('./src').cleanup)
+afterEach(require('./lib').unmountAll)

src/index.js

@@ -1,7 +1,7 @@
 import React, { Suspense } from 'react'
 import { act, create } from 'react-test-renderer'
 
-const unmounts = []
+let unmounts = []
 
 function TestHook({ callback, hookProps, onError, children }) {
   try {
@@ -53,9 +53,8 @@ function resultContainer() {
   }
 }
 
-function cleanup() {
+function unmountAll() {
   unmounts.forEach((unmount) => unmount())
-  unmounts.length = 0
 }
 
 function renderHook(callback, { initialProps, wrapper } = {}) {
@@ -82,6 +81,7 @@ function renderHook(callback, { initialProps, wrapper } = {}) {
 
   function unmountHook() {
     act(() => {
+      unmounts = unmounts.filter((u) => u !== unmountHook)
       unmount()
     })
   }
@@ -101,4 +101,4 @@ function renderHook(callback, { initialProps, wrapper } = {}) {
   }
 }
 
-export { renderHook, cleanup, act }
+export { renderHook, unmountAll, act }

test/cleanup.test.js renamed to test/unmountAll.test.js

@@ -1,7 +1,7 @@
 import { useEffect } from 'react'
-import { renderHook, cleanup } from 'src'
+import { renderHook, unmountAll } from 'src'
 
-describe('cleanup tests', () => {
+describe('unmountAll tests', () => {
   let sideEffect = {}
 
   function useEffectsCounter({ initialProps }) {
@@ -25,24 +25,24 @@ describe('cleanup tests', () => {
     useEffectsCounter({ initialProps: { id: 10 } })
     useEffectsCounter({ initialProps: { id: 100 } })
 
-    cleanup()
+    unmountAll()
 
     expect(sideEffect).toEqual({ 1: 1, 10: 1, 100: 1 })
   })
 
-  test('should not cleanup a hook that have already unmounted', () => {
+  test('should not unmountAll a hook that have already unmounted', () => {
     const { unmount } = useEffectsCounter({ initialProps: { id: 1 } })
 
     unmount()
-    cleanup()
+    unmountAll()
 
     expect(sideEffect).toEqual({ 1: 1 })
   })
 
-  test('should not unmount a hook that have already cleaned up', () => {
+  test('should not unmount a hook that have already unmounted up', () => {
     const { unmount } = useEffectsCounter({ initialProps: { id: 1 } })
 
-    cleanup()
+    unmountAll()
     unmount()
 
     expect(sideEffect).toEqual({ 1: 1 })

typings/index.d.ts

@@ -23,4 +23,4 @@ export function renderHook<P, R>(
   options?: RenderHookOptions<P>
 ): RenderHookResult<P, R>
 
-export function cleanup(): void
+export function unmountAll(): void