node-postgres/test/parser-tests.js

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


var authOkBuffer = new BufferList()
  .addInt32(8)
  .join(true, 'R');

var paramStatusBuffer = new BufferList()
  .addCString("client_encoding")
  .addCString("UTF8")
  .join(true, 'S');

var backendKeyDataBuffer = new BufferList()
  .addInt32(1)
  .addInt32(2)
  .join(true,'K');

var readyForQueryBuffer = new BufferList()
  .add(Buffer('I'))
  .join(true,'Z');

var expectedAuthenticationOkayMessage = {
  name: 'AuthenticationOk',
  id: 'R',
  length: 8
};

var expectedParameterStatusMessage = {
  name: 'ParameterStatus',
  id: 'S',
  length: 25,
  parameterName: 'client_encoding',
  parameterValue: 'UTF8'
};

var expectedBackendKeyDataMessage = {
  name: 'BackendKeyData',
  id: 'K',
  processID: 1,
  secretKey: 2
};

var expectedReadyForQueryMessage = {
  name: 'ReadyForQuery',
  id: 'Z',
  length: 5,
  status: 'I'
};


test('Parser on single messages', function() {
  test('parses AuthenticationOk message', function() {
    var result = new Parser(authOkBuffer).parse()[0];
    assert.same(result, expectedAuthenticationOkayMessage);
  });

  test('parses ParameterStatus message', function() {
    var result = new Parser(paramStatusBuffer).parse()[0];
    assert.same(result, expectedParameterStatusMessage);
  });

  test('parses BackendKeyData message', function() {
    var result = new Parser(backendKeyDataBuffer).parse()[0];
    assert.same(result, expectedBackendKeyDataMessage);
  });

  test('parses ReadyForQuery message', function() {
    var result = new Parser(readyForQueryBuffer).parse()[0];
    assert.same(result, expectedReadyForQueryMessage);
  });


  var commandCompleteBuffer = new BufferList()
    .addCString("SELECT 3")
    .join(true,'C');
  test('parses CommandComplete message', function() {
    var result = new Parser(commandCompleteBuffer).parse()[0];
    assert.same(result, {
      length: 13,
      id: 'C',
      text: "SELECT 3"
    });
  });

  var emptyRowDescriptionBuffer = new BufferList()
    .addInt16(0) //number of fields
    .join(true,'T');
  test('parses RowDescriptions', function() {

    test('parses empty row description', function() {
      var result = new Parser(emptyRowDescriptionBuffer).parse()[0];
      assert.same(result, {
        name: 'RowDescription',
        id: 'T',
        length: 6,
        fieldCount: 0
      });
      assert.equal(result.fields.length, 0);
    });

    var addRow = function(bufferList, name, offset) {
      return bufferList.addCString(name) //field name
        .addInt32(offset++) //table id
        .addInt16(offset++) //attribute of column number
        .addInt32(offset++) //objectId of field's data type
        .addInt16(offset++) //datatype size
        .addInt32(offset++) //type modifier
        .addInt16(0) //format code, 0 => text
    };

    var oneRowDescBuff = new BufferList()
      .addInt16(1);
    oneRowDescBuff = addRow(oneRowDescBuff, 'id', 1)
      .join(true,'T');

    test('parses single row description',function() {
      var result = new Parser(oneRowDescBuff).parse()[0];
      assert.same(result, {
        name: 'RowDescription',
        id: 'T',
        length: 27,
        fieldCount: 1
      });
      assert.equal(result.fields.length, 1);

      assert.same(result.fields[0], {
        name: 'id',
        tableID: 1,
        columnID: 2,
        dataType: 3,
        dataTypeSize: 4,
        dataTypeModifier: 5,
        format: 'text'
      });
    });

    test('parses two row descriptions', function() {
      var twoRowDesc = new BufferList()
        .addInt16(2);
      twoRowDesc = addRow(twoRowDesc, 'bang', 1);
      twoRowDesc = addRow(twoRowDesc, 'whoah', 10);
      twoRowBuf = twoRowDesc.join(true, 'T');

      var result = new Parser(twoRowBuf).parse()[0];
      assert.same(result, {
        name: 'RowDescription',
        id: 'T',
        length: 53,
        fieldCount: 2
      });
      assert.equal(result.fields.length, 2);

      assert.same(result.fields[0], {
        name: 'bang',
        tableID: 1,
        columnID: 2,
        dataType: 3,
        dataTypeSize: 4,
        dataTypeModifier: 5,
        format: 'text'
      });

      assert.same(result.fields[1], {
        name: 'whoah',
        tableID: 10,
        columnID: 11,
        dataType: 12,
        dataTypeSize: 13,
        dataTypeModifier: 14,
        format: 'text'
      });


    });


  });



  test('parses empty CString', function() {
    var result = new Parser(Buffer([0])).parseCString();
    assert.equal(result, '');
  });

  test('parses length', function() {
    var parser = new Parser(Buffer([0,0,0,3]));
    var result = parser.parseLength();
    assert.equal(result, 3);
    assert.equal(parser.offset, 4);
  });

  test('parsing empty buffer returns false', function() {
    var parser = new Parser(Buffer(0));
    assert.equal(parser.parse(), false);
  });
});