2020-04-10 10:34:34 -05:00

24 lines
661 B
JavaScript

var concat = require('concat-stream');
var tester = require('stream-tester');
var JSONStream = require('JSONStream');
var QueryStream = require('../');
require('./helper')('pauses', function (client) {
it('pauses', function (done) {
this.timeout(5000);
var stream = new QueryStream('SELECT * FROM generate_series(0, $1) num', [200], { batchSize: 2, highWaterMark: 2 });
var query = client.query(stream);
var pauser = tester.createPauseStream(0.1, 100);
query
.pipe(JSONStream.stringify())
.pipe(pauser)
.pipe(
concat(function (json) {
JSON.parse(json);
done();
})
);
});
});