feat: expose setupState from mountSuspended

packages/vitest-environment-nuxt/src/runtime/mount.ts

@@ -59,6 +59,18 @@ export async function mountSuspended<T>(
   const { render, setup } = component as DefineComponent<any, any>
 
   let setupContext: SetupContext
+  let setupState: any
+
+  const wrappedSetup = async (
+    props: Record<string, any>,
+    setupContext: SetupContext
+  ) => {
+    if (setup) {
+      setupState = await setup(props, setupContext)
+      return setupState
+    }
+  }
+
   return new Promise<ReturnType<typeof mount<T>>>(resolve => {
     const vm = mount(
       {
@@ -72,7 +84,13 @@ export async function mountSuspended<T>(
         render: (renderContext: any) =>
           h(
             Suspense,
-            { onResolve: () => nextTick().then(() => resolve(vm as any)) },
+            {
+              onResolve: () =>
+                nextTick().then(() => {
+                  ;(vm as any).setupState = setupState
+                  resolve(vm as any)
+                }),
+            },
             {
               default: () =>
                 h({
@@ -95,7 +113,7 @@ export async function mountSuspended<T>(
                         : undefined,
                       setup: setup
                         ? (props: Record<string, any>) =>
-                            setup(props, setupContext)
+                            wrappedSetup(props, setupContext)
                         : undefined,
                     }
 

playground/tests/nuxt/mock-component-expose-setup-state.spec.ts

@@ -0,0 +1,9 @@
+import { expect, it } from 'vitest'
+import { mountSuspended } from 'vitest-environment-nuxt/utils'
+import MockComponent from './mocks/MockComponent.vue'
+
+it('should expose setup state', async () => {
+  const component = await mountSuspended(MockComponent)
+  expect(component.setupState.counter.value).toBe(1)
+  expect(component.setupState.doubled.value).toBe(2)
+})