retain return types on ObjectMethods in transform-es2015-shorthand-properties (#4670)

packages/babel-plugin-transform-es2015-shorthand-properties/src/index.js

@@ -6,9 +6,12 @@ export default function () {
       ObjectMethod(path) {
         let { node } = path;
         if (node.kind === "method") {
+          const func = t.functionExpression(null, node.params, node.body, node.generator, node.async);
+          func.returnType = node.returnType;
+
           path.replaceWith(t.objectProperty(
             node.key,
-            t.functionExpression(null, node.params, node.body, node.generator, node.async),
+            func,
             node.computed
           ));
         }

packages/babel-plugin-transform-es2015-shorthand-properties/test/fixtures/shorthand-properties/method-type-annotations/actual.js

@@ -0,0 +1,6 @@
+// @flow
+var obj = {
+  method(a: string): number {
+    return 5 + 5;
+  }
+};

packages/babel-plugin-transform-es2015-shorthand-properties/test/fixtures/shorthand-properties/method-type-annotations/expected.js

@@ -0,0 +1,6 @@
+// @flow
+var obj = {
+  method: function (a: string): number {
+    return 5 + 5;
+  }
+};

packages/babel-plugin-transform-es2015-shorthand-properties/test/fixtures/shorthand-properties/method-type-annotations/options.json

@@ -0,0 +1,3 @@
+{
+  "plugins": ["transform-es2015-shorthand-properties", "syntax-flow"]
+}

packages/babel-plugin-transform-flow-comments/test/fixtures/flow-comments/object-method-type-annotations/actual.js

@@ -0,0 +1,6 @@
+// @flow
+var obj = {
+  method(a: string): number {
+    return 5 + 5;
+  }
+};

packages/babel-plugin-transform-flow-comments/test/fixtures/flow-comments/object-method-type-annotations/expected.js

@@ -0,0 +1,6 @@
+// @flow
+var obj = {
+  method: function (a /*: string*/) /*: number*/ {
+    return 5 + 5;
+  }
+};

packages/babel-plugin-transform-flow-comments/test/fixtures/flow-comments/object-method-type-annotations/options.json

@@ -0,0 +1,3 @@
+{
+  "plugins": ["transform-es2015-shorthand-properties", "transform-flow-comments"]
+}

packages/babel-plugin-transform-flow-strip-types/test/fixtures/regression/object-method-type-annotations/actual.js

@@ -0,0 +1,6 @@
+// @flow
+var obj = {
+  method(a: string): number {
+    return 5 + 5;
+  }
+};

packages/babel-plugin-transform-flow-strip-types/test/fixtures/regression/object-method-type-annotations/expected.js

@@ -0,0 +1,5 @@
+var obj = {
+  method: function (a) {
+    return 5 + 5;
+  }
+};

packages/babel-plugin-transform-flow-strip-types/test/fixtures/regression/object-method-type-annotations/options.json

@@ -0,0 +1,3 @@
+{
+  "plugins": ["transform-es2015-shorthand-properties", "transform-flow-strip-types"]
+}