fix(cleanup): Cleanup should flush microtask queue after unmount (#632)

Author: kamranayub

src/__tests__/cleanup.js

@@ -26,3 +26,17 @@ test('cleanup does not error when an element is not a child', async () => {
   render(<div />, {container: document.createElement('div')})
   await cleanup()
 })
+
+test('cleanup waits for queued microtasks during unmount sequence', async () => {
+  const spy = jest.fn()
+
+  const Test = () => {
+    React.useEffect(() => () => setImmediate(spy))
+
+    return null
+  }
+
+  render(<Test />)
+  await cleanup()
+  expect(spy).toHaveBeenCalledTimes(1)
+})

src/pure.js

@@ -97,8 +97,10 @@ function render(
 }
 
 async function cleanup() {
-  await flush()
   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).