Small code cleanup

Author: mpeyper

src/asyncUtils.js

@@ -11,7 +11,7 @@ function asyncUtils(addResolver) {
 
   const waitForNextUpdate = async (options = {}) => {
     if (!nextUpdatePromise) {
-      const resolveOnNextUpdate = (resolve, reject) => {
+      nextUpdatePromise = new Promise((resolve, reject) => {
         let timeoutId
         if (options.timeout > 0) {
           timeoutId = setTimeout(
@@ -24,12 +24,10 @@ function asyncUtils(addResolver) {
           nextUpdatePromise = null
           resolve()
         })
-      }
-
-      nextUpdatePromise = new Promise(resolveOnNextUpdate)
+      })
       await act(() => nextUpdatePromise)
     }
-    return await nextUpdatePromise
+    await nextUpdatePromise
   }
 
   const wait = async (callback, { timeout, suppressErrors = true } = {}) => {

src/pure.js

@@ -75,6 +75,13 @@ function renderHook(callback, { initialProps, wrapper } = {}) {
   })
   const { unmount, update } = testRenderer
 
+  function rerenderHook(newProps = hookProps.current) {
+    hookProps.current = newProps
+    act(() => {
+      update(toRender())
+    })
+  }
+
   function unmountHook() {
     act(() => {
       removeCleanup(unmountHook)
@@ -86,12 +93,7 @@ function renderHook(callback, { initialProps, wrapper } = {}) {
 
   return {
     result,
-    rerender: (newProps = hookProps.current) => {
-      hookProps.current = newProps
-      act(() => {
-        update(toRender())
-      })
-    },
+    rerender: rerenderHook,
     unmount: unmountHook,
     ...asyncUtils(addResolver)
   }