Add regression tests

amcasey · amcasey · commit 1e93967ec416 · 2018-01-04T15:34:19.000-08:00
diff --git a/src/services/codefixes/importFixes.ts b/src/services/codefixes/importFixes.ts
@@ -798,6 +798,7 @@ namespace ts.codefix {
                         }
                     }
                     else if (isExportSpecifier(declaration)) {
+                        Debug.assert(declaration.name.escapedText === InternalSymbolName.Default);
                         if (declaration.propertyName) {
                             return declaration.propertyName.escapedText;
                         }
diff --git a/tests/cases/fourslash/importNameCodeFixDefaultExport2.ts b/tests/cases/fourslash/importNameCodeFixDefaultExport2.ts
@@ -0,0 +1,21 @@
+/// <reference path="fourslash.ts" />
+
+// @allowJs: true
+// @checkJs: true
+
+// @Filename: /lib.js
+////class Base { }
+////export default Base;
+
+// @Filename: /test.js
+////[|class Derived extends Base { }|]
+
+goTo.file("/test.js");
+verify.importFixAtPosition([
+`// @ts-ignore
+class Derived extends Base { }`,
+`// @ts-nocheck
+class Derived extends Base { }`,
+`import Base from "./lib";
+
+class Derived extends Base { }`,]);
diff --git a/tests/cases/fourslash/importNameCodeFixDefaultExport3.ts b/tests/cases/fourslash/importNameCodeFixDefaultExport3.ts
@@ -0,0 +1,21 @@
+/// <reference path="fourslash.ts" />
+
+// @allowJs: true
+// @checkJs: true
+
+// @Filename: /lib.js
+////class Base { }
+////export { Base as default };
+
+// @Filename: /test.js
+////[|class Derived extends Base { }|]
+
+goTo.file("/test.js");
+verify.importFixAtPosition([
+`// @ts-ignore
+class Derived extends Base { }`,
+`// @ts-nocheck
+class Derived extends Base { }`,
+`import Base from "./lib";
+
+class Derived extends Base { }`,]);

Original file line number	Diff line number	Diff line change
`@@ -798,6 +798,7 @@ namespace ts.codefix {`
`798`	`798`	`}`
`799`	`799`	`}`
`800`	`800`	`else if (isExportSpecifier(declaration)) {`
	`801`	`+ Debug.assert(declaration.name.escapedText === InternalSymbolName.Default);`
`801`	`802`	`if (declaration.propertyName) {`
`802`	`803`	`return declaration.propertyName.escapedText;`
`803`	`804`	`}`