From a3a58e15223a5d8694d1315c3056bea7f1a9793f Mon Sep 17 00:00:00 2001
From: Illya Klymov <xanf@xanf.me>
Date: Thu, 3 Nov 2022 08:36:59 +0200
Subject: [PATCH] chore(mount): simplify mount operation

* do not use ugly console.warn hack
---
 src/mount.ts               | 15 +++++++++------
 tests/shallowMount.spec.ts |  2 +-
 2 files changed, 10 insertions(+), 7 deletions(-)

diff --git a/src/mount.ts b/src/mount.ts
index 31ddda39dc..2f9c7bdc0b 100644
--- a/src/mount.ts
+++ b/src/mount.ts
@@ -25,7 +25,8 @@ import {
   ComponentOptions,
   ConcreteComponent,
   Prop,
-  transformVNodeArgs
+  transformVNodeArgs,
+  ref
 } from 'vue'
 
 import { MountingOptions, Slot } from './types'
@@ -432,9 +433,15 @@ export function mount(
     component.components = { ...component.components, ...global.components }
   }
 
+  const componentRef = ref(null)
   // create the wrapper component
   const Parent = defineComponent({
     name: 'VTU_ROOT',
+    setup() {
+      return {
+        [MOUNT_COMPONENT_REF]: componentRef
+      }
+    },
     render() {
       return h(component as ComponentOptions, { ...props, ...refs }, slots)
     }
@@ -562,16 +569,12 @@ export function mount(
   const vm = app.mount(el)
 
   // Ignore "Avoid app logic that relies on enumerating keys on a component instance..." warning
-  const warnSave = console.warn
-  console.warn = () => {}
-
-  const appRef = vm.$refs[MOUNT_COMPONENT_REF] as ComponentPublicInstance
+  const appRef = componentRef.value! as ComponentPublicInstance
   // we add `hasOwnProperty` so Jest can spy on the proxied vm without throwing
   // note that this is not necessary with Jest v27+ or Vitest, but is kept for compatibility with older Jest versions
   appRef.hasOwnProperty = (property) => {
     return Reflect.has(appRef, property)
   }
-  console.warn = warnSave
   const wrapper = createVueWrapper(app, appRef, setProps)
   trackInstance(wrapper)
   return wrapper
diff --git a/tests/shallowMount.spec.ts b/tests/shallowMount.spec.ts
index a852cbca1b..5adc1e4abf 100644
--- a/tests/shallowMount.spec.ts
+++ b/tests/shallowMount.spec.ts
@@ -1,5 +1,5 @@
 import { describe, expect, it } from 'vitest'
-import { defineAsyncComponent, defineComponent } from 'vue'
+import { defineAsyncComponent, defineComponent, onMounted } from 'vue'
 import { mount, shallowMount, VueWrapper } from '../src'
 import ComponentWithChildren from './components/ComponentWithChildren.vue'
 import ScriptSetupWithChildren from './components/ScriptSetupWithChildren.vue'