packages/compiler-cli/src/ngtsc/util/test/visitor_spec.ts - main - githubmirror/angular

packages/compiler-cli/src/ngtsc/util/test/visitor_spec.ts
108 строк3 KB
Kristiyan Kostadinov
feat(core): add support for TypeScript 6
17 фев 2026, 19:40
17 фев 2026, 19:4081cabc1
/**
 * @license
 * Copyright Google LLC 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.dev/license
 */
import ts from 'typescript';

import {absoluteFrom, getSourceFileOrError} from '../../file_system';
import {runInEachFileSystem} from '../../file_system/testing';
import {makeProgram} from '../../testing';
import {visit, VisitListEntryResult, Visitor} from '../src/visitor';

class TestAstVisitor extends Visitor {
  override visitClassDeclaration(
    node: ts.ClassDeclaration,
  ): VisitListEntryResult<ts.Statement, ts.ClassDeclaration> {
    const name = node.name!.text;
    const statics = node.members.filter((member) => {
      const modifiers = ts.canHaveModifiers(member) ? ts.getModifiers(member) : undefined;
      return (modifiers || []).some((mod) => mod.kind === ts.SyntaxKind.StaticKeyword);
    });
    const idStatic = statics.find(
      (el) => ts.isPropertyDeclaration(el) && ts.isIdentifier(el.name) && el.name.text === 'id',
    ) as ts.PropertyDeclaration | undefined;
    if (idStatic !== undefined) {
      return {
        node,
        before: [
          ts.factory.createVariableStatement(undefined, [
            ts.factory.createVariableDeclaration(
              `${name}_id`,
              undefined,
              undefined,
              idStatic.initializer,
            ),
          ]),
        ],
      };
    }
    return {node};
  }
}

function testTransformerFactory(context: ts.TransformationContext): ts.Transformer<ts.SourceFile> {
  return (file: ts.SourceFile) => visit(file, new TestAstVisitor(), context);
}

runInEachFileSystem(() => {
  describe('AST Visitor', () => {
    let _: typeof absoluteFrom;
    beforeEach(() => (_ = absoluteFrom));

    it('should add a statement before class in plain file', () => {
      const {program, host} = makeProgram([
        {name: _('/main.ts'), contents: `class A { static id = 3; }`},
      ]);
      const sf = getSourceFileOrError(program, _('/main.ts'));
      program.emit(sf, undefined, undefined, undefined, {before: [testTransformerFactory]});
      const main = host.readFile('/main.js');
      expect(main).toMatch(/var A_id = 3;/);
    });

    it('should add a statement before class inside function definition', () => {
      const {program, host} = makeProgram([
        {
          name: _('/main.ts'),
          contents: `
      export function foo() {
        var x = 3;
        class A { static id = 2; }
        return A;
      }
    `,
        },
      ]);
      const sf = getSourceFileOrError(program, _('/main.ts'));
      program.emit(sf, undefined, undefined, undefined, {before: [testTransformerFactory]});
      const main = host.readFile(_('/main.js'));
      expect(main).toMatch(/var x = 3;\s+var A_id = 2;\s+class A {/);
    });

    it('handles nested statements', () => {
      const {program, host} = makeProgram([
        {
          name: _('/main.ts'),
          contents: `
      export class A {
        static id = 3;

        foo() {
          class B {
            static id = 4;
          }
          return B;
        }
    }`,
        },
      ]);
      const sf = getSourceFileOrError(program, _('/main.ts'));
      program.emit(sf, undefined, undefined, undefined, {before: [testTransformerFactory]});
      const main = host.readFile(_('/main.js'));
      expect(main).toMatch(/var A_id = 3;\s+export class A {/);
      expect(main).toMatch(/var B_id = 4;\s+class B {/);
    });
  });
});