fix: shallowMount failed for script setup components

Author: rrittwag

src/stubs.ts

@@ -152,8 +152,8 @@ export function stubComponents(
           }
         }
       }
-      if (!name && !shallow) {
-        return args
+      if (!name) {
+        return shallow ? ['stub'] : args
       }
 
       const stub = resolveComponentStubByName(name, stubs)

tests/components/ComponentWithChildren.vue

@@ -3,6 +3,7 @@
     <hello />
     <component-with-input />
     <component-without-name />
+    <script-setup />
     <with-props />
   </div>
 </template>
@@ -11,6 +12,7 @@
 import Hello from './Hello'
 import ComponentWithInput from './ComponentWithInput'
 import ComponentWithoutName from './ComponentWithoutName'
+import ScriptSetup from './ScriptSetup'
 import WithProps from './WithProps'
 
 export default {
@@ -19,9 +21,10 @@ export default {
     Hello,
     ComponentWithInput,
     ComponentWithoutName,
+    ScriptSetup,
     WithProps
   },
-  data () {
+  data() {
     return {}
   }
 }

tests/components/ScriptSetupWithChildren.vue

@@ -0,0 +1,15 @@
+<script setup lang="ts">
+import Hello from './Hello'
+import ComponentWithInput from './ComponentWithInput'
+import ComponentWithoutName from './ComponentWithoutName'
+import ScriptSetup from './ScriptSetup'
+import WithProps from './WithProps'
+</script>
+
+<template>
+  <hello />
+  <component-with-input />
+  <component-without-name />
+  <script-setup />
+  <with-props />
+</template>

tests/shallowMount.spec.ts

@@ -1,6 +1,7 @@
 import { defineComponent } from 'vue'
 import { mount, shallowMount, VueWrapper } from '../src'
 import ComponentWithChildren from './components/ComponentWithChildren.vue'
+import ScriptSetupWithChildren from './components/ScriptSetupWithChildren.vue'
 
 describe('shallowMount', () => {
   it('renders props for stubbed component in a snapshot', () => {
@@ -43,6 +44,7 @@ describe('shallowMount', () => {
         '<hello-stub></hello-stub>' +
         '<component-with-input-stub></component-with-input-stub>' +
         '<component-without-name-stub></component-without-name-stub>' +
+        '<script-setup-stub></script-setup-stub>' +
         '<with-props-stub></with-props-stub>' +
         '</div>'
     )
@@ -55,6 +57,7 @@ describe('shallowMount', () => {
         '<hello-stub></hello-stub>' +
         '<component-with-input-stub></component-with-input-stub>' +
         '<component-without-name-stub></component-without-name-stub>' +
+        '<script-setup-stub></script-setup-stub>' +
         '<with-props-stub></with-props-stub>' +
         '</div>'
     )
@@ -74,8 +77,27 @@ describe('shallowMount', () => {
         '<div>Override</div>' +
         '<component-with-input-stub></component-with-input-stub>' +
         '<component-without-name-stub></component-without-name-stub>' +
+        '<script-setup-stub></script-setup-stub>' +
         '<with-props-stub></with-props-stub>' +
         '</div>'
     )
   })
+
+  it('stubs all components in a script setup component', () => {
+    const wrapper = mount(ScriptSetupWithChildren, {
+      shallow: true,
+      global: {
+        stubs: {
+          Hello: { template: '<div>Override</div>' }
+        }
+      }
+    })
+    expect(wrapper.html()).toEqual(
+      '<div>Override</div>' +
+        '<component-with-input-stub></component-with-input-stub>' +
+        '<stub></stub>' +
+        '<stub></stub>' +
+        '<with-props-stub></with-props-stub>'
+    )
+  })
 })