From: Evan You <yyx990803@gmail.com>
Date: Fri, 6 Aug 2021 17:01:39 +0000 (-0400)
Subject: test: improve test case for 865b84bf
X-Git-Tag: v3.2.0-beta.8~5
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=1994f1200d12c05e509a812f2455de9da101c649;p=thirdparty%2Fvuejs%2Fcore.git

test: improve test case for 865b84bf
---

diff --git a/packages/compiler-sfc/__tests__/__snapshots__/compileScriptRefSugar.spec.ts.snap b/packages/compiler-sfc/__tests__/__snapshots__/compileScriptRefSugar.spec.ts.snap
index 6dc4229ce2..32acabe3db 100644
--- a/packages/compiler-sfc/__tests__/__snapshots__/compileScriptRefSugar.spec.ts.snap
+++ b/packages/compiler-sfc/__tests__/__snapshots__/compileScriptRefSugar.spec.ts.snap
@@ -99,11 +99,18 @@ export default _defineComponent({
   setup(__props, { expose }) {
   expose()
 
-        let n = _ref<number | undefined>()
-        console.log(n.value!)
-        console.log(n.value as number)
+      let a = _ref(1)
+      console.log(a.value!) 
+      console.log(a.value! + 1) 
+      console.log(a.value as number) 
+      console.log((a.value as number) + 1) 
+      console.log(<number>a.value) 
+      console.log(<number>a.value + 1) 
+      console.log(a.value! + (a.value as number)) 
+      console.log(a.value! + <number>a.value) 
+      console.log((a.value as number) + <number>a.value)
       
-return { n }
+return { a }
 }
 
 })"
diff --git a/packages/compiler-sfc/__tests__/compileScriptRefSugar.spec.ts b/packages/compiler-sfc/__tests__/compileScriptRefSugar.spec.ts
index d479dbbf9f..59964d8510 100644
--- a/packages/compiler-sfc/__tests__/compileScriptRefSugar.spec.ts
+++ b/packages/compiler-sfc/__tests__/compileScriptRefSugar.spec.ts
@@ -286,17 +286,25 @@ describe('<script setup> ref sugar', () => {
     const { content } = compile(
       `
       <script setup lang="ts">
-        let n = $ref<number | undefined>()
-        console.log(n!)
-        console.log(n as number)
+      let a = $ref(1)
+      console.log(a!) 
+      console.log(a! + 1) 
+      console.log(a as number) 
+      console.log((a as number) + 1) 
+      console.log(<number>a) 
+      console.log(<number>a + 1) 
+      console.log(a! + (a as number)) 
+      console.log(a! + <number>a) 
+      console.log((a as number) + <number>a)
       </script>`,
       {
         refSugar: true
       }
     )
     assertCode(content)
-    expect(content).toMatch('console.log(n.value!)')
-    expect(content).toMatch('console.log(n.value as number)')
+    expect(content).toMatch('console.log(a.value!)')
+    expect(content).toMatch('console.log(a.value as number)')
+    expect(content).toMatch('console.log(<number>a.value)')
   })
 
   describe('errors', () => {
diff --git a/packages/compiler-sfc/src/compileScript.ts b/packages/compiler-sfc/src/compileScript.ts
index bef32c8fe7..320c00b521 100644
--- a/packages/compiler-sfc/src/compileScript.ts
+++ b/packages/compiler-sfc/src/compileScript.ts
@@ -1785,7 +1785,8 @@ export function walkIdentifiers(
         parent &&
         parent.type.startsWith('TS') &&
         parent.type !== 'TSAsExpression' &&
-        parent.type !== 'TSNonNullExpression'
+        parent.type !== 'TSNonNullExpression' &&
+        parent.type !== 'TSTypeAssertion'
       ) {
         return this.skip()
       }