concat.js

var assert = require('assert')
var concat = require('concat-stream')
var through = require('through')
var helper = require('./helper')

var QueryStream = require('../')

helper('concat', function (client) {
  it('concats correctly', function (done) {
    var stream = new QueryStream('SELECT * FROM generate_series(0, 200) num', [])
    var query = client.query(stream)
    query.pipe(through(function (row) {
      this.push(row.num)
    })).pipe(concat(function (result) {
      var total = result.reduce(function (prev, cur) {
        return prev + cur
      })
      assert.equal(total, 20100)
    }))
    stream.on('end', done)
  })
})