angular/packages/compiler-cli/test/ngtsc/attach_source_location_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 {setEnableTemplateSourceLocations} from '@angular/compiler';
import {runInEachFileSystem} from '../../src/ngtsc/file_system/testing';
import {loadStandardTestFiles} from '../../src/ngtsc/testing';
import {NgtscTestEnvironment} from './env';

const testFiles = loadStandardTestFiles({fakeCommon: true});

runInEachFileSystem(() => {
  describe('source location instruction generation', () => {
    let env!: NgtscTestEnvironment;

    beforeEach(() => {
      setEnableTemplateSourceLocations(true);
      env = NgtscTestEnvironment.setup(testFiles);
      env.tsconfig();
    });

    afterEach(() => {
      setEnableTemplateSourceLocations(false);
    });

    it('should attach the source location in an inline template', () => {
      env.write(
        `test.ts`,
        `
          import {Component} from '@angular/core';

          @Component({
            template: \`
              <div><span>
                <strong>Hello</strong>
              </span></div>
            \`,
          })
          class Comp {}
         `,
      );
      env.driveMain();
      const content = env.getContents('test.js');
      expect(content).toContain('ɵɵelementStart(0, "div")(1, "span")(2, "strong");');
      expect(content).toContain(
        'ɵɵattachSourceLocations("test.ts", [[0, 114, 5, 14], [1, 119, 5, 19], [2, 142, 6, 16]]);',
      );
    });

    it('should attach the source location in an external template', () => {
      env.write(
        'test.html',
        `
        <div><span>
          <strong>Hello</strong>
        </span></div>
      `,
      );

      env.write(
        `test.ts`,
        `
          import {Component} from '@angular/core';

          @Component({templateUrl: './test.html'})
          class Comp {}
        `,
      );
      env.driveMain();
      const content = env.getContents('test.js');
      expect(content).toContain('ɵɵelementStart(0, "div")(1, "span")(2, "strong");');
      expect(content).toContain(
        'ɵɵattachSourceLocations("test.html", [[0, 9, 1, 8], [1, 14, 1, 13], [2, 31, 2, 10]]);',
      );
    });

    it('should attach the source location to structural directives', () => {
      env.write(
        `test.ts`,
        `
          import {Component} from '@angular/core';
          import {CommonModule} from '@angular/common';

          @Component({
            imports: [CommonModule],
            template: \`
              <div *ngIf="true">
                <span></span>
              </div>
            \`,
          })
          class Comp {}
        `,
      );
      env.driveMain();
      const content = env.getContents('test.js');
      expect(content).toContain('ɵɵtemplate(0,');
      expect(content).toContain('ɵɵelementStart(0, "div");');
      expect(content).toContain('ɵɵelement(1, "span");');
      expect(content).toContain(
        'ɵɵattachSourceLocations("test.ts", [[0, 207, 7, 14], [1, 242, 8, 16]]);',
      );
    });

    it('should not attach the source location to ng-container', () => {
      env.write(
        `test.ts`,
        `
          import {Component} from '@angular/core';

          @Component({
            template: \`
              <ng-container>
                <div>
                  <ng-container>
                    <span></span>
                  </ng-container>
                </div>
              </ng-container>
            \`,
          })
          class Comp {}
        `,
      );
      env.driveMain();
      const content = env.getContents('test.js');
      expect(content).toContain('ɵɵelementContainerStart(0);');
      expect(content).toContain('ɵɵelementStart(1, "div");');
      expect(content).toContain('ɵɵelementContainerStart(2);');
      expect(content).toContain('ɵɵelement(3, "span");');
      expect(content).toContain(
        'ɵɵattachSourceLocations("test.ts", [[1, 145, 6, 16], [3, 204, 8, 20]]);',
      );
    });
  });
});