refactor: wrap <script setup> assertion into isScriptSetup util and u…

…se it

- drop hasSetupState check, if a component is <script setup> i always has setup state
- do note that isScriptSetup and hasSetupState check for different things. I tried to replaced hasSetupScript with isScriptSetup but some specs around expose started to fail so we are required to keep it for now.

src/mount.ts

@@ -34,6 +34,7 @@ import {
   getComponentsFromStubs,
   getDirectivesFromStubs,
   hasSetupState,
+  isScriptSetup,
   isFunctionalComponent,
   isObject,
   isObjectComponent,
@@ -485,7 +486,7 @@ export function mount(
         // Also ensures not to include option API components in this this block
         // since they can also have setup state but need to be patched using
         // the regular method.
-        if (hasSetupState(this) && this.$.setupState.__isScriptSetup) {
+        if (isScriptSetup(this)) {
           // add the mocks to setupState
           for (const [k, v] of Object.entries(
             global.mocks as { [key: string]: any }

src/utils.ts

@@ -185,6 +185,7 @@ export function getDirectivesFromStubs(
       .map(([key, value]) => [key.substring(1), value])
   ) as Record<string, Directive>
 }
+
 export function hasSetupState(
   vm: ComponentPublicInstance
 ): vm is ComponentPublicInstance & {
@@ -195,3 +196,13 @@ export function hasSetupState(
     (vm.$ as unknown as { devtoolsRawSetupState: any }).devtoolsRawSetupState
   )
 }
+
+export function isScriptSetup(
+  vm: ComponentPublicInstance
+): vm is ComponentPublicInstance & {
+  $: { setupState: Record<string, unknown> }
+} {
+  return (
+    vm && (vm.$ as unknown as { setupState: any }).setupState.__isScriptSetup
+  )
+}