feat: Use React scheduler to flush instead of custom implementation

Author: eps1lon

Diff for: src/__tests__/cleanup.js

@@ -1,7 +1,7 @@
 import React from 'react'
-import {render, cleanup, act} from '../'
+import {render, cleanup} from '../'
 
-test('cleans up the document', async () => {
+test('cleans up the document', () => {
   const spy = jest.fn()
   const divId = 'my-div'
 
@@ -17,17 +17,17 @@ test('cleans up the document', async () => {
   }
 
   render(<Test />)
-  await cleanup()
+  cleanup()
   expect(document.body).toBeEmptyDOMElement()
   expect(spy).toHaveBeenCalledTimes(1)
 })
 
-test('cleanup does not error when an element is not a child', async () => {
+test('cleanup does not error when an element is not a child', () => {
   render(<div />, {container: document.createElement('div')})
-  await cleanup()
+  cleanup()
 })
 
-test('cleanup runs effect cleanup functions', async () => {
+test('cleanup runs effect cleanup functions', () => {
   const spy = jest.fn()
 
   const Test = () => {
@@ -37,7 +37,7 @@ test('cleanup runs effect cleanup functions', async () => {
   }
 
   render(<Test />)
-  await cleanup()
+  cleanup()
   expect(spy).toHaveBeenCalledTimes(1)
 })
 
@@ -54,7 +54,7 @@ describe('fake timers and missing act warnings', () => {
     jest.useRealTimers()
   })
 
-  test('cleanup does not flush immediates', async () => {
+  test('cleanup does not flush immediates', () => {
     const microTaskSpy = jest.fn()
     function Test() {
       const counter = 1
@@ -77,15 +77,15 @@ describe('fake timers and missing act warnings', () => {
     }
     render(<Test />)
 
-    await cleanup()
+    cleanup()
 
     expect(microTaskSpy).toHaveBeenCalledTimes(0)
     // console.error is mocked
     // eslint-disable-next-line no-console
     expect(console.error).toHaveBeenCalledTimes(0)
   })
 
-  test('cleanup does not swallow missing act warnings', async () => {
+  test('cleanup does not swallow missing act warnings', () => {
     const deferredStateUpdateSpy = jest.fn()
     function Test() {
       const counter = 1
@@ -109,7 +109,7 @@ describe('fake timers and missing act warnings', () => {
     render(<Test />)
 
     jest.runAllImmediates()
-    await cleanup()
+    cleanup()
 
     expect(deferredStateUpdateSpy).toHaveBeenCalledTimes(1)
     // console.error is mocked

Diff for: src/pure.js

@@ -7,7 +7,6 @@ import {
 } from '@testing-library/dom'
 import act, {asyncAct} from './act-compat'
 import {fireEvent} from './fire-event'
-import flush from './flush-microtasks'
 
 configureDTL({
   asyncWrapper: async cb => {
@@ -100,17 +99,16 @@ function render(
   }
 }
 
-async function cleanup() {
+function cleanup() {
   mountedContainers.forEach(cleanupAtContainer)
-  // flush microtask queue after unmounting in case
-  // unmount sequence generates new microtasks
-  await flush()
 }
 
 // maybe one day we'll expose this (perhaps even as a utility returned by render).
 // but let's wait until someone asks for it.
 function cleanupAtContainer(container) {
-  ReactDOM.unmountComponentAtNode(container)
+  act(() => {
+    ReactDOM.unmountComponentAtNode(container)
+  })
   if (container.parentNode === document.body) {
     document.body.removeChild(container)
   }