moar

jestjs · Feb 10, 2019 · c12feb3 · c12feb3
1 parent 47f5522
commit c12feb3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/packages/jest-mock/src/index.ts b/packages/jest-mock/src/index.ts
@@ -911,7 +911,7 @@ class ModuleMockerClass {
     return metadata;
   }
 
-  isMockFunction(fn: any): boolean {
+  isMockFunction<T, Y extends unknown[]>(fn: any): fn is Mock<T, Y> {
     return !!fn && fn._isMockFunction === true;
   }
 
@@ -943,7 +943,7 @@ class ModuleMockerClass {
     ? SpyInstance<ReturnType<T[M]>, ArgsType<T[M]>>
     : never;
 
-  spyOn<T extends {}, M extends keyof T>(
+  spyOn<T extends {}, M extends NonFunctionPropertyNames<T>>(
     object: T,
     methodName: M,
     accessType?: 'get' | 'set',
@@ -985,7 +985,7 @@ class ModuleMockerClass {
     return object[methodName];
   }
 
-  private _spyOnProperty<T extends {}, M extends keyof T>(
+  private _spyOnProperty<T extends {}, M extends NonFunctionPropertyNames<T>>(
     obj: T,
     propertyName: M,
     accessType: 'get' | 'set' = 'get',