fix(mock-intance): supports tokens

help-me-mom · Dec 25, 2020 · 92abb82 · 92abb82
1 parent 4eb0245
commit 92abb82
Showing 1 changed file with 22 additions and 4 deletions.
diff --git a/lib/mock-instance/mock-instance.ts b/lib/mock-instance/mock-instance.ts
@@ -1,14 +1,32 @@
-import { Injector } from '@angular/core';
+import { InjectionToken, Injector } from '@angular/core';
 
 import { AbstractType, Type } from '../common/core.types';
 import ngMocksUniverse from '../common/ng-mocks-universe';
 
+/**
+ * @see https://github.com/ike18t/ng-mocks#mockinstance
+ */
+export function MockInstance<T>(
+  declaration: InjectionToken<T>,
+  init?: (instance: T | undefined, injector: Injector | undefined) => Partial<T>,
+): void;
+
+/**
+ * @see https://github.com/ike18t/ng-mocks#mockinstance
+ */
+export function MockInstance<T>(
+  declaration: InjectionToken<T>,
+  config?: {
+    init?: (instance: T | undefined, injector: Injector | undefined) => Partial<T>;
+  },
+): void;
+
 /**
  * @see https://github.com/ike18t/ng-mocks#mockinstance
  */
 export function MockInstance<T>(
   declaration: Type<T> | AbstractType<T>,
-  init?: (instance: T, injector: Injector | undefined) => void,
+  init?: (instance: T, injector: Injector | undefined) => void | Partial<T>,
 ): void;
 
 /**
@@ -17,11 +35,11 @@ export function MockInstance<T>(
 export function MockInstance<T>(
   declaration: Type<T> | AbstractType<T>,
   config?: {
-    init?(instance: T, injector: Injector | undefined): void;
+    init?: (instance: T, injector: Injector | undefined) => void | Partial<T>;
   },
 ): void;
 
-export function MockInstance<T>(declaration: Type<T> | AbstractType<T>, data?: any) {
+export function MockInstance<T>(declaration: Type<T> | AbstractType<T> | InjectionToken<T>, data?: any) {
   const config = typeof data === 'function' ? { init: data } : data;
   const universeConfig = ngMocksUniverse.config.has(declaration) ? ngMocksUniverse.config.get(declaration) : {};
   if (config) {