fix: improve vm types

.github/workflows/ci.yml

@@ -30,3 +30,4 @@ jobs:
     - run: yarn build
       env:
         CI: true
+    - run: yarn tsd

package.json

@@ -9,7 +9,8 @@
   "module": "dist/vue-test-utils.esm-bundler.js",
   "files": [
     "dist",
-    "README.md"
+    "README.md",
+    "dist/index.d.ts"
   ],
   "dependencies": {
     "lodash": "^4.17.15"
@@ -21,6 +22,7 @@
     "@rollup/plugin-node-resolve": "^7.1.3",
     "@types/estree": "^0.0.42",
     "@types/jest": "^24.9.1",
+    "@types/node": "12.12.35",
     "@types/lodash": "^4.14.149",
     "@vue/compiler-sfc": "^3.0.0-alpha.13",
     "babel-jest": "^25.2.3",
@@ -33,6 +35,7 @@
     "rollup": "^1.31.1",
     "rollup-plugin-typescript2": "^0.26.0",
     "ts-jest": "^25.0.0",
+    "tsd": "0.11.0",
     "typescript": "^3.7.5",
     "vue": "^3.0.0-alpha.13",
     "vue-jest": "vuejs/vue-jest#next",
@@ -48,6 +51,7 @@
   },
   "scripts": {
     "test": "yarn jest --runInBand tests",
+    "tsd": "tsd",
     "build": "yarn rollup -c rollup.config.js",
     "lint": "prettier -c --parser typescript \"(src|tests)/**/*.ts?(x)\"",
     "lint:fix": "yarn lint --write"
@@ -61,5 +65,12 @@
     "*.ts": [
       "prettier --parser=typescript --write"
     ]
+  },
+  "tsd": {
+    "directory": "test-dts",
+    "compilerOptions": {
+      "strict": false,
+      "lib": ["esnext", "dom"]
+    }
   }
 }

src/mount.ts

@@ -41,14 +41,18 @@ interface MountingOptions {
   stubs?: Record<string, any>
 }
 
-export function mount<T extends any>(
-  originalComponent: any,
+export function mount<TestedComponent extends ComponentPublicInstance>(
+  originalComponent: new () => TestedComponent,
+  options?: MountingOptions
+): VueWrapper<TestedComponent>
+export function mount(
+  originalComponent: Component,
   options?: MountingOptions
 ): VueWrapper<any>
-export function mount<T extends ComponentPublicInstance>(
-  originalComponent: new () => T,
+export function mount(
+  originalComponent: any,
   options?: MountingOptions
-): VueWrapper<T> {
+): VueWrapper<any> {
   const component = { ...originalComponent }
 
   // Reset the document.body
@@ -158,5 +162,5 @@ export function mount<T extends ComponentPublicInstance>(
   // mount the app!
   const app = vm.mount(el)
 
-  return createWrapper<T>(app, events, setProps)
+  return createWrapper(app, events, setProps)
 }

test-dts/index.d-test.ts

@@ -0,0 +1,23 @@
+import { expectType } from 'tsd'
+import { defineComponent } from 'vue'
+import { mount } from '../src'
+
+const App = defineComponent({
+  props: {
+    a: String
+  },
+  template: ''
+})
+
+let wrapper = mount(App)
+expectType<string>(wrapper.vm.a)
+
+const AppWithoutDefine = {
+  props: {
+    a: String
+  },
+  template: ''
+}
+
+wrapper = mount(AppWithoutDefine)
+expectType<string>(wrapper.vm.a)