@remotex-labs/xjet 1.2.2 → 1.2.3

package/dist/index.d.ts

@@ -178,7 +178,7 @@ declare function fnImplementation<ReturnType, Args extends Array<unknown>, Conte
  *
  * @since 1.2.2
  */
-declare function mockImplementation<F extends abstract new (...args: any) => any>(method: F, implementation?: (...args: ConstructorParameters<F>) => Partial<InstanceType<F>>): MockState<(...args: ConstructorParameters<F>) => Partial<InstanceType<F>>>;
+declare function mockImplementation<F extends abstract new (...args: any) => any>(method: F, implementation?: (...args: ConstructorParameters<F>) => PartialResolvedType<InstanceType<F>>): MockState<(...args: ConstructorParameters<F>) => PartialResolvedType<InstanceType<F>>>;
 /**
  * Creates a mock implementation of the provided function.
  *
@@ -213,7 +213,7 @@ declare function mockImplementation<F extends abstract new (...args: any) => any
  *
  * @since 1.2.2
  */
-declare function mockImplementation<F extends FunctionType>(method: F, implementation?: (...args: Parameters<F>) => Partial<ReturnType<F>>): MockState<(this: ThisParameterType<F>, ...args: Parameters<F>) => Partial<ReturnType<F>>>;
+declare function mockImplementation<F extends FunctionType>(method: F, implementation?: (...args: Parameters<F>) => PartialResolvedType<ReturnType<F>>): MockState<(this: ThisParameterType<F>, ...args: Parameters<F>) => PartialResolvedType<ReturnType<F>>>;
 /**
  * Creates a mock for an element with an optional custom implementation.
  *
@@ -292,6 +292,27 @@ interface MockableFunctionInterface<F extends FunctionType> extends MockState<F>
      */
     (this: ThisParameterType<F>, ...args: Parameters<F>): ReturnType<F>;
 }
+/**
+ * Makes properties of a type or its resolved promise value optional.
+ *
+ * @template T - The type to transform
+ *
+ * @remarks
+ * If T is a Promise-like type, this utility unwraps it and makes the resolved value's
+ * properties optional. Otherwise, it directly makes T's properties optional.
+ *
+ * @example
+ * ```ts
+ * // Makes properties of User optional
+ * type MaybeUser = PartialResolvedType<User>;
+ *
+ * // Makes properties of resolved User optional
+ * type MaybeAsyncUser = PartialResolvedType<Promise<User>>;
+ * ```
+ *
+ * @since 1.2.2
+ */
+type PartialResolvedType<T> = T extends PromiseLike<infer U> ? Promise<Partial<U>> : Partial<T>;
 
 /**
  * Import will remove at compile time
@@ -1560,7 +1581,7 @@ declare function spyOnProxyGet<T extends Record<string | symbol, unknown>, K ext
  *
  * @since 1.0.0
  */
-declare function spyOnImplementation<T extends object, K extends keyof T>(target: T, key: K): T[K] extends FunctionType ? MockState<(this: ThisParameterType<T[K]>, ...args: Parameters<T[K]>) => ReturnType<T[K]>> : MockState<() => T[K]>;
+declare function spyOnImplementation<T extends object, K extends keyof T>(target: T, key: K): T[K] extends FunctionType ? MockState<(this: ThisParameterType<T[K]>, ...args: Parameters<T[K]>) => PartialResolvedType<ReturnType<T[K]>>> : MockState<() => T[K]>;
 
 /**
  * Interface representing the internal state of a mock proxy.