/** * @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 {AnimationEntryMetadata, Compiler, ComponentFactory, Inject, Injectable, Injector, NgZone, Type, ViewMetadata} from '@angular/core'; import {ComponentFixture, ComponentFixtureNoNgZone, TestBed, TestComponentBuilder} from '@angular/core/testing'; import {DirectiveResolver} from '../index'; import {MapWrapper} from '../src/facade/collection'; import {isPresent} from '../src/facade/lang'; /** * A TestComponentBuilder that allows overriding based on the compiler. * * @deprecated Use `TestBed.configureTestModule` / `TestBed.override...` / `TestBed.createComponent` * instead. */ @Injectable() export class OverridingTestComponentBuilder extends TestComponentBuilder { /** @internal */ _bindingsOverrides = new Map<Type<any>, any[]>(); /** @internal */ _directiveOverrides = new Map<Type<any>, Map<Type<any>, Type<any>>>(); /** @internal */ _templateOverrides = new Map<Type<any>, string>(); /** @internal */ _animationOverrides = new Map<Type<any>, AnimationEntryMetadata[]>(); /** @internal */ _viewBindingsOverrides = new Map<Type<any>, any[]>(); /** @internal */ _viewOverrides = new Map<Type<any>, ViewMetadata>(); constructor(@Inject(TestBed) injector: Injector) { super(injector); } /** @internal */ _clone(): OverridingTestComponentBuilder { let clone = new OverridingTestComponentBuilder(this._injector); clone._viewOverrides = MapWrapper.clone(this._viewOverrides); clone._directiveOverrides = MapWrapper.clone(this._directiveOverrides); clone._templateOverrides = MapWrapper.clone(this._templateOverrides); clone._bindingsOverrides = MapWrapper.clone(this._bindingsOverrides); clone._viewBindingsOverrides = MapWrapper.clone(this._viewBindingsOverrides); return clone; } overrideTemplate(componentType: Type<any>, template: string): OverridingTestComponentBuilder { let clone = this._clone(); clone._templateOverrides.set(componentType, template); return clone; } overrideAnimations(componentType: Type<any>, animations: AnimationEntryMetadata[]): TestComponentBuilder { var clone = this._clone(); clone._animationOverrides.set(componentType, animations); return clone; } overrideView(componentType: Type<any>, view: ViewMetadata): OverridingTestComponentBuilder { let clone = this._clone(); clone._viewOverrides.set(componentType, view); return clone; } overrideDirective(componentType: Type<any>, from: Type<any>, to: Type<any>): OverridingTestComponentBuilder { let clone = this._clone(); let overridesForComponent = clone._directiveOverrides.get(componentType); if (!isPresent(overridesForComponent)) { clone._directiveOverrides.set(componentType, new Map<Type<any>, Type<any>>()); overridesForComponent = clone._directiveOverrides.get(componentType); } overridesForComponent.set(from, to); return clone; } overrideProviders(type: Type<any>, providers: any[]): OverridingTestComponentBuilder { let clone = this._clone(); clone._bindingsOverrides.set(type, providers); return clone; } overrideViewProviders(type: Type<any>, providers: any[]): OverridingTestComponentBuilder { let clone = this._clone(); clone._viewBindingsOverrides.set(type, providers); return clone; } createAsync<T>(rootComponentType: Type<T>): Promise<ComponentFixture<T>> { this._applyMetadataOverrides(); return super.createAsync(rootComponentType); } createSync<T>(rootComponentType: Type<T>): ComponentFixture<T> { this._applyMetadataOverrides(); return super.createSync(rootComponentType); } private _applyMetadataOverrides() { let mockDirectiveResolver = this._injector.get(DirectiveResolver); this._viewOverrides.forEach((view, type) => { mockDirectiveResolver.setView(type, view); }); this._templateOverrides.forEach( (template, type) => mockDirectiveResolver.setInlineTemplate(type, template)); this._animationOverrides.forEach( (animationsEntry, type) => mockDirectiveResolver.setAnimations(type, animationsEntry)); this._directiveOverrides.forEach((overrides, component) => { overrides.forEach( (to, from) => { mockDirectiveResolver.overrideViewDirective(component, from, to); }); }); this._bindingsOverrides.forEach( (bindings, type) => mockDirectiveResolver.setProvidersOverride(type, bindings)); this._viewBindingsOverrides.forEach( (bindings, type) => mockDirectiveResolver.setViewProvidersOverride(type, bindings)); } }