node-postgres/test/fast-reader.js
Brian M. Carlson 722296f8d2 Initial commit
2013-10-21 23:57:50 -05:00

30 lines
780 B
JavaScript

var pg = require('pg')
var assert = require('assert')
var gonna = require('gonna')
var _ = require('lodash')
var QueryStream = require('../')
var client = new pg.Client()
var connected = gonna('connect', 100, function() {
var stream = new QueryStream('SELECT * FROM generate_series(0, 200) num', [])
var query = client.query(stream)
var result = []
stream.on('readable', function() {
var res = stream.read()
assert(res, 'should not return null on evented reader')
result.push(res.num)
})
stream.on('end', client.end.bind(client))
stream.on('end', function() {
var total = result.reduce(function(prev, cur) {
return prev + cur
})
assert.equal(total, 20100)
})
assert.strictEqual(query.read(2), null)
})
client.connect(connected)