fix(act): ensure that the result is returned from our wrapper

Closes #345

Author: Kent C. Dodds

src/__tests__/act.js

@@ -16,10 +16,8 @@ test('render calls useEffect immediately', () => {
 
 test('findByTestId returns the element', async () => {
   const ref = React.createRef()
-  const {findByTestId, getByTestId} = render(
-    <div ref={ref} data-testid="foo" />,
-  )
-  await expect(findByTestId('foo')).resolves.toEqual(getByTestId('foo'))
+  const {findByTestId} = render(<div ref={ref} data-testid="foo" />)
+  expect(await findByTestId('foo')).toBe(ref.current)
 })
 
 test('fireEvent triggers useEffect calls', () => {

src/act-compat.js

@@ -56,10 +56,9 @@ async function asyncActPolyfill(cb) {
     )
     youHaveBeenWarned = true
   }
-  const result = await cb()
+  await cb()
   // make all effects resolve after
   act(() => {})
-  return result
 }
 
 // istanbul ignore next

src/index.js

@@ -9,7 +9,13 @@ import {
 import act, {asyncAct} from './act-compat'
 
 configureDTL({
-  asyncWrapper: asyncAct,
+  asyncWrapper: async cb => {
+    let result
+    await asyncAct(async () => {
+      result = await cb()
+    })
+    return result
+  },
 })
 
 const mountedContainers = new Set()