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