83 lines
3.1 KiB
Plaintext
83 lines
3.1 KiB
Plaintext
declare const S: unique symbol;
|
|
|
|
interface GetState {
|
|
<A extends any[], R>(spy: SpyInternalImpl<A, R>): SpyInternalImplState<A, R>;
|
|
<A extends any[], R>(spy: SpyInternal<A, R>): SpyInternalState<A, R>;
|
|
}
|
|
declare let spies: Set<SpyImpl<any[], any>>;
|
|
declare let getInternalState: GetState;
|
|
declare type ReturnError = ['error', any];
|
|
declare type ReturnOk<R> = ['ok', R];
|
|
declare type ResultFn<R> = ReturnError | ReturnOk<R>;
|
|
interface SpyInternal<A extends any[] = any[], R = any> {
|
|
(this: any, ...args: A): R;
|
|
[S]: SpyInternalState<A, R>;
|
|
}
|
|
interface SpyInternalImpl<A extends any[] = any[], R = any> extends SpyInternal<A, R> {
|
|
[S]: SpyInternalImplState<A, R>;
|
|
}
|
|
interface SpyInternalState<A extends any[] = any[], R = any> {
|
|
called: boolean;
|
|
callCount: number;
|
|
calls: A[];
|
|
results: ResultFn<R>[];
|
|
reset(): void;
|
|
impl: ((...args: A) => R) | undefined;
|
|
next: ResultFn<R>[];
|
|
}
|
|
interface SpyInternalImplState<A extends any[] = any[], R = any> extends SpyInternalState<A, R> {
|
|
getOriginal(): (...args: A) => R;
|
|
willCall(cb: (...args: A) => R): this;
|
|
restore(): void;
|
|
}
|
|
interface Spy<A extends any[] = any[], R = any> extends SpyInternalState<A, R> {
|
|
returns: R[];
|
|
length: number;
|
|
nextError(error: any): this;
|
|
nextResult(result: R): this;
|
|
}
|
|
interface SpyImpl<A extends any[] = any[], R = any> extends Spy<A, R> {
|
|
getOriginal(): (...args: A) => R;
|
|
willCall(cb: (...args: A) => R): this;
|
|
restore(): void;
|
|
}
|
|
declare function createInternalSpy<A extends any[], R>(cb?: ((...args: A) => R) | {
|
|
new (...args: A): R;
|
|
}): SpyInternal<A, R>;
|
|
|
|
interface SpyFn<A extends any[] = any[], R = any> extends Spy<A, R> {
|
|
new (...args: A): R extends void ? any : R;
|
|
(...args: A): R;
|
|
}
|
|
declare function spy<A extends any[], R>(cb?: ((...args: A) => R) | {
|
|
new (...args: A): R;
|
|
}): SpyFn<A, R>;
|
|
|
|
declare type Procedure = (...args: any[]) => any;
|
|
declare type Methods<T> = {
|
|
[K in keyof T]: T[K] extends Procedure ? K : never;
|
|
}[keyof T];
|
|
declare type Getters<T> = {
|
|
[K in keyof T]: T[K] extends Procedure ? never : K;
|
|
}[keyof T];
|
|
declare type Constructors<T> = {
|
|
[K in keyof T]: T[K] extends new (...args: any[]) => any ? K : never;
|
|
}[keyof T];
|
|
declare function internalSpyOn<T, K extends string & keyof T>(obj: T, methodName: K | {
|
|
getter: K;
|
|
} | {
|
|
setter: K;
|
|
}, mock?: Procedure): SpyInternalImpl<any[], any>;
|
|
declare function spyOn<T, S extends Getters<Required<T>>>(obj: T, methodName: {
|
|
setter: S;
|
|
}, mock?: (arg: T[S]) => void): SpyImpl<[T[S]], void>;
|
|
declare function spyOn<T, G extends Getters<Required<T>>>(obj: T, methodName: {
|
|
getter: G;
|
|
}, mock?: () => T[G]): SpyImpl<[], T[G]>;
|
|
declare function spyOn<T, M extends Constructors<Required<T>>>(object: T, method: M): Required<T>[M] extends new (...args: infer A) => infer R ? SpyImpl<A, R> : never;
|
|
declare function spyOn<T, M extends Methods<Required<T>>>(obj: T, methodName: M, mock?: T[M]): Required<T>[M] extends (...args: infer A) => infer R ? SpyImpl<A, R> : never;
|
|
|
|
declare function restoreAll(): void;
|
|
|
|
export { Spy, SpyFn, SpyImpl, SpyInternal, SpyInternalImpl, createInternalSpy, getInternalState, internalSpyOn, restoreAll, spies, spy, spyOn };
|