Merge pull request #64 from cexbrayat/fix/vm-type

fix: vm should be the vm of the tested component

Author: lmiller1990

src/vue-wrapper.ts

@@ -19,7 +19,9 @@ export class VueWrapper implements WrapperAPI {
   ) {
     this.__vm = vm
     this.__setProps = setProps
-    this.componentVM = this.vm.$refs['VTU_COMPONENT'] as ComponentPublicInstance
+    this.componentVM = this.__vm.$refs[
+      'VTU_COMPONENT'
+    ] as ComponentPublicInstance
     this.__emitted = events
   }
 
@@ -42,7 +44,7 @@ export class VueWrapper implements WrapperAPI {
   }
 
   get vm(): ComponentPublicInstance {
-    return this.__vm
+    return this.componentVM
   }
 
   classes(className?: string) {

tests/vm.spec.ts

@@ -0,0 +1,21 @@
+import { defineComponent, ref } from 'vue'
+
+import { mount } from '../src'
+
+describe('vm', () => {
+  it('returns the component vm', () => {
+    const Component = defineComponent({
+      template: '<div>{{ msg }}</div>',
+      setup() {
+        const msg = 'hello'
+        const isEnabled = ref(true)
+        return { msg, isEnabled }
+      }
+    })
+
+    const wrapper = mount(Component)
+
+    expect((wrapper.vm as any).msg).toBe('hello')
+    expect((wrapper.vm as any).isEnabled).toBe(true)
+  })
+})