node-postgres/test/unit/copystream/copyfrom-tests.js

var helper = require(__dirname + '/../test-helper');
var CopyFromStream =  require(__dirname + '/../../../lib/copystream').CopyFromStream;
var ConnectionImitation = function () {
  this.send = 0;
  this.hasToBeSend = 0;
  this.finished = 0;
};
ConnectionImitation.prototype = {
  endCopyFrom: function () {
    assert.ok(this.finished++ === 0, "end shoud be called only once");
    assert.equal(this.send, this.hasToBeSend, "at the moment of the end all data has to be sent");
  },
  sendCopyFromChunk: function (chunk) {
    this.send += chunk.length; 
    return true;
  },
  updateHasToBeSend: function (chunk) {
    this.hasToBeSend += chunk.length;
    return chunk;
  }
};
var buf1 = new Buffer("asdfasd"),
  buf2 = new Buffer("q03r90arf0aospd;"),
  buf3 = new Buffer(542),
  buf4 = new Buffer("93jfemialfjkasjlfas");

test('CopyFromStream, start streaming before data, end after data. no drain event', function () {
  var stream = new CopyFromStream();
  var conn = new ConnectionImitation();
  stream.on('drain', function () {
    assert.ok(false, "there has not be drain event");
  });
  stream.startStreamingToConnection(conn);
  assert.ok(stream.write(conn.updateHasToBeSend(buf1)));
  assert.ok(stream.write(conn.updateHasToBeSend(buf2)));
  assert.ok(stream.write(conn.updateHasToBeSend(buf3)));
  assert.ok(stream.writable, "stream has to be writable");
  stream.end(conn.updateHasToBeSend(buf4));
  assert.ok(!stream.writable, "stream has not to be writable");
  stream.end();
  assert.equal(conn.hasToBeSend, conn.send);
});
test('CopyFromStream, start streaming after end, end after data. drain event', function () {
  var stream = new CopyFromStream();
  assert.emits(stream, 'drain', function() {}, 'drain have to be emitted');
  var conn = new ConnectionImitation() 
  assert.ok(!stream.write(conn.updateHasToBeSend(buf1)));
  assert.ok(!stream.write(conn.updateHasToBeSend(buf2)));
  assert.ok(!stream.write(conn.updateHasToBeSend(buf3)));
  assert.ok(stream.writable, "stream has to be writable");
  stream.end(conn.updateHasToBeSend(buf4));
  assert.ok(!stream.writable, "stream has not to be writable");
  stream.end();
  stream.startStreamingToConnection(conn);
  assert.equal(conn.hasToBeSend, conn.send);
});
test('CopyFromStream, start streaming between data chunks. end after data. drain event', function () {
  var stream = new CopyFromStream();
  var conn = new ConnectionImitation() 
  assert.emits(stream, 'drain', function() {}, 'drain have to be emitted');
  stream.write(conn.updateHasToBeSend(buf1));
  stream.write(conn.updateHasToBeSend(buf2));
  stream.startStreamingToConnection(conn);
  stream.write(conn.updateHasToBeSend(buf3));
  assert.ok(stream.writable, "stream has to be writable");
  stream.end(conn.updateHasToBeSend(buf4));
  assert.equal(conn.hasToBeSend, conn.send);
  assert.ok(!stream.writable, "stream has not to be writable");
  stream.end();
});
test('CopyFromStream, start sreaming before end. end stream with data. drain event', function () {
  var stream = new CopyFromStream();
  var conn = new ConnectionImitation() 
  assert.emits(stream, 'drain', function() {}, 'drain have to be emitted');
  stream.write(conn.updateHasToBeSend(buf1));
  stream.write(conn.updateHasToBeSend(buf2));
  stream.write(conn.updateHasToBeSend(buf3));
  stream.startStreamingToConnection(conn);
  assert.ok(stream.writable, "stream has to be writable");
  stream.end(conn.updateHasToBeSend(buf4));
  assert.equal(conn.hasToBeSend, conn.send);
  assert.ok(!stream.writable, "stream has not to be writable");
  stream.end();
});
test('CopyFromStream, start streaming after end. end with data. drain event', function(){
  var stream = new CopyFromStream();
  var conn = new ConnectionImitation() 
  assert.emits(stream, 'drain', function() {}, 'drain have to be emitted');
  stream.write(conn.updateHasToBeSend(buf1));
  stream.write(conn.updateHasToBeSend(buf2));
  stream.write(conn.updateHasToBeSend(buf3));
  stream.startStreamingToConnection(conn);
  assert.ok(stream.writable, "stream has to be writable");
  stream.end(conn.updateHasToBeSend(buf4));
  stream.startStreamingToConnection(conn);
  assert.equal(conn.hasToBeSend, conn.send);
  assert.ok(!stream.writable, "stream has not to be writable");
  stream.end();
});