[CONJS-303] DMLs are not returning any output when streaming #293

Author: rusher

lib/cmd/parser.js

@@ -192,6 +192,10 @@ class Parser extends Command {
         }
       }
     }
+
+    if (this.inStream) {
+      this.handleNewRows(okPacket);
+    }
     if (!mustRedirect) {
       if (
         info.redirectRequest &&

test/integration/test-resultset-streaming.js

@@ -33,6 +33,16 @@ describe('results-set streaming', () => {
     assert.equal(10000, currRow);
   });
 
+  it('Streaming Update for-await-of', async function () {
+    let currRow = 0;
+    const stream = shareConn.queryStream('DO 1');
+    for await (const row of stream) {
+      console.log(row);
+      currRow++;
+    }
+    assert.equal(1, currRow);
+  });
+
   it('Streaming Error', function (done) {
     const stream = shareConn.queryStream('wrong');
     stream.on('error', (error) => {