repl: support top-level for-await-of

PR-URL: #23841 Fixes: #23836 Reviewed-By: John-David Dalton <[email protected]> Reviewed-By: Yuta Hiroto <[email protected]> Reviewed-By: Gus Caplan <[email protected]> Reviewed-By: Rich Trott <[email protected]> Reviewed-By: Tiancheng "Timothy" Gu <[email protected]> Reviewed-By: Colin Ihrig <[email protected]> Reviewed-By: Matheus Marchini <[email protected]> Reviewed-By: James M Snell <[email protected]>
nodejs · Oct 28, 2018 · d6aeb1e · d6aeb1e
1 parent 927c6f8
commit d6aeb1e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/lib/internal/repl/await.js b/lib/internal/repl/await.js
@@ -11,6 +11,12 @@ const visitorsWithoutAncestors = {
     }
     walk.base.ClassDeclaration(node, state, c);
   },
+  ForOfStatement(node, state, c) {
+    if (node.await === true) {
+      state.containsAwait = true;
+    }
+    walk.base.ForOfStatement(node, state, c);
+  },
   FunctionDeclaration(node, state, c) {
     state.prepend(node, `${node.id.name}=`);
   },

diff --git a/test/parallel/test-repl-top-level-await.js b/test/parallel/test-repl-top-level-await.js
@@ -130,7 +130,8 @@ async function ordinaryTests() {
     [ 'let o = await 1, p', 'undefined' ],
     [ 'p', 'undefined' ],
     [ 'let q = 1, s = await 2', 'undefined' ],
-    [ 's', '2' ]
+    [ 's', '2' ],
+    [ 'for await (let i of [1,2,3]) console.log(i)', 'undefined', { line: 3 } ]
   ];
 
   for (const [input, expected, options = {}] of testCases) {