Workaround reference error when using async arrow functions in code (#…

…486)

src/babel/index.js

@@ -191,6 +191,16 @@ module.exports = function plugin(args) {
             // class property node value is nullable
             if (node.value && node.value.type === 'ArrowFunctionExpression') {
               const isAsync = node.value.async;
+
+              // TODO:
+              // Remove this check when babel issue is resolved: https://github.com/babel/babel/issues/5078
+              // RHL Issue: https://github.com/gaearon/react-hot-loader/issues/391
+              // This code makes async arrow functions not reloadable,
+              // but doesn't break code any more when using 'this' inside AAF
+              if (isAsync) {
+                return;
+              }
+
               const params = node.value.params;
               const newIdentifier = t.identifier(`__${node.key.name}__REACT_HOT_LOADER__`);
 

test/babel/fixtures/class-properties/async-functions-expression-body/expected.js

@@ -1,10 +1,6 @@
 class Foo {
   constructor() {
-    this.bar = async (...params) => await this.__bar__REACT_HOT_LOADER__(...params);
-  }
-
-  async __bar__REACT_HOT_LOADER__(a, b) {
-    return await b(a);
+    this.bar = async (a, b) => await b(a);
   }
 
 }

test/babel/fixtures/class-properties/async-functions/expected.js

@@ -1,10 +1,8 @@
 class Foo {
   constructor() {
-    this.bar = async (...params) => await this.__bar__REACT_HOT_LOADER__(...params);
-  }
-
-  async __bar__REACT_HOT_LOADER__(a, b) {
-    return await a(b);
+    this.bar = async (a, b) => {
+      return await a(b);
+    };
   }
 
 }

test/babel/fixtures/class-properties/default-params/expected.js

@@ -4,7 +4,7 @@ class Foo {
   }
 
   __bar__REACT_HOT_LOADER__(a = "foo") {
-    return `${ a }bar`;
+    return `${a}bar`;
   }
 
 }

test/babel/fixtures/class-properties/destructured-params/expected.js

@@ -4,7 +4,7 @@ class Foo {
   }
 
   __bar__REACT_HOT_LOADER__({ a }, { b }) {
-    return `${ a }${ b }`;
+    return `${a}${b}`;
   }
 
 }