From d12206db8efac33401399b7f38e7894b24b3a0f4 Mon Sep 17 00:00:00 2001
From: Evan You <yyx990803@gmail.com>
Date: Tue, 6 Jul 2021 11:39:27 -0400
Subject: [PATCH] test: add test case for proper effect teardown w/
 withAsyncContext

---
 .../__tests__/apiSetupHelpers.spec.ts         | 34 ++++++++++++++++++-
 1 file changed, 33 insertions(+), 1 deletion(-)

diff --git a/packages/runtime-core/__tests__/apiSetupHelpers.spec.ts b/packages/runtime-core/__tests__/apiSetupHelpers.spec.ts
index cf698d7b..5244a9a8 100644
--- a/packages/runtime-core/__tests__/apiSetupHelpers.spec.ts
+++ b/packages/runtime-core/__tests__/apiSetupHelpers.spec.ts
@@ -9,7 +9,9 @@ import {
   render,
   serializeInner,
   SetupContext,
-  Suspense
+  Suspense,
+  computed,
+  ComputedRef
 } from '@vue/runtime-test'
 import {
   defineEmits,
@@ -253,5 +255,35 @@ describe('SFC <script setup> helpers', () => {
       await ready
       expect(getCurrentInstance()).toBeNull()
     })
+
+    test('should teardown in-scope effects', async () => {
+      let resolve: (val?: any) => void
+      const ready = new Promise(r => {
+        resolve = r
+      })
+
+      let c: ComputedRef
+
+      const Comp = defineComponent({
+        async setup() {
+          await withAsyncContext(Promise.resolve())
+
+          c = computed(() => {})
+          // register the lifecycle after an await statement
+          onMounted(resolve)
+          return () => ''
+        }
+      })
+
+      const app = createApp(() => h(Suspense, () => h(Comp)))
+      const root = nodeOps.createElement('div')
+      app.mount(root)
+
+      await ready
+      expect(c!.effect.active).toBe(true)
+
+      app.unmount()
+      expect(c!.effect.active).toBe(false)
+    })
   })
 })