71ada483bf
Remove unnecessary private method `getDeclarationFromNode` and moved some logic to utils instead so that it can be tested in isolation of the Language Service infrastructure. The use of typechecker to check the directive is also not necessary, since resolve.getNonNormalizedDirectiveMetadata() will check if the directive is actually an Angular entity. PR Close #32156
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 {getClassDeclFromTemplateNode} from '../src/template';
|
|
import {toh} from './test_data';
|
|
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 {
|
|
return getClassDeclFromTemplateNode(node) || 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'], toh);
|
|
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 {
|
|
const candidate = getClassDeclFromTemplateNode(node);
|
|
if (candidate) {
|
|
return candidate;
|
|
}
|
|
return node.forEachChild(visit);
|
|
});
|
|
expect(classDecl).toBeTruthy();
|
|
expect(ts.isClassDeclaration(classDecl !)).toBe(true);
|
|
expect((classDecl as ts.ClassDeclaration).name !.text).toBe('AppComponent');
|
|
});
|
|
});
|