node-postgres/test/unit/communication-tests.js

require(__dirname+'/test-helper');

test('client can take existing stream', function() {
  var stream = new MemoryStream();
  var client = new Client({
    stream: stream
  });
  assert.equal(client.stream, stream);
});

test('using closed stream', function() {
  var stream = new MemoryStream();
  stream.readyState = 'closed';
  stream.connect = function(port, host) {
    this.connectCalled = true;
    this.port = port;
    this.host = host;
  }
  var client = new Client({
    stream: stream,
    host: 'bang',
    port: 1234
  });
  client.connect();
  test('makes stream connect', function() {
    assert.equal(stream.connectCalled, true);
  });
  test('uses configured port', function() {
    assert.equal(stream.port, 1234);
  });
  test('uses configured host', function() {
    assert.equal(stream.host, 'bang');
  });
  
  test('after stream connects', function() {
    stream.emit('connect');
    //TODO - test more of the connection packets
  });
});

test('using opened stream', function() {
  var stream = new MemoryStream();
  stream.readyState = 'open';
  stream.connect = function() {
    assert.ok(false, "Should not call open");
  };
  var client = new Client({stream: stream});
  test('does not call open', function() {
    client.connect();
  });
});

test('query queue', function() {

  var stream = new MemoryStream();

  stream.readyState = 'open';

  var client = new Client({
    stream: stream
  });
  client.connect();

  test('new client has empty queue', function() {
    assert.empty(client.queryQueue);
  });

  test('calling query queues the query object', function() {
    var query = client.query('!');
    assert.length(client.queryQueue, 1);
  });

  test('sends query after stream emits ready for query packet', function() {
    assert.empty(stream.packets);
    var handled = stream.emit('data', buffers.readyForQuery());
    assert.ok(handled, "Stream should have had data handled");
    assert.length(stream.packets, 1);
    assert.equalBuffers(stream.packets[0], [0x51,0,0,0,6,33,0])
  });

});