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

* failing test

* act-compat change

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

Closes #345
Closes #347

Author: joual

src/__tests__/act.js

@@ -14,6 +14,12 @@ test('render calls useEffect immediately', () => {
   expect(effectCb).toHaveBeenCalledTimes(1)
 })
 
+test('findByTestId returns the element', async () => {
+  const ref = React.createRef()
+  const {findByTestId} = render(<div ref={ref} data-testid="foo" />)
+  expect(await findByTestId('foo')).toBe(ref.current)
+})
+
 test('fireEvent triggers useEffect calls', () => {
   const effectCb = jest.fn()
   function Counter() {

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()