Do not remove generator markers (#463)

Fixes #462
alangpierce · Aug 26, 2019 · 8882acd · 8882acd
1 parent 6ed6dff
commit 8882acd
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 0 deletions.
diff --git a/src/util/getClassInfo.ts b/src/util/getClassInfo.ts
@@ -257,6 +257,7 @@ function isAccessModifier(token: Token): boolean {
     tt._private,
     tt._protected,
     tt._abstract,
+    tt.star,
   ].includes(token.type);
 }
 

diff --git a/test/sucrase-test.ts b/test/sucrase-test.ts
@@ -249,6 +249,44 @@ describe("sucrase", () => {
     );
   });
 
+  it("handles generator class methods", () => {
+    assertResult(
+      `
+      class A {
+        *foo() {
+        }
+      }
+    `,
+      `"use strict";
+      class A {
+        *foo() {
+        }
+      }
+    `,
+      {transforms: ["jsx", "imports", "typescript"]},
+    );
+  });
+
+  it("handles async generator class methods", () => {
+    assertResult(
+      `
+      class C {
+        async *m() {
+          yield await 1;
+        }
+      }
+    `,
+      `"use strict";
+      class C {
+        async *m() {
+          yield await 1;
+        }
+      }
+    `,
+      {transforms: ["jsx", "imports", "typescript"]},
+    );
+  });
+
   it("removes numeric separators from number literals", () => {
     assertResult(
       `