Add tests

Author: brianc

packages/pg-query-stream/src/index.ts

@@ -27,11 +27,13 @@ class QueryStream extends Readable implements Submittable {
 
     super({ objectMode: true, autoDestroy: true, highWaterMark: batchSize || highWaterMark })
     this.cursor = new Cursor(text, values, config)
-    this.cursor.on('end', (result) => {
-      this.callback && this.callback(null, result)
-    }).on('error', (err) => {
-      this.callback && this.callback(err)
-    })
+    this.cursor
+      .on('end', (result) => {
+        this.callback && this.callback(null, result)
+      })
+      .on('error', (err) => {
+        this.callback && this.callback(err)
+      })
 
     // delegate Submittable callbacks to cursor
     this.handleRowDescription = this.cursor.handleRowDescription.bind(this.cursor)

packages/pg-query-stream/test/pool.ts

@@ -0,0 +1,19 @@
+import assert from 'assert'
+import { Pool } from 'pg'
+import QueryStream from '../src'
+
+describe('pool', function () {
+  it('works', async function () {
+    const pool = new Pool()
+    const query = new QueryStream('SELECT * FROM generate_series(0, 10) num', [])
+    const q = pool.query(query)
+    console.log(q) // Promise { <pending>, ...
+    query.on('data', (row) => {
+      // just consume the whole stream
+    })
+    await q //!
+    query.on('end', () => {
+      pool.end()
+    })
+  })
+})