Merge pull request #57 from fossamagna/fix-in-sequcen-expression

Add global assignment in SequenceExpression to target for function generation.
fossamagna · Jan 21, 2019 · 8176603 · 8176603
2 parents da6a490 + 7d4ebf1
commit 8176603
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 1 deletion.
diff --git a/index.js b/index.js
@@ -39,12 +39,21 @@ function createStubFunctionASTNode(functionName, leadingComments, params) {
 function _generateStubs(data, options) {
   var ast = esprima.parseScript(data, { attachComment: options.comment });
   var stubs = [];
+  var functionName;
   estraverse.traverse(ast, {
     leave: function (node) {
       if (node.type === 'ExpressionStatement'
         && isGlobalAssignmentExpression(node.expression)) {
-        var functionName = node.expression.left.property.name;
+        functionName = node.expression.left.property.name;
         stubs.push(createStubFunctionASTNode(functionName, node.leadingComments, node.expression.right.params));
+      } else if (node.type === 'ExpressionStatement' 
+        && node.expression.type === 'SequenceExpression') {
+        node.expression.expressions.forEach(function (expression) {
+          if (isGlobalAssignmentExpression(expression)) {
+            functionName = expression.left.property.name;
+            stubs.push(createStubFunctionASTNode(functionName, node.leadingComments, expression.right.params));
+          }
+        });
       }
     }
   });

diff --git a/test/fixtures/expected.js b/test/fixtures/expected.js
@@ -9,4 +9,8 @@ function foo() {
  */
 // leading line comment for boo
 function boo(message) {
+}
+function test() {
+}
+function X() {
 }
diff --git a/test/fixtures/source.js b/test/fixtures/source.js
@@ -10,3 +10,7 @@ global.foo = function () {
 // leading line comment for boo
 global.boo = function (message) {
 };
+function test() {
+};
+var X = 'x';
+global.test = test, global.X = X;