test: add tests to verify language-service supports output() function (#54217)

devversion · thePunderWoman · commit 02fdd8d6eec7 · 2024-02-05T15:08:35.000Z
Adds unit tests to verify that the language service supports the `output()` function with completions, and definition jumping. PR Close #54217
diff --git a/packages/language-service/test/completions_spec.ts b/packages/language-service/test/completions_spec.ts
@@ -342,6 +342,52 @@ describe('completions', () => {
     });
   });
 
+  describe('initializer-based output() API', () => {
+    const initializerOutputDirectiveWithUnionType = {
+      'Dir': `
+         @Directive({
+           selector: '[dir]',
+         })
+         export class Dir {
+           bla = output<string>();
+         }
+    `
+    };
+
+    it('should return event completion', () => {
+      const {templateFile} =
+          setup(`<button dir ></button>`, ``, initializerOutputDirectiveWithUnionType);
+      templateFile.moveCursorToText(`<button dir ¦>`);
+      const completions = templateFile.getCompletionsAtPosition();
+      expectContain(completions, DisplayInfoKind.EVENT, ['(bla)']);
+    });
+
+    it('should return property access completions', () => {
+      const {templateFile} =
+          setup(`<input dir (bla)="'foo'.">`, '', initializerOutputDirectiveWithUnionType);
+      templateFile.moveCursorToText(`dir (bla)="'foo'.¦">`);
+
+      const completions = templateFile.getCompletionsAtPosition();
+      expectContain(
+          completions, ts.ScriptElementKind.memberFunctionElement,
+          [`charAt`, 'toLowerCase', /* etc. */]);
+    });
+
+    it('should return completions of string literals, number literals, `true`, ' +
+           '`false`, `null` and `undefined`',
+       () => {
+         const {templateFile} =
+             setup(`<input dir (bla)="$event.">`, '', initializerOutputDirectiveWithUnionType);
+         templateFile.moveCursorToText('dir (bla)="$event.¦">');
+
+         const completions = templateFile.getCompletionsAtPosition();
+
+         expectContain(
+             completions, ts.ScriptElementKind.memberFunctionElement,
+             [`charAt`, 'toLowerCase', /* etc. */]);
+       });
+  });
+
   describe('for blocks', () => {
     const completionPrefixes = ['@', '@i'];
 
@@ -1634,7 +1680,7 @@ function setup(
   const env = LanguageServiceTestEnv.setup();
   const project = env.addProject('test', {
     'test.ts': `
-         import {Component, input, Directive, NgModule, Pipe, TemplateRef} from '@angular/core';
+         import {Component, input, output, Directive, NgModule, Pipe, TemplateRef} from '@angular/core';
 
          ${functionDeclarations}
 
diff --git a/packages/language-service/test/definitions_spec.ts b/packages/language-service/test/definitions_spec.ts
@@ -171,6 +171,13 @@ describe('definitions', () => {
        export class MyDir2 {
          @Output() someEvent = new EventEmitter<void>();
        }`,
+      'dir3.ts': `
+       import {Directive, output, EventEmitter} from '@angular/core';
+
+       @Directive({selector: '[dir]'})
+       export class MyDir3 {
+         someEvent = output();
+       }`,
       'app.ts': `
          import {Component, NgModule} from '@angular/core';
          import {CommonModule} from '@angular/common';
@@ -190,13 +197,15 @@ describe('definitions', () => {
     expect(template.contents.slice(textSpan.start, textSpan.start + textSpan.length))
         .toEqual('someEvent');
 
-    expect(definitions.length).toEqual(2);
-    const [def, def2] = definitions;
+    expect(definitions.length).toEqual(3);
+    const [def3, def2, def] = definitions;
     expect(def.textSpan).toContain('someEvent');
     expect(def2.textSpan).toContain('someEvent');
+    expect(def3.textSpan).toContain('someEvent');
+    expect(def3.contextSpan!).toBe('someEvent = output();');
     // TODO(atscott): investigate why the text span includes more than just 'someEvent'
     // assertTextSpans([def, def2], ['someEvent']);
-    assertFileNames([def, def2], ['dir2.ts', 'dir.ts']);
+    assertFileNames([def3, def2, def], ['dir3.ts', 'dir2.ts', 'dir.ts']);
   });
 
   it('should go to the pre-compiled style sheet', () => {
diff --git a/packages/language-service/test/type_definitions_spec.ts b/packages/language-service/test/type_definitions_spec.ts
@@ -74,6 +74,42 @@ describe('type definitions', () => {
     });
   });
 
+  describe('initializer-based output() API', () => {
+    it('return the definition for an output', () => {
+      initMockFileSystem('Native');
+      const files = {
+        'app.ts': `
+          import {Component, Directive, output} from '@angular/core';
+
+          @Directive({
+            selector: 'my-dir',
+            standalone: true
+          })
+          export class MyDir {
+            nameChanges = output<string>();
+          }
+
+          @Component({
+            templateUrl: 'app.html',
+            standalone: true,
+            imports: [MyDir],
+          })
+          export class AppCmp {
+            doSmth() {}
+          }
+        `,
+        'app.html': `Will be overridden`,
+      };
+      env = LanguageServiceTestEnv.setup();
+      const project = env.addProject('test', files);
+      const definitions = getTypeDefinitionsAndAssertBoundSpan(
+          project, {templateOverride: `<my-dir (name¦Changes)="doSmth()" />`});
+      expect(definitions!.length).toEqual(1);
+
+      assertTextSpans(definitions, ['EventEmitter']);
+      assertFileNames(definitions, ['index.d.ts']);
+    });
+  });
 
   function getTypeDefinitionsAndAssertBoundSpan(
       project: Project, {templateOverride}: {templateOverride: string}) {
