Add tests to query stream

sidorares · Jan 22, 2024 · 97f1507 · 97f1507
1 parent b263402
commit 97f1507
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/test/integration/connection/test-stream.js b/test/integration/connection/test-stream.js
@@ -7,6 +7,7 @@ const assert = require('assert');
 let rows;
 const rows1 = [];
 const rows2 = [];
+const rows3 = [];
 
 connection.query(
   [
@@ -62,8 +63,14 @@ connection.execute('SELECT * FROM announcements', (err, _rows) => {
   });
 });
 
+const s3 = connection.query('SELECT * FROM announcements').stream();
+for await (const row of s3) {
+  rows3.push(row);
+}
+
 process.on('exit', () => {
   assert.deepEqual(rows.length, 2);
   assert.deepEqual(rows, rows1);
   assert.deepEqual(rows, rows2);
+  assert.deepEqual(rows, rows3);
 });