add test cases for auto-import (#214)

Author: mizdra

packages/ts-plugin/e2e/named-exports.test.ts

@@ -9,6 +9,7 @@ import {
   launchTsserver,
   mergeSpanGroups,
   normalizeCodeFixActions,
+  normalizeCompletionDetails,
   normalizeCompletionEntry,
   normalizeRefItems,
   normalizeSpanGroups,
@@ -259,6 +260,44 @@ describe('supports completions', async () => {
         normalizeCompletionEntry(res.body?.entries.filter((entry) => entry.name === entryName) ?? []),
       ).toStrictEqual(normalizeCompletionEntry(expected));
     });
+    test('if the user completes `styles`, auto-import the namespace import', async () => {
+      const res = await tsserver.sendCompletionDetails({
+        file: iff.paths['index.ts'],
+        line: 1,
+        offset: 7,
+        entryNames: [
+          {
+            name: 'styles',
+            source: './a.module.css',
+            data: {
+              exportName: ts.InternalSymbolName.Default,
+              fileName: iff.paths['a.module.css'],
+              moduleSpecifier: './a.module.css',
+            } satisfies ts.CompletionEntryData,
+          },
+        ],
+      });
+      expect(normalizeCompletionDetails(res.body!)).toStrictEqual([
+        {
+          codeActions: [
+            {
+              changes: [
+                {
+                  fileName: formatPath(iff.paths['index.ts']),
+                  textChanges: [
+                    {
+                      start: { line: 1, offset: 1 },
+                      end: { line: 1, offset: 1 },
+                      newText: `import * as styles from "./a.module.css";${ts.sys.newLine}${ts.sys.newLine}`,
+                    },
+                  ],
+                },
+              ],
+            },
+          ],
+        },
+      ]);
+    });
   });
   describe('prioritize named imports if prioritizeNamedImports is true', async () => {
     const iff = await baseIff.fork({

packages/ts-plugin/e2e/test-util/tsserver.ts

@@ -28,6 +28,9 @@ interface Tsserver {
   sendCompletionInfo(
     args: server.protocol.CompletionsRequest['arguments'],
   ): Promise<server.protocol.CompletionInfoResponse>;
+  sendCompletionDetails(
+    args: server.protocol.CompletionDetailsRequest['arguments'],
+  ): Promise<server.protocol.CompletionDetailsResponse>;
   sendGetCodeFixes(args: server.protocol.CodeFixRequest['arguments']): Promise<server.protocol.CodeFixResponse>;
 }
 
@@ -77,6 +80,7 @@ export function launchTsserver(): Tsserver {
       sendRequest(ts.server.protocol.CommandTypes.GetApplicableRefactors, args),
     sendGetEditsForRefactor: async (args) => sendRequest(ts.server.protocol.CommandTypes.GetEditsForRefactor, args),
     sendCompletionInfo: async (args) => sendRequest(ts.server.protocol.CommandTypes.CompletionInfo, args),
+    sendCompletionDetails: async (args) => sendRequest(ts.server.protocol.CommandTypes.CompletionDetails, args),
     sendGetCodeFixes: async (args) => sendRequest(ts.server.protocol.CommandTypes.GetCodeFixes, args),
   };
 }
@@ -200,6 +204,30 @@ export function normalizeCompletionEntry(entries: readonly SimplifiedCompletionE
     );
 }
 
+type SimplifiedCodeAction = {
+  changes: ts.server.protocol.FileCodeEdits[];
+};
+
+type SimplifiedCompletionDetails = {
+  codeActions?: SimplifiedCodeAction[];
+};
+
+export function normalizeCompletionDetails(
+  entries: readonly SimplifiedCompletionDetails[],
+): SimplifiedCompletionDetails[] {
+  return entries.map((entry) => {
+    return {
+      ...(entry.codeActions ?
+        {
+          codeActions: entry.codeActions.map((action) => {
+            return { changes: action.changes };
+          }),
+        }
+      : {}),
+    };
+  });
+}
+
 type SimplifiedCodeFixAction = {
   fixName: string;
   changes: ts.server.protocol.FileCodeEdits[];