Merge pull request #524 from gitKrystan/fix-521

Retain async / generator behavior on class methods (Fixes #521)

Author: wycats

transforms/ember-object/__testfixtures__/async-function-property.input.js

@@ -0,0 +1,5 @@
+const Foo = Test.extend({
+  myAsyncMethod: async function() {
+    await Promise.resolve('hello');
+  }
+});

transforms/ember-object/__testfixtures__/async-function-property.output.js

@@ -0,0 +1,8 @@
+import classic from 'ember-classic-decorator';
+
+@classic
+class Foo extends Test {
+  async myAsyncMethod() {
+    await Promise.resolve('hello');
+  }
+}

transforms/ember-object/__testfixtures__/async-method.input.js

@@ -0,0 +1,5 @@
+const Foo = Test.extend({
+  async myAsyncMethod() {
+    await Promise.resolve('hello');
+  }
+});

transforms/ember-object/__testfixtures__/async-method.output.js

@@ -0,0 +1,8 @@
+import classic from 'ember-classic-decorator';
+
+@classic
+class Foo extends Test {
+  async myAsyncMethod() {
+    await Promise.resolve('hello');
+  }
+}

transforms/ember-object/__testfixtures__/generator-method.input.js

@@ -0,0 +1,5 @@
+const Foo = Test.extend({
+  *gen() {
+    yield 'hello';
+  }
+});

transforms/ember-object/__testfixtures__/generator-method.output.js

@@ -0,0 +1,8 @@
+import classic from 'ember-classic-decorator';
+
+@classic
+class Foo extends Test {
+  *gen() {
+    yield 'hello';
+  }
+}

transforms/ember-object/__testfixtures__/generator-property.input.js

@@ -0,0 +1,5 @@
+const Foo = Test.extend({
+  gen: function*() {
+    yield 'hello';
+  }
+});

transforms/ember-object/__testfixtures__/generator-property.output.js

@@ -0,0 +1,8 @@
+import classic from 'ember-classic-decorator';
+
+@classic
+class Foo extends Test {
+  *gen() {
+    yield 'hello';
+  }
+}

transforms/helpers/eo-prop/private/function-expression.ts

@@ -51,6 +51,8 @@ export default class EOFunctionExpressionProp extends AbstractEOProp<
         body: this.value.body,
         comments: this.comments,
         decorators: this.existingDecorators,
+        generator: this.value.generator ?? false,
+        async: this.value.async ?? false,
       }),
       this.replaceSuperWithUndefined
     );

transforms/helpers/eo-prop/private/method.ts

@@ -54,6 +54,8 @@ export default class EOMethod extends AbstractEOProp<
         body: this.body,
         comments: this.comments,
         decorators: this.existingDecorators,
+        generator: this.value.generator ?? false,
+        async: this.value.async ?? false,
       }),
       this.replaceSuperWithUndefined
     );