feat: support async act

src/pure.ts

@@ -7,15 +7,15 @@ import ReactDOMClient from 'react-dom/client'
 // we call act only when rendering to flush any possible effects
 // usually the async nature of Vitest browser mode ensures consistency,
 // but rendering is sync and controlled by React directly
-function act(cb: () => unknown) {
+async function act(cb: () => unknown) {
   // @ts-expect-error unstable_act is not typed, but exported
   const _act = React.act || React.unstable_act
   if (typeof _act !== 'function') {
     cb()
   }
   else {
     (globalThis as any).IS_REACT_ACT_ENVIRONMENT = true
-    _act(cb)
+    await _act(cb)
     ;(globalThis as any).IS_REACT_ACT_ENVIRONMENT = false
   }
 }
@@ -28,8 +28,8 @@ export interface RenderResult extends LocatorSelectors {
     maxLength?: number,
     options?: PrettyDOMOptions
   ) => void
-  unmount: () => void
-  rerender: (ui: React.ReactNode) => void
+  unmount: () => Promise<void>
+  rerender: (ui: React.ReactNode) => Promise<void>
   asFragment: () => DocumentFragment
 }
 
@@ -47,10 +47,10 @@ const mountedRootEntries: {
   root: ReturnType<typeof createConcurrentRoot>
 }[] = []
 
-export function render(
+export async function render(
   ui: React.ReactNode,
   { container, baseElement, wrapper: WrapperComponent }: ComponentRenderOptions = {},
-): RenderResult {
+): Promise<RenderResult> {
   if (!baseElement) {
     // default to document.body instead of documentElement to avoid output of potentially-large
     // head elements (such as JSS style blocks) in debug output
@@ -83,7 +83,7 @@ export function render(
     })
   }
 
-  act(() => {
+  await act(() => {
     root!.render(
       strictModeIfNeeded(wrapUiIfNeeded(ui, WrapperComponent)),
     )
@@ -93,13 +93,13 @@ export function render(
     container,
     baseElement,
     debug: (el, maxLength, options) => debug(el, maxLength, options),
-    unmount: () => {
-      act(() => {
+    unmount: async () => {
+      await act(() => {
         root.unmount()
       })
     },
-    rerender: (newUi: React.ReactNode) => {
-      act(() => {
+    rerender: async (newUi: React.ReactNode) => {
+      await act(() => {
         root.render(
           strictModeIfNeeded(wrapUiIfNeeded(newUi, WrapperComponent)),
         )
@@ -112,9 +112,9 @@ export function render(
   }
 }
 
-export function cleanup(): void {
-  mountedRootEntries.forEach(({ root, container }) => {
-    act(() => {
+export async function cleanup(): Promise<void> {
+  mountedRootEntries.forEach(async ({ root, container }) => {
+    await act(() => {
       root.unmount()
     })
     if (container.parentNode === document.body) {

test/fixtures/HelloWorld.tsx

@@ -1,3 +1,7 @@
-export function HelloWorld(): React.ReactElement {
-  return <div>Hello World</div>
+export function HelloWorld({
+  name = 'World',
+}: {
+  name?: string
+}): React.ReactElement {
+  return <div>{`Hello ${name}`}</div>
 }

test/render.test.tsx

@@ -1,19 +1,29 @@
+import { Suspense } from 'react'
 import { expect, test } from 'vitest'
 import { page } from '@vitest/browser/context'
 import { render } from '../src/index'
 import { HelloWorld } from './fixtures/HelloWorld'
 import { Counter } from './fixtures/Counter'
 
 test('renders simple component', async () => {
-  const screen = render(<HelloWorld />)
+  const screen = await render(<HelloWorld />)
   await expect.element(page.getByText('Hello World')).toBeVisible()
   expect(screen.container).toMatchSnapshot()
 })
 
 test('renders counter', async () => {
-  const screen = render(<Counter initialCount={1} />)
+  const screen = await render(<Counter initialCount={1} />)
 
   await expect.element(screen.getByText('Count is 1')).toBeVisible()
   await screen.getByRole('button', { name: 'Increment' }).click()
   await expect.element(screen.getByText('Count is 2')).toBeVisible()
 })
+
+test('renders child component on mount for suspense boundary which is not suspending', async () => {
+  const { getByText } = await render(<HelloWorld name="Vitest" />, {
+    wrapper: ({ children }) => (
+      <Suspense fallback={<div>Suspended!</div>}>{children}</Suspense>
+    ),
+  })
+  await expect.element(getByText('Hello Vitest')).toBeInTheDocument()
+})