cerxes/babel
3
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge pull request #2861 from hzoo/i-2838

Browse Source 
helper-remap-async-to-generator: account for ObjectMethod - fixes #2838
This commit is contained in:
Sebastian McKenzie 2015-11-06 19:40:30 -05:00
commit 9a1cf40296
3 changed files with 17 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
packages/babel-core/test/fixtures/transformation/async-to-generator/object-method/actual.js vendored Normal file
View File

@ -0,0 +1,6 @@
let obj = {
a: 123,
async foo(bar) {
return await baz(bar);
}
}

8
packages/babel-core/test/fixtures/transformation/async-to-generator/object-method/expected.js vendored Normal file
View File

@ -0,0 +1,8 @@
let obj = {
a: 123,
foo(bar) {
return babelHelpers.asyncToGenerator(function* () {
return yield baz(bar);
})();
}
};

6
packages/babel-helper-remap-async-to-generator/src/index.js
View File

@ -24,7 +24,7 @@ let awaitVisitor = {
}
};
function classMethod(path: NodePath, callId: Object) {
function classOrObjectMethod(path: NodePath, callId: Object) {
let node = path.node;
let body = node.body;
@ -99,8 +99,8 @@ export default function (path: NodePath, callId: Object) {
path.traverse(awaitVisitor);
if (path.isClassMethod()) {
return classMethod(path, callId);
if (path.isClassMethod() || path.isObjectMethod()) {
return classOrObjectMethod(path, callId);
} else {
return plainFunction(path, callId);
}