/** * @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, ViewMetadata} from '@angular/core'; import {ComponentFixture, ComponentFixtureNoNgZone, TestBed, TestComponentBuilder} from '@angular/core/testing'; import {DirectiveResolver} from '../index'; import {MapWrapper} from '../src/facade/collection'; import {ConcreteType, IS_DART, Type, 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(); /** @internal */ _directiveOverrides = new Map>(); /** @internal */ _templateOverrides = new Map(); /** @internal */ _animationOverrides = new Map(); /** @internal */ _viewBindingsOverrides = new Map(); /** @internal */ _viewOverrides = new Map(); 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, template: string): OverridingTestComponentBuilder { let clone = this._clone(); clone._templateOverrides.set(componentType, template); return clone; } overrideAnimations(componentType: Type, animations: AnimationEntryMetadata[]): TestComponentBuilder { var clone = this._clone(); clone._animationOverrides.set(componentType, animations); return clone; } overrideView(componentType: Type, view: ViewMetadata): OverridingTestComponentBuilder { let clone = this._clone(); clone._viewOverrides.set(componentType, view); return clone; } overrideDirective(componentType: Type, from: Type, to: Type): OverridingTestComponentBuilder { let clone = this._clone(); let overridesForComponent = clone._directiveOverrides.get(componentType); if (!isPresent(overridesForComponent)) { clone._directiveOverrides.set(componentType, new Map()); overridesForComponent = clone._directiveOverrides.get(componentType); } overridesForComponent.set(from, to); return clone; } overrideProviders(type: Type, providers: any[]): OverridingTestComponentBuilder { let clone = this._clone(); clone._bindingsOverrides.set(type, providers); return clone; } overrideViewProviders(type: Type, providers: any[]): OverridingTestComponentBuilder { let clone = this._clone(); clone._viewBindingsOverrides.set(type, providers); return clone; } createAsync(rootComponentType: ConcreteType): Promise> { this._applyMetadataOverrides(); return super.createAsync(rootComponentType); } createSync(rootComponentType: ConcreteType): ComponentFixture { 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)); } }