node-postgres/test/concat.js
Calvin Metcalf 41b7d7d4de fix up tests
2014-05-14 09:51:56 -04:00

23 lines
636 B
JavaScript

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)
})
})