fix: take data option into account

vuejs · Apr 11, 2023 · f2be9d5 · f2be9d5
1 parent 686a5d6
commit f2be9d5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/mount.ts b/src/mount.ts
@@ -16,8 +16,10 @@ type ShimSlotReturnType<T> = T extends (...args: infer P) => any
 
 type WithArray<T> = T | T[]
 
+type ComponentData<T> = T extends { data?(...args: any): infer D } ? D : {}
+
 export type ComponentMountingOptions<T> = Omit<
-  MountingOptions<ComponentProps<T>>,
+  MountingOptions<ComponentProps<T>, ComponentData<T>>,
   'slots'
 > & {
   slots?: {
@@ -45,7 +47,7 @@ export function mount<
 >(
   originalComponent: T,
   options?: ComponentMountingOptions<C>
-): VueWrapper<ComponentExposed<C> & ComponentProps<C>>
+): VueWrapper<ComponentExposed<C> & ComponentProps<C> & ComponentData<C>>
 
 // implementation
 export function mount(

diff --git a/src/vueWrapper.ts b/src/vueWrapper.ts
@@ -70,7 +70,7 @@ function createVMProxy<T extends ComponentPublicInstance>(
 
 export class VueWrapper<
   VM = unknown,
-  T extends VM & ComponentPublicInstance = VM & ComponentPublicInstance
+  T extends ComponentPublicInstance = VM & ComponentPublicInstance
 > extends BaseWrapper<Node> {
   private readonly componentVM: T
   private readonly rootVM: ComponentPublicInstance | undefined | null