40 lines
1.2 KiB
TypeScript
40 lines
1.2 KiB
TypeScript
/**
|
|
* @license
|
|
* Copyright Google Inc. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by an MIT-style license that can be
|
|
* found in the LICENSE file at https://angular.io/license
|
|
*/
|
|
|
|
import {NgModuleResolver} from '@angular/compiler';
|
|
import {Compiler, Injectable, Injector, NgModule, Type} from '@angular/core';
|
|
|
|
@Injectable()
|
|
export class MockNgModuleResolver extends NgModuleResolver {
|
|
private _ngModules = new Map<Type<any>, NgModule>();
|
|
|
|
constructor(private _injector: Injector) { super(); }
|
|
|
|
/**
|
|
* Overrides the {@link NgModule} for a module.
|
|
*/
|
|
setNgModule(type: Type<any>, metadata: NgModule): void {
|
|
this._ngModules.set(type, metadata);
|
|
this._clearCacheFor(type);
|
|
}
|
|
|
|
/**
|
|
* Returns the {@link NgModule} for a module:
|
|
* - Set the {@link NgModule} to the overridden view when it exists or fallback to the
|
|
* default
|
|
* `NgModuleResolver`, see `setNgModule`.
|
|
*/
|
|
resolve(type: Type<any>, throwIfNotFound = true): NgModule {
|
|
return this._ngModules.get(type) || super.resolve(type, throwIfNotFound);
|
|
}
|
|
|
|
private get _compiler(): Compiler { return this._injector.get(Compiler); }
|
|
|
|
private _clearCacheFor(component: Type<any>) { this._compiler.clearCacheFor(component); }
|
|
}
|