Fixed comments following simple expressions

estools · Nov 7, 2012 · 00fab71 · 00fab71
1 parent 216e383
commit 00fab71
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/escodegen.js b/escodegen.js
@@ -701,6 +701,7 @@
         if (stmt.trailingComments) {
             tailingToStatement = !endsWithLineTerminator(toSourceNode(result).toString());
             specialBase = stringRepeat(' ', calculateSpaces(toSourceNode([base, result, indent]).toString()));
+            result = isArray(result) ? result : [result];
             for (i = 0, len = stmt.trailingComments.length; i < len; i += 1) {
                 comment = stmt.trailingComments[i];
                 if (tailingToStatement) {

diff --git a/test/comment/simple-statement-comment.expected.js b/test/comment/simple-statement-comment.expected.js
@@ -0,0 +1 @@
+;    // Trailing
diff --git a/test/comment/simple-statement-comment.js b/test/comment/simple-statement-comment.js
@@ -0,0 +1 @@
+;    // Trailing