From 91b27e0eb707cd068999eb89a11cdd5a6bd846c8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Renato=20de=20Le=C3=A3o?= <renatodeleao@gmail.com>
Date: Wed, 4 Jan 2023 10:08:17 +0000
Subject: [PATCH] refactor: wrap <script setup> assertion into isScriptSetup
 util and use 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 |  4 ++--
 src/utils.ts | 11 +++++++++++
 2 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/src/mount.ts b/src/mount.ts
index bff98a23e..d4cb996a0 100644
--- a/src/mount.ts
+++ b/src/mount.ts
@@ -33,7 +33,7 @@ import { MountingOptions, Slot } from './types'
 import {
   getComponentsFromStubs,
   getDirectivesFromStubs,
-  hasSetupState,
+  isScriptSetup,
   isFunctionalComponent,
   isObject,
   isObjectComponent,
@@ -485,7 +485,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 }
diff --git a/src/utils.ts b/src/utils.ts
index be0cad37b..058124ec6 100644
--- a/src/utils.ts
+++ b/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
+  )
+}