angular/packages/compiler/test/output/output_jit_spec.ts

/**
 * @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 {EmitterVisitorContext} from '@angular/compiler/src/output/abstract_emitter';
import * as o from '@angular/compiler/src/output/output_ast';
import {JitEmitterVisitor, JitEvaluator} from '@angular/compiler/src/output/output_jit';
import {R3JitReflector} from '@angular/compiler/src/render3/r3_jit';
import {newArray} from '@angular/compiler/src/util';

describe('Output JIT', () => {
  describe('regression', () => {
    it('should generate unique argument names', () => {
      const externalIds = newArray(10, 1).map(
        (_, index) => new o.ExternalReference('@angular/core', `id_${index}_`),
      );
      const externalIds1 = newArray(10, 1).map(
        (_, index) => new o.ExternalReference('@angular/core', `id_${index}_1`),
      );
      const ctx = EmitterVisitorContext.createRoot();
      const reflectorContext: {[key: string]: string} = {};
      for (const {name} of externalIds) {
        reflectorContext[name!] = name!;
      }
      for (const {name} of externalIds1) {
        reflectorContext[name!] = name!;
      }
      const converter = new JitEmitterVisitor(new R3JitReflector(reflectorContext));
      converter.visitAllStatements(
        [o.literalArr([...externalIds1, ...externalIds].map((id) => o.importExpr(id))).toStmt()],
        ctx,
      );
      const args = converter.getArgs();
      expect(Object.keys(args).length).toBe(20);
    });
  });

  it('should use strict mode', () => {
    const evaluator = new JitEvaluator();
    expect(() => {
      evaluator.evaluateStatements(
        'http://angular.io/something.ts',
        [
          // Set an undeclared variable
          // foo = "bar";
          o.variable('foo').equals(o.literal('bar')).toStmt(),
        ],
        new R3JitReflector({}),
        false,
      );
    }).toThrowError();
  });

  it('should not add more than one strict mode statement if there is already one present', () => {
    const converter = new JitEmitterVisitor(new R3JitReflector({}));
    const ctx = EmitterVisitorContext.createRoot();
    converter.visitAllStatements([o.literal('use strict').toStmt()], ctx);
    const matches = ctx.toSource().match(/'use strict';/g)!;
    expect(matches.length).toBe(1);
  });
});