53 lines
2.0 KiB
TypeScript
53 lines
2.0 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 {Reference, TypeScriptReflectionHost, staticallyResolve} from '../../../ngtsc/metadata';
|
||
|
|
import {getDeclaration, makeProgram} from '../../../ngtsc/testing/in_memory_typescript';
|
||
|
|
import {NgccReferencesRegistry} from '../../src/analysis/ngcc_references_registry';
|
||
|
|
|
||
|
|
describe('NgccReferencesRegistry', () => {
|
||
|
|
it('should return a mapping from resolved reference identifiers to their declarations', () => {
|
||
|
|
const {program} = makeProgram([{
|
||
|
|
name: 'index.ts',
|
||
|
|
contents: `
|
||
|
|
export class SomeClass {}
|
||
|
|
export function someFunction() {}
|
||
|
|
export const someVariable = 42;
|
||
|
|
|
||
|
|
export const testArray = [SomeClass, someFunction, someVariable];
|
||
|
|
`
|
||
|
|
}]);
|
||
|
|
|
||
|
|
const checker = program.getTypeChecker();
|
||
|
|
|
||
|
|
const testArrayDeclaration =
|
||
|
|
getDeclaration(program, 'index.ts', 'testArray', ts.isVariableDeclaration);
|
||
|
|
const someClassDecl = getDeclaration(program, 'index.ts', 'SomeClass', ts.isClassDeclaration);
|
||
|
|
const someFunctionDecl =
|
||
|
|
getDeclaration(program, 'index.ts', 'someFunction', ts.isFunctionDeclaration);
|
||
|
|
const someVariableDecl =
|
||
|
|
getDeclaration(program, 'index.ts', 'someVariable', ts.isVariableDeclaration);
|
||
|
|
const testArrayExpression = testArrayDeclaration.initializer !;
|
||
|
|
|
||
|
|
const host = new TypeScriptReflectionHost(checker);
|
||
|
|
const registry = new NgccReferencesRegistry(host);
|
||
|
|
|
||
|
|
const references =
|
||
|
|
staticallyResolve(testArrayExpression, host, checker) as Reference<ts.Declaration>[];
|
||
|
|
registry.add(...references);
|
||
|
|
|
||
|
|
const map = registry.getDeclarationMap();
|
||
|
|
expect(map.size).toEqual(2);
|
||
|
|
expect(map.get(someClassDecl.name !) !.node).toBe(someClassDecl);
|
||
|
|
expect(map.get(someFunctionDecl.name !) !.node).toBe(someFunctionDecl);
|
||
|
|
expect(map.has(someVariableDecl.name as ts.Identifier)).toBe(false);
|
||
|
|
});
|
||
|
|
});
|