node-postgres/packages/pg-query-stream/test/passing-options.js

var assert = require('assert')
var helper = require('./helper')
var QueryStream = require('../')

helper('passing options', function (client) {
  it('passes row mode array', function (done) {
    var stream = new QueryStream('SELECT * FROM generate_series(0, 10) num', [], { rowMode: 'array' })
    var query = client.query(stream)
    var result = []
    query.on('data', (datum) => {
      result.push(datum)
    })
    query.on('end', () => {
      const expected = new Array(11).fill(0).map((_, i) => [i])
      assert.deepEqual(result, expected)
      done()
    })
  })

  it('passes custom types', function (done) {
    const types = {
      getTypeParser: () => (string) => string,
    }
    var stream = new QueryStream('SELECT * FROM generate_series(0, 10) num', [], { types })
    var query = client.query(stream)
    var result = []
    query.on('data', (datum) => {
      result.push(datum)
    })
    query.on('end', () => {
      const expected = new Array(11).fill(0).map((_, i) => ({
        num: i.toString(),
      }))
      assert.deepEqual(result, expected)
      done()
    })
  })
})