improve mount of functionalComponent and add couple more tests

Author: pikax

src/mount.ts

@@ -64,36 +64,20 @@ export type ObjectEmitsOptions = Record<
 export type EmitsOptions = ObjectEmitsOptions | string[]
 
 // Functional component
-export function mount<TestedComponent extends FunctionalComponent>(
+export function mount<
+  TestedComponent extends FunctionalComponent<Props>,
+  Props
+>(
   originalComponent: TestedComponent,
-  options?: MountingOptions<any>
-): VueWrapper<ComponentPublicInstance>
+  options?: MountingOptions<Props>
+): VueWrapper<ComponentPublicInstance<Props>>
+
 // Component declared with defineComponent
 export function mount<TestedComponent extends ComponentPublicInstance>(
   originalComponent: { new (): TestedComponent } & Component,
   options?: MountingOptions<TestedComponent['$props'], TestedComponent['$data']>
 ): VueWrapper<TestedComponent>
 
-// Functional Component declared with (props, ctx)=> render
-export function mount<Props, RawBindings = object>(
-  setup: (
-    props: Readonly<Props>,
-    ctx: SetupContext
-  ) => RawBindings | RenderFunction,
-
-  options: MountingOptions<Props, {}>
-): VueWrapper<
-  ComponentPublicInstance<
-    Props,
-    RawBindings,
-    {},
-    {},
-    {},
-    // public props
-    VNodeProps & Props
-  >
->
-
 // Component declared with no props
 export function mount<
   Props = {},

test-dts/mount.d-test.ts

@@ -91,9 +91,11 @@ expectType<string>(
 
 // can receive extra props
 // as they are declared as `string[]`
-mount(AppWithArrayProps, {
-  props: { a: 'Hello', b: 2 }
-})
+expectType<number>(
+  mount(AppWithArrayProps, {
+    props: { a: 'Hello', b: 2 }
+  }).vm.b
+)
 
 // cannot receive extra props
 // if they pass use object inside
@@ -125,3 +127,20 @@ expectError(
 mount(AppWithoutProps, {
   props: { b: 'Hello' } as never
 })
+
+// Functional tests
+
+// wrong props
+expectError((props: { a: 1 }) => {}, {
+  props: {
+    a: '222'
+  }
+})
+
+expectType<number>(
+  mount((props: { a: 1 }, ctx) => {}, {
+    props: {
+      a: 22
+    }
+  }).vm.a
+)