Moved cleanup logic into seperate file

Author: mpeyper

src/cleanup.js

@@ -0,0 +1,26 @@
+import { act } from 'react-test-renderer'
+
+let cleanupCallbacks = []
+
+async function cleanup() {
+  await act(async () => {})
+  cleanupCallbacks.forEach((cb) => cb())
+  cleanupCallbacks = []
+}
+
+function addCleanup(callback) {
+  cleanupCallbacks.push(callback)
+}
+
+function removeCleanup(callback) {
+  cleanupCallbacks = cleanupCallbacks.filter((cb) => cb !== callback)
+}
+
+// Automatically registers cleanup in supported testing frameworks
+if (typeof afterEach === 'function' && !process.env.RHTL_SKIP_AUTO_CLEANUP) {
+  afterEach(async () => {
+    await cleanup()
+  })
+}
+
+export { cleanup, addCleanup, removeCleanup }

src/index.js

@@ -1,7 +1,6 @@
 import React, { Suspense } from 'react'
 import { act, create } from 'react-test-renderer'
-
-let cleanupCallbacks = []
+import { cleanup, addCleanup, removeCleanup } from './cleanup'
 
 function TestHook({ callback, hookProps, onError, children }) {
   try {
@@ -77,12 +76,12 @@ function renderHook(callback, { initialProps, wrapper } = {}) {
 
   function unmountHook() {
     act(() => {
-      cleanupCallbacks = cleanupCallbacks.filter((cb) => cb !== unmountHook)
+      removeCleanup(unmountHook)
       unmount()
     })
   }
 
-  cleanupCallbacks.push(unmountHook)
+  addCleanup(unmountHook)
 
   let waitingForNextUpdate = null
   const resolveOnNextUpdate = (resolve) => {
@@ -108,18 +107,4 @@ function renderHook(callback, { initialProps, wrapper } = {}) {
   }
 }
 
-async function cleanup() {
-  await act(async () => {
-    await act(async () => {})
-    cleanupCallbacks.forEach((cb) => cb())
-    cleanupCallbacks = []
-  })
-}
-
-if (typeof afterEach === 'function' && !process.env.RHTL_SKIP_AUTO_CLEANUP) {
-  afterEach(async () => {
-    await cleanup()
-  })
-}
-
 export { renderHook, cleanup, act }