Merge pull request microsoft#24760 from krk/codefix-asterisk

mhegazy · web-flow · commit b4dea5eccefb · 2018-06-08T08:57:49.000-07:00
addMethodDeclaration codefix creates a generator function when target…
diff --git a/src/services/codefixes/helpers.ts b/src/services/codefixes/helpers.ts
@@ -111,7 +111,7 @@ namespace ts.codefix {
     }
 
     export function createMethodFromCallExpression(
-        { typeArguments, arguments: args }: CallExpression,
+        { typeArguments, arguments: args, parent: parent }: CallExpression,
         methodName: string,
         inJs: boolean,
         makeStatic: boolean,
@@ -120,7 +120,7 @@ namespace ts.codefix {
         return createMethod(
             /*decorators*/ undefined,
             /*modifiers*/ makeStatic ? [createToken(SyntaxKind.StaticKeyword)] : undefined,
-            /*asteriskToken*/ undefined,
+            /*asteriskToken*/ isYieldExpression(parent) ? createToken(SyntaxKind.AsteriskToken) : undefined,
             methodName,
             /*questionToken*/ undefined,
             /*typeParameters*/ inJs ? undefined : map(typeArguments, (_, i) =>
diff --git a/tests/cases/fourslash/codeFixAddMissingMember_generator_function.ts b/tests/cases/fourslash/codeFixAddMissingMember_generator_function.ts
@@ -0,0 +1,21 @@
+/// <reference path='fourslash.ts' />
+
+////class C {
+////    *method() {
+////        yield* this.y();
+////    }
+////}
+
+verify.codeFixAll({
+    fixId: "addMissingMember",
+    fixAllDescription: "Add all missing members",
+    newFileContent:
+        `class C {
+    *method() {
+        yield* this.y();
+    }
+    *y(): any {
+        throw new Error("Method not implemented.");
+    }
+}`,
+});
diff --git a/tests/cases/fourslash/codeFixAddMissingMember_non_generator_function.ts b/tests/cases/fourslash/codeFixAddMissingMember_non_generator_function.ts
@@ -0,0 +1,21 @@
+/// <reference path='fourslash.ts' />
+
+////class C {
+////    method() {
+////        yield* this.y();
+////    }
+////}
+
+verify.codeFixAll({
+    fixId: "addMissingMember",
+    fixAllDescription: "Add all missing members",
+    newFileContent:
+        `class C {
+    method() {
+        yield* this.y();
+    }
+    y(): any {
+        throw new Error("Method not implemented.");
+    }
+}`,
+});
