fix(runtime-utils): don't stub helpers when shallow (#632)

nuxt · Dec 5, 2023 · 55cbda2 · 55cbda2
1 parent 92f6a65
commit 55cbda2
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 9 deletions.
diff --git a/examples/app-vitest-full/tests/nuxt/index.spec.ts b/examples/app-vitest-full/tests/nuxt/index.spec.ts
@@ -58,6 +58,16 @@ describe('test utils', () => {
     `)
   })
 
+  it('should work with shallow mounting within suspense', async () => {
+    const component = await mountSuspended(App, { shallow: true })
+    expect(component.html()).toMatchInlineSnapshot(`
+      "<async-component-wrapper-stub></async-component-wrapper-stub>
+      <anonymous-stub></anonymous-stub>
+      <nuxt-page-stub></nuxt-page-stub>
+      <nuxt-link-stub to="/test"></nuxt-link-stub>"
+    `)
+  })
+
   it('should render default props within nuxt suspense', async () => {
     const component = await mountSuspended(OptionsComponent)
     expect(component.find('h2').html()).toMatchInlineSnapshot(

diff --git a/src/runtime-utils/mount.ts b/src/runtime-utils/mount.ts
@@ -94,12 +94,14 @@ export async function mountSuspended<T>(
               {
                 default: () =>
                   h({
+                    name: 'MountSuspendedHelper',
                     async setup() {
                       const router = useRouter()
                       await router.replace(route)
 
                       // Proxy top-level setup/render context so test wrapper resolves child component
                       const clonedComponent = {
+                        name: 'MountSuspendedComponent',
                         ...component,
                         render: render
                           ? (_ctx: any, ...args: any[]) => {
@@ -127,16 +129,24 @@ export async function mountSuspended<T>(
               }
             ),
         },
-        defu(_options, {
-          slots,
-          global: {
-            config: {
-              globalProperties: vueApp.config.globalProperties,
+        defu(
+          _options,
+          {
+            slots,
+            global: {
+              config: {
+                globalProperties: vueApp.config.globalProperties,
+              },
+              provide: vueApp._context.provides,
+              stubs: {
+                Suspense: false,
+                MountSuspendedHelper: false,
+                [typeof (component as any).name === 'string' ? (component as any).name : 'MountSuspendedComponent']: false
+              },
+              components: { RouterLink },
             },
-            provide: vueApp._context.provides,
-            components: { RouterLink },
-          },
-        } satisfies ComponentMountingOptions<T>) as ComponentMountingOptions<T>
+          } satisfies ComponentMountingOptions<T>
+        ) as ComponentMountingOptions<T>
       )
     }
   )