stream: ended streams should resolve the async iteration

Fixes: #23891 PR-URL: #23901 Reviewed-By: Colin Ihrig <[email protected]> Reviewed-By: James M Snell <[email protected]>
nodejs · Nov 26, 2018 · 5ea25c2 · 5ea25c2
1 parent dbd37ab
commit 5ea25c2
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 1 deletion.
diff --git a/lib/internal/streams/async_iterator.js b/lib/internal/streams/async_iterator.js
@@ -127,7 +127,10 @@ const createReadableStreamAsyncIterator = (stream) => {
     [kLastResolve]: { value: null, writable: true },
     [kLastReject]: { value: null, writable: true },
     [kError]: { value: null, writable: true },
-    [kEnded]: { value: false, writable: true },
+    [kEnded]: {
+      value: stream._readableState.endEmitted,
+      writable: true
+    },
     [kLastPromise]: { value: null, writable: true },
     // the function passed to new Promise
     // is cached so we avoid allocating a new

diff --git a/test/parallel/test-stream-readable-async-iterators.js b/test/parallel/test-stream-readable-async-iterators.js
@@ -362,6 +362,24 @@ async function tests() {
       assert.strictEqual(e, err);
     }
   })();
+
+  await (async () => {
+    console.log('iterating on an ended stream completes');
+    const r = new Readable({
+      objectMode: true,
+      read() {
+        this.push('asdf');
+        this.push('hehe');
+        this.push(null);
+      }
+    });
+    // eslint-disable-next-line no-unused-vars
+    for await (const a of r) {
+    }
+    // eslint-disable-next-line no-unused-vars
+    for await (const b of r) {
+    }
+  })();
 }
 
 // to avoid missing some tests if a promise does not resolve