491 lines
21 KiB
TypeScript
491 lines
21 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 MagicString from 'magic-string';
|
||
|
|
import * as ts from 'typescript';
|
||
|
|
import {NoopImportRewriter} from '../../../src/ngtsc/imports';
|
||
|
|
import {AbsoluteFsPath} from '../../../src/ngtsc/path';
|
||
|
|
import {DecorationAnalyzer} from '../../src/analysis/decoration_analyzer';
|
||
|
|
import {NgccReferencesRegistry} from '../../src/analysis/ngcc_references_registry';
|
||
|
|
import {SwitchMarkerAnalyzer} from '../../src/analysis/switch_marker_analyzer';
|
||
|
|
import {UmdReflectionHost} from '../../src/host/umd_host';
|
||
|
|
import {ImportManager} from '../../../src/ngtsc/translator';
|
||
|
|
import {UmdRenderer} from '../../src/rendering/umd_renderer';
|
||
|
|
import {MockFileSystem} from '../helpers/mock_file_system';
|
||
|
|
import {MockLogger} from '../helpers/mock_logger';
|
||
|
|
import {getDeclaration, makeTestEntryPointBundle, createFileSystemFromProgramFiles} from '../helpers/utils';
|
||
|
|
|
||
|
|
const _ = AbsoluteFsPath.fromUnchecked;
|
||
|
|
|
||
|
|
function setup(file: {name: string, contents: string}) {
|
||
|
|
const fs = new MockFileSystem(createFileSystemFromProgramFiles([file]));
|
||
|
|
const logger = new MockLogger();
|
||
|
|
const bundle = makeTestEntryPointBundle('esm5', 'esm5', false, [file]);
|
||
|
|
const src = bundle.src;
|
||
|
|
const typeChecker = src.program.getTypeChecker();
|
||
|
|
const host = new UmdReflectionHost(logger, false, src.program, src.host);
|
||
|
|
const referencesRegistry = new NgccReferencesRegistry(host);
|
||
|
|
const decorationAnalyses = new DecorationAnalyzer(
|
||
|
|
fs, src.program, src.options, src.host, typeChecker, host,
|
||
|
|
referencesRegistry, [AbsoluteFsPath.fromUnchecked('/')], false)
|
||
|
|
.analyzeProgram();
|
||
|
|
const switchMarkerAnalyses = new SwitchMarkerAnalyzer(host).analyzeProgram(src.program);
|
||
|
|
const renderer = new UmdRenderer(fs, logger, host, false, bundle);
|
||
|
|
const importManager = new ImportManager(new NoopImportRewriter(), 'i');
|
||
|
|
return {
|
||
|
|
decorationAnalyses,
|
||
|
|
host,
|
||
|
|
importManager,
|
||
|
|
program: src.program, renderer,
|
||
|
|
sourceFile: src.file, switchMarkerAnalyses
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
const PROGRAM = {
|
||
|
|
name: _('/some/file.js'),
|
||
|
|
contents: `
|
||
|
|
/* A copyright notice */
|
||
|
|
(function (global, factory) {
|
||
|
|
typeof exports === 'object' && typeof module !== 'undefined' ? factory(exports,require('some-side-effect'),require('/local-dep'),require('@angular/core')) :
|
||
|
|
typeof define === 'function' && define.amd ? define('file', ['exports','some-side-effect','/local-dep','@angular/core'], factory) :
|
||
|
|
(factory(global.file,global.someSideEffect,global.localDep,global.ng.core));
|
||
|
|
}(this, (function (exports,someSideEffect,localDep,core) {'use strict';
|
||
|
|
var A = (function() {
|
||
|
|
function A() {}
|
||
|
|
A.decorators = [
|
||
|
|
{ type: core.Directive, args: [{ selector: '[a]' }] },
|
||
|
|
{ type: OtherA }
|
||
|
|
];
|
||
|
|
A.prototype.ngDoCheck = function() {
|
||
|
|
//
|
||
|
|
};
|
||
|
|
return A;
|
||
|
|
}());
|
||
|
|
|
||
|
|
var B = (function() {
|
||
|
|
function B() {}
|
||
|
|
B.decorators = [
|
||
|
|
{ type: OtherB },
|
||
|
|
{ type: core.Directive, args: [{ selector: '[b]' }] }
|
||
|
|
];
|
||
|
|
return B;
|
||
|
|
}());
|
||
|
|
|
||
|
|
var C = (function() {
|
||
|
|
function C() {}
|
||
|
|
C.decorators = [
|
||
|
|
{ type: core.Directive, args: [{ selector: '[c]' }] },
|
||
|
|
];
|
||
|
|
return C;
|
||
|
|
}());
|
||
|
|
|
||
|
|
function NoIife() {}
|
||
|
|
|
||
|
|
var BadIife = (function() {
|
||
|
|
function BadIife() {}
|
||
|
|
BadIife.decorators = [
|
||
|
|
{ type: core.Directive, args: [{ selector: '[c]' }] },
|
||
|
|
];
|
||
|
|
}());
|
||
|
|
|
||
|
|
var compileNgModuleFactory = compileNgModuleFactory__PRE_R3__;
|
||
|
|
var badlyFormattedVariable = __PRE_R3__badlyFormattedVariable;
|
||
|
|
function compileNgModuleFactory__PRE_R3__(injector, options, moduleType) {
|
||
|
|
var compilerFactory = injector.get(CompilerFactory);
|
||
|
|
var compiler = compilerFactory.createCompiler([options]);
|
||
|
|
return compiler.compileModuleAsync(moduleType);
|
||
|
|
}
|
||
|
|
|
||
|
|
function compileNgModuleFactory__POST_R3__(injector, options, moduleType) {
|
||
|
|
ngDevMode && assertNgModuleType(moduleType);
|
||
|
|
return Promise.resolve(new R3NgModuleFactory(moduleType));
|
||
|
|
}
|
||
|
|
// Some other content
|
||
|
|
exports.A = A;
|
||
|
|
exports.B = B;
|
||
|
|
exports.C = C;
|
||
|
|
exports.NoIife = NoIife;
|
||
|
|
exports.BadIife = BadIife;
|
||
|
|
})));`,
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
const PROGRAM_DECORATE_HELPER = {
|
||
|
|
name: '/some/file.js',
|
||
|
|
contents: `
|
||
|
|
/* A copyright notice */
|
||
|
|
(function (global, factory) {
|
||
|
|
typeof exports === 'object' && typeof module !== 'undefined' ? factory(exports,require('tslib'),require('@angular/core')) :
|
||
|
|
typeof define === 'function' && define.amd ? define('file', ['exports','/tslib','@angular/core'], factory) :
|
||
|
|
(factory(global.file,global.tslib,global.ng.core));
|
||
|
|
}(this, (function (exports,tslib,core) {'use strict';
|
||
|
|
var OtherA = function () { return function (node) { }; };
|
||
|
|
var OtherB = function () { return function (node) { }; };
|
||
|
|
var A = /** @class */ (function () {
|
||
|
|
function A() {
|
||
|
|
}
|
||
|
|
A = tslib.__decorate([
|
||
|
|
core.Directive({ selector: '[a]' }),
|
||
|
|
OtherA()
|
||
|
|
], A);
|
||
|
|
return A;
|
||
|
|
}());
|
||
|
|
export { A };
|
||
|
|
var B = /** @class */ (function () {
|
||
|
|
function B() {
|
||
|
|
}
|
||
|
|
B = tslib.__decorate([
|
||
|
|
OtherB(),
|
||
|
|
core.Directive({ selector: '[b]' })
|
||
|
|
], B);
|
||
|
|
return B;
|
||
|
|
}());
|
||
|
|
export { B };
|
||
|
|
var C = /** @class */ (function () {
|
||
|
|
function C() {
|
||
|
|
}
|
||
|
|
C = tslib.__decorate([
|
||
|
|
core.Directive({ selector: '[c]' })
|
||
|
|
], C);
|
||
|
|
return C;
|
||
|
|
}());
|
||
|
|
export { C };
|
||
|
|
var D = /** @class */ (function () {
|
||
|
|
function D() {
|
||
|
|
}
|
||
|
|
D_1 = D;
|
||
|
|
var D_1;
|
||
|
|
D = D_1 = tslib.__decorate([
|
||
|
|
core.Directive({ selector: '[d]', providers: [D_1] })
|
||
|
|
], D);
|
||
|
|
return D;
|
||
|
|
}());
|
||
|
|
exports.D = D;
|
||
|
|
// Some other content
|
||
|
|
})));`
|
||
|
|
};
|
||
|
|
|
||
|
|
describe('UmdRenderer', () => {
|
||
|
|
|
||
|
|
describe('addImports', () => {
|
||
|
|
it('should append the given imports into the CommonJS factory call', () => {
|
||
|
|
const {renderer, program} = setup(PROGRAM);
|
||
|
|
const file = program.getSourceFile('some/file.js') !;
|
||
|
|
const output = new MagicString(PROGRAM.contents);
|
||
|
|
renderer.addImports(
|
||
|
|
output,
|
||
|
|
[
|
||
|
|
{specifier: '@angular/core', qualifier: 'i0'},
|
||
|
|
{specifier: '@angular/common', qualifier: 'i1'}
|
||
|
|
],
|
||
|
|
file);
|
||
|
|
expect(output.toString())
|
||
|
|
.toContain(
|
||
|
|
`typeof exports === 'object' && typeof module !== 'undefined' ? factory(exports,require('some-side-effect'),require('/local-dep'),require('@angular/core'),require('@angular/core'),require('@angular/common')) :`);
|
||
|
|
});
|
||
|
|
|
||
|
|
it('should append the given imports into the AMD initialization', () => {
|
||
|
|
const {renderer, program} = setup(PROGRAM);
|
||
|
|
const file = program.getSourceFile('some/file.js') !;
|
||
|
|
const output = new MagicString(PROGRAM.contents);
|
||
|
|
renderer.addImports(
|
||
|
|
output,
|
||
|
|
[
|
||
|
|
{specifier: '@angular/core', qualifier: 'i0'},
|
||
|
|
{specifier: '@angular/common', qualifier: 'i1'}
|
||
|
|
],
|
||
|
|
file);
|
||
|
|
expect(output.toString())
|
||
|
|
.toContain(
|
||
|
|
`typeof define === 'function' && define.amd ? define('file', ['exports','some-side-effect','/local-dep','@angular/core','@angular/core','@angular/common'], factory) :`);
|
||
|
|
});
|
||
|
|
|
||
|
|
it('should append the given imports into the global initialization', () => {
|
||
|
|
const {renderer, program} = setup(PROGRAM);
|
||
|
|
const file = program.getSourceFile('some/file.js') !;
|
||
|
|
const output = new MagicString(PROGRAM.contents);
|
||
|
|
renderer.addImports(
|
||
|
|
output,
|
||
|
|
[
|
||
|
|
{specifier: '@angular/core', qualifier: 'i0'},
|
||
|
|
{specifier: '@angular/common', qualifier: 'i1'}
|
||
|
|
],
|
||
|
|
file);
|
||
|
|
expect(output.toString())
|
||
|
|
.toContain(
|
||
|
|
`(factory(global.file,global.someSideEffect,global.localDep,global.ng.core,global.ng.core,global.ng.common));`);
|
||
|
|
});
|
||
|
|
|
||
|
|
it('should append the given imports as parameters into the factory function definition', () => {
|
||
|
|
const {renderer, program} = setup(PROGRAM);
|
||
|
|
const file = program.getSourceFile('some/file.js') !;
|
||
|
|
const output = new MagicString(PROGRAM.contents);
|
||
|
|
renderer.addImports(
|
||
|
|
output,
|
||
|
|
[
|
||
|
|
{specifier: '@angular/core', qualifier: 'i0'},
|
||
|
|
{specifier: '@angular/common', qualifier: 'i1'}
|
||
|
|
],
|
||
|
|
file);
|
||
|
|
expect(output.toString())
|
||
|
|
.toContain(`(function (exports,someSideEffect,localDep,core,i0,i1) {'use strict';`);
|
||
|
|
});
|
||
|
|
});
|
||
|
|
|
||
|
|
describe('addExports', () => {
|
||
|
|
it('should insert the given exports at the end of the source file', () => {
|
||
|
|
const {importManager, renderer, sourceFile} = setup(PROGRAM);
|
||
|
|
const output = new MagicString(PROGRAM.contents);
|
||
|
|
const generateNamedImportSpy = spyOn(importManager, 'generateNamedImport').and.callThrough();
|
||
|
|
renderer.addExports(
|
||
|
|
output, PROGRAM.name.replace(/\.js$/, ''),
|
||
|
|
[
|
||
|
|
{from: _('/some/a.js'), identifier: 'ComponentA1'},
|
||
|
|
{from: _('/some/a.js'), identifier: 'ComponentA2'},
|
||
|
|
{from: _('/some/foo/b.js'), identifier: 'ComponentB'},
|
||
|
|
{from: PROGRAM.name, identifier: 'TopLevelComponent'},
|
||
|
|
],
|
||
|
|
importManager, sourceFile);
|
||
|
|
|
||
|
|
expect(output.toString()).toContain(`
|
||
|
|
exports.A = A;
|
||
|
|
exports.B = B;
|
||
|
|
exports.C = C;
|
||
|
|
exports.NoIife = NoIife;
|
||
|
|
exports.BadIife = BadIife;
|
||
|
|
exports.ComponentA1 = i0.ComponentA1;
|
||
|
|
exports.ComponentA2 = i0.ComponentA2;
|
||
|
|
exports.ComponentB = i1.ComponentB;
|
||
|
|
exports.TopLevelComponent = TopLevelComponent;
|
||
|
|
})));`);
|
||
|
|
|
||
|
|
expect(generateNamedImportSpy).toHaveBeenCalledWith('./a', 'ComponentA1');
|
||
|
|
expect(generateNamedImportSpy).toHaveBeenCalledWith('./a', 'ComponentA2');
|
||
|
|
expect(generateNamedImportSpy).toHaveBeenCalledWith('./foo/b', 'ComponentB');
|
||
|
|
});
|
||
|
|
|
||
|
|
it('should not insert alias exports in js output', () => {
|
||
|
|
const {importManager, renderer, sourceFile} = setup(PROGRAM);
|
||
|
|
const output = new MagicString(PROGRAM.contents);
|
||
|
|
renderer.addExports(
|
||
|
|
output, PROGRAM.name.replace(/\.js$/, ''),
|
||
|
|
[
|
||
|
|
{from: _('/some/a.js'), alias: 'eComponentA1', identifier: 'ComponentA1'},
|
||
|
|
{from: _('/some/a.js'), alias: 'eComponentA2', identifier: 'ComponentA2'},
|
||
|
|
{from: _('/some/foo/b.js'), alias: 'eComponentB', identifier: 'ComponentB'},
|
||
|
|
{from: PROGRAM.name, alias: 'eTopLevelComponent', identifier: 'TopLevelComponent'},
|
||
|
|
],
|
||
|
|
importManager, sourceFile);
|
||
|
|
const outputString = output.toString();
|
||
|
|
expect(outputString).not.toContain(`eComponentA1`);
|
||
|
|
expect(outputString).not.toContain(`eComponentB`);
|
||
|
|
expect(outputString).not.toContain(`eTopLevelComponent`);
|
||
|
|
});
|
||
|
|
});
|
||
|
|
|
||
|
|
describe('addConstants', () => {
|
||
|
|
it('should insert the given constants after imports in the source file', () => {
|
||
|
|
const {renderer, program} = setup(PROGRAM);
|
||
|
|
const file = program.getSourceFile('some/file.js');
|
||
|
|
if (file === undefined) {
|
||
|
|
throw new Error(`Could not find source file`);
|
||
|
|
}
|
||
|
|
const output = new MagicString(PROGRAM.contents);
|
||
|
|
renderer.addConstants(output, 'var x = 3;', file);
|
||
|
|
expect(output.toString()).toContain(`
|
||
|
|
}(this, (function (exports,someSideEffect,localDep,core) {
|
||
|
|
var x = 3;
|
||
|
|
'use strict';
|
||
|
|
var A = (function() {`);
|
||
|
|
});
|
||
|
|
|
||
|
|
it('should insert constants after inserted imports',
|
||
|
|
() => {
|
||
|
|
// This test (from ESM5) is not needed as constants go in the body
|
||
|
|
// of the UMD IIFE, so cannot come before imports.
|
||
|
|
});
|
||
|
|
});
|
||
|
|
|
||
|
|
describe('rewriteSwitchableDeclarations', () => {
|
||
|
|
it('should switch marked declaration initializers', () => {
|
||
|
|
const {renderer, program, sourceFile, switchMarkerAnalyses} = setup(PROGRAM);
|
||
|
|
const file = program.getSourceFile('some/file.js');
|
||
|
|
if (file === undefined) {
|
||
|
|
throw new Error(`Could not find source file`);
|
||
|
|
}
|
||
|
|
const output = new MagicString(PROGRAM.contents);
|
||
|
|
renderer.rewriteSwitchableDeclarations(
|
||
|
|
output, file, switchMarkerAnalyses.get(sourceFile) !.declarations);
|
||
|
|
expect(output.toString())
|
||
|
|
.not.toContain(`var compileNgModuleFactory = compileNgModuleFactory__PRE_R3__;`);
|
||
|
|
expect(output.toString())
|
||
|
|
.toContain(`var badlyFormattedVariable = __PRE_R3__badlyFormattedVariable;`);
|
||
|
|
expect(output.toString())
|
||
|
|
.toContain(`var compileNgModuleFactory = compileNgModuleFactory__POST_R3__;`);
|
||
|
|
expect(output.toString())
|
||
|
|
.toContain(`function compileNgModuleFactory__PRE_R3__(injector, options, moduleType) {`);
|
||
|
|
expect(output.toString())
|
||
|
|
.toContain(`function compileNgModuleFactory__POST_R3__(injector, options, moduleType) {`);
|
||
|
|
});
|
||
|
|
});
|
||
|
|
|
||
|
|
describe('addDefinitions', () => {
|
||
|
|
it('should insert the definitions directly before the return statement of the class IIFE',
|
||
|
|
() => {
|
||
|
|
const {renderer, decorationAnalyses, sourceFile} = setup(PROGRAM);
|
||
|
|
const output = new MagicString(PROGRAM.contents);
|
||
|
|
const compiledClass =
|
||
|
|
decorationAnalyses.get(sourceFile) !.compiledClasses.find(c => c.name === 'A') !;
|
||
|
|
renderer.addDefinitions(output, compiledClass, 'SOME DEFINITION TEXT');
|
||
|
|
expect(output.toString()).toContain(`
|
||
|
|
A.prototype.ngDoCheck = function() {
|
||
|
|
//
|
||
|
|
};
|
||
|
|
SOME DEFINITION TEXT
|
||
|
|
return A;
|
||
|
|
`);
|
||
|
|
});
|
||
|
|
|
||
|
|
it('should error if the compiledClass is not valid', () => {
|
||
|
|
const {renderer, sourceFile, program} = setup(PROGRAM);
|
||
|
|
const output = new MagicString(PROGRAM.contents);
|
||
|
|
|
||
|
|
const noIifeDeclaration =
|
||
|
|
getDeclaration(program, sourceFile.fileName, 'NoIife', ts.isFunctionDeclaration);
|
||
|
|
const mockNoIifeClass: any = {declaration: noIifeDeclaration, name: 'NoIife'};
|
||
|
|
expect(() => renderer.addDefinitions(output, mockNoIifeClass, 'SOME DEFINITION TEXT'))
|
||
|
|
.toThrowError(
|
||
|
|
'Compiled class declaration is not inside an IIFE: NoIife in /some/file.js');
|
||
|
|
|
||
|
|
const badIifeDeclaration =
|
||
|
|
getDeclaration(program, sourceFile.fileName, 'BadIife', ts.isVariableDeclaration);
|
||
|
|
const mockBadIifeClass: any = {declaration: badIifeDeclaration, name: 'BadIife'};
|
||
|
|
expect(() => renderer.addDefinitions(output, mockBadIifeClass, 'SOME DEFINITION TEXT'))
|
||
|
|
.toThrowError(
|
||
|
|
'Compiled class wrapper IIFE does not have a return statement: BadIife in /some/file.js');
|
||
|
|
});
|
||
|
|
});
|
||
|
|
|
||
|
|
describe('removeDecorators', () => {
|
||
|
|
|
||
|
|
it('should delete the decorator (and following comma) that was matched in the analysis', () => {
|
||
|
|
const {renderer, decorationAnalyses, sourceFile} = setup(PROGRAM);
|
||
|
|
const output = new MagicString(PROGRAM.contents);
|
||
|
|
const compiledClass =
|
||
|
|
decorationAnalyses.get(sourceFile) !.compiledClasses.find(c => c.name === 'A') !;
|
||
|
|
const decorator = compiledClass.decorators[0];
|
||
|
|
const decoratorsToRemove = new Map<ts.Node, ts.Node[]>();
|
||
|
|
decoratorsToRemove.set(decorator.node.parent !, [decorator.node]);
|
||
|
|
renderer.removeDecorators(output, decoratorsToRemove);
|
||
|
|
expect(output.toString())
|
||
|
|
.not.toContain(`{ type: core.Directive, args: [{ selector: '[a]' }] },`);
|
||
|
|
expect(output.toString()).toContain(`{ type: OtherA }`);
|
||
|
|
expect(output.toString()).toContain(`{ type: core.Directive, args: [{ selector: '[b]' }] }`);
|
||
|
|
expect(output.toString()).toContain(`{ type: OtherB }`);
|
||
|
|
expect(output.toString()).toContain(`{ type: core.Directive, args: [{ selector: '[c]' }] }`);
|
||
|
|
});
|
||
|
|
|
||
|
|
|
||
|
|
it('should delete the decorator (but cope with no trailing comma) that was matched in the analysis',
|
||
|
|
() => {
|
||
|
|
const {renderer, decorationAnalyses, sourceFile} = setup(PROGRAM);
|
||
|
|
const output = new MagicString(PROGRAM.contents);
|
||
|
|
const compiledClass =
|
||
|
|
decorationAnalyses.get(sourceFile) !.compiledClasses.find(c => c.name === 'B') !;
|
||
|
|
const decorator = compiledClass.decorators[0];
|
||
|
|
const decoratorsToRemove = new Map<ts.Node, ts.Node[]>();
|
||
|
|
decoratorsToRemove.set(decorator.node.parent !, [decorator.node]);
|
||
|
|
renderer.removeDecorators(output, decoratorsToRemove);
|
||
|
|
expect(output.toString())
|
||
|
|
.toContain(`{ type: core.Directive, args: [{ selector: '[a]' }] },`);
|
||
|
|
expect(output.toString()).toContain(`{ type: OtherA }`);
|
||
|
|
expect(output.toString())
|
||
|
|
.not.toContain(`{ type: core.Directive, args: [{ selector: '[b]' }] }`);
|
||
|
|
expect(output.toString()).toContain(`{ type: OtherB }`);
|
||
|
|
expect(output.toString())
|
||
|
|
.toContain(`{ type: core.Directive, args: [{ selector: '[c]' }] }`);
|
||
|
|
});
|
||
|
|
|
||
|
|
|
||
|
|
it('should delete the decorator (and its container if there are not other decorators left) that was matched in the analysis',
|
||
|
|
() => {
|
||
|
|
const {renderer, decorationAnalyses, sourceFile} = setup(PROGRAM);
|
||
|
|
const output = new MagicString(PROGRAM.contents);
|
||
|
|
const compiledClass =
|
||
|
|
decorationAnalyses.get(sourceFile) !.compiledClasses.find(c => c.name === 'C') !;
|
||
|
|
const decorator = compiledClass.decorators[0];
|
||
|
|
const decoratorsToRemove = new Map<ts.Node, ts.Node[]>();
|
||
|
|
decoratorsToRemove.set(decorator.node.parent !, [decorator.node]);
|
||
|
|
renderer.removeDecorators(output, decoratorsToRemove);
|
||
|
|
renderer.addDefinitions(output, compiledClass, 'SOME DEFINITION TEXT');
|
||
|
|
expect(output.toString())
|
||
|
|
.toContain(`{ type: core.Directive, args: [{ selector: '[a]' }] },`);
|
||
|
|
expect(output.toString()).toContain(`{ type: OtherA }`);
|
||
|
|
expect(output.toString())
|
||
|
|
.toContain(`{ type: core.Directive, args: [{ selector: '[b]' }] }`);
|
||
|
|
expect(output.toString()).toContain(`{ type: OtherB }`);
|
||
|
|
expect(output.toString()).not.toContain(`C.decorators`);
|
||
|
|
});
|
||
|
|
|
||
|
|
});
|
||
|
|
|
||
|
|
describe('[__decorate declarations]', () => {
|
||
|
|
it('should delete the decorator (and following comma) that was matched in the analysis', () => {
|
||
|
|
const {renderer, decorationAnalyses, sourceFile} = setup(PROGRAM_DECORATE_HELPER);
|
||
|
|
const output = new MagicString(PROGRAM_DECORATE_HELPER.contents);
|
||
|
|
const compiledClass =
|
||
|
|
decorationAnalyses.get(sourceFile) !.compiledClasses.find(c => c.name === 'A') !;
|
||
|
|
const decorator = compiledClass.decorators.find(d => d.name === 'Directive') !;
|
||
|
|
const decoratorsToRemove = new Map<ts.Node, ts.Node[]>();
|
||
|
|
decoratorsToRemove.set(decorator.node.parent !, [decorator.node]);
|
||
|
|
renderer.removeDecorators(output, decoratorsToRemove);
|
||
|
|
expect(output.toString()).not.toContain(`core.Directive({ selector: '[a]' }),`);
|
||
|
|
expect(output.toString()).toContain(`OtherA()`);
|
||
|
|
expect(output.toString()).toContain(`core.Directive({ selector: '[b]' })`);
|
||
|
|
expect(output.toString()).toContain(`OtherB()`);
|
||
|
|
expect(output.toString()).toContain(`core.Directive({ selector: '[c]' })`);
|
||
|
|
});
|
||
|
|
|
||
|
|
it('should delete the decorator (but cope with no trailing comma) that was matched in the analysis',
|
||
|
|
() => {
|
||
|
|
const {renderer, decorationAnalyses, sourceFile} = setup(PROGRAM_DECORATE_HELPER);
|
||
|
|
const output = new MagicString(PROGRAM_DECORATE_HELPER.contents);
|
||
|
|
const compiledClass =
|
||
|
|
decorationAnalyses.get(sourceFile) !.compiledClasses.find(c => c.name === 'B') !;
|
||
|
|
const decorator = compiledClass.decorators.find(d => d.name === 'Directive') !;
|
||
|
|
const decoratorsToRemove = new Map<ts.Node, ts.Node[]>();
|
||
|
|
decoratorsToRemove.set(decorator.node.parent !, [decorator.node]);
|
||
|
|
renderer.removeDecorators(output, decoratorsToRemove);
|
||
|
|
expect(output.toString()).toContain(`core.Directive({ selector: '[a]' }),`);
|
||
|
|
expect(output.toString()).toContain(`OtherA()`);
|
||
|
|
expect(output.toString()).not.toContain(`core.Directive({ selector: '[b]' })`);
|
||
|
|
expect(output.toString()).toContain(`OtherB()`);
|
||
|
|
expect(output.toString()).toContain(`core.Directive({ selector: '[c]' })`);
|
||
|
|
});
|
||
|
|
|
||
|
|
|
||
|
|
it('should delete the decorator (and its container if there are no other decorators left) that was matched in the analysis',
|
||
|
|
() => {
|
||
|
|
const {renderer, decorationAnalyses, sourceFile} = setup(PROGRAM_DECORATE_HELPER);
|
||
|
|
const output = new MagicString(PROGRAM_DECORATE_HELPER.contents);
|
||
|
|
const compiledClass =
|
||
|
|
decorationAnalyses.get(sourceFile) !.compiledClasses.find(c => c.name === 'C') !;
|
||
|
|
const decorator = compiledClass.decorators.find(d => d.name === 'Directive') !;
|
||
|
|
const decoratorsToRemove = new Map<ts.Node, ts.Node[]>();
|
||
|
|
decoratorsToRemove.set(decorator.node.parent !, [decorator.node]);
|
||
|
|
renderer.removeDecorators(output, decoratorsToRemove);
|
||
|
|
expect(output.toString()).toContain(`core.Directive({ selector: '[a]' }),`);
|
||
|
|
expect(output.toString()).toContain(`OtherA()`);
|
||
|
|
expect(output.toString()).toContain(`core.Directive({ selector: '[b]' })`);
|
||
|
|
expect(output.toString()).toContain(`OtherB()`);
|
||
|
|
expect(output.toString()).not.toContain(`core.Directive({ selector: '[c]' })`);
|
||
|
|
expect(output.toString()).not.toContain(`C = tslib_1.__decorate([`);
|
||
|
|
expect(output.toString()).toContain(`function C() {\n }\n return C;`);
|
||
|
|
});
|
||
|
|
});
|
||
|
|
});
|