(copy-from) add test with await pipeline()

Author: jeromew

test/copy-from.js

@@ -5,6 +5,8 @@ const assert = require('assert')
 const _ = require('lodash')
 const pg = require('pg')
 const { finished, pipeline, PassThrough } = require('stream')
+const { promisify } = require('util')
+const { spawn } = require('child_process')
 
 const copy = require('../').from
 
@@ -201,6 +203,23 @@ describe('copy-from', () => {
       })
     })
 
+    it('works with await pipeline()', async () => {
+      if (!pipeline) return
+      const client = new pg.Client()
+      await client.connect()
+      await client.query('CREATE TEMP TABLE numbers (num1 int)')
+      try {
+        const total = 1000
+        const seq = spawn('seq', [1, total]).stdout
+        const copyFromStream = client.query(copy(`COPY numbers FROM STDIN`))
+        await promisify(pipeline)(seq, copyFromStream)
+        const res = await client.query('SELECT count(*) FROM numbers')
+        assert.equal(res.rows[0].count, total)
+      } finally {
+        await client.end()
+      }
+    })
+
     describe('erroneous stream (syntax error)', () => {
       it("emits 0 'finish'", (done) => {
         assertCopyFromResult('syntaxError', '(field1 int)', [Buffer.from('1\n')], (err, rows, stream) => {