1140bbc25c
clang-format was recently updated and any PRs that touch files in the language service will have to reformat all the files. Instead of changing the formatting in those PRs, this PR formats all files in language-service package once and for all. PR Close #36426
51 lines
1.8 KiB
TypeScript
51 lines
1.8 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 * as ts from 'typescript';
|
|
import {getClassDeclFromDecoratorProp} from '../src/template';
|
|
import {MockTypescriptHost} from './test_utils';
|
|
|
|
describe('getClassDeclFromTemplateNode', () => {
|
|
it('should find class declaration in syntax-only mode', () => {
|
|
const sourceFile = ts.createSourceFile(
|
|
'foo.ts', `
|
|
@Component({
|
|
template: '<div></div>'
|
|
})
|
|
class MyComponent {}`,
|
|
ts.ScriptTarget.ES2015, true /* setParentNodes */);
|
|
function visit(node: ts.Node): ts.ClassDeclaration|undefined {
|
|
if (ts.isPropertyAssignment(node)) {
|
|
return getClassDeclFromDecoratorProp(node);
|
|
}
|
|
return node.forEachChild(visit);
|
|
}
|
|
const classDecl = sourceFile.forEachChild(visit);
|
|
expect(classDecl).toBeTruthy();
|
|
expect(classDecl!.kind).toBe(ts.SyntaxKind.ClassDeclaration);
|
|
expect((classDecl as ts.ClassDeclaration).name!.text).toBe('MyComponent');
|
|
});
|
|
|
|
|
|
it('should return class declaration for AppComponent', () => {
|
|
const host = new MockTypescriptHost(['/app/app.component.ts']);
|
|
const tsLS = ts.createLanguageService(host);
|
|
const sourceFile = tsLS.getProgram()!.getSourceFile('/app/app.component.ts');
|
|
expect(sourceFile).toBeTruthy();
|
|
const classDecl = sourceFile!.forEachChild(function visit(node): ts.Node|undefined {
|
|
if (ts.isPropertyAssignment(node)) {
|
|
return getClassDeclFromDecoratorProp(node);
|
|
}
|
|
return node.forEachChild(visit);
|
|
});
|
|
expect(classDecl).toBeTruthy();
|
|
expect(ts.isClassDeclaration(classDecl!)).toBe(true);
|
|
expect((classDecl as ts.ClassDeclaration).name!.text).toBe('AppComponent');
|
|
});
|
|
});
|