node-postgres/test/unit/client/simple-query-tests.js

var helper = require(__dirname + "/test-helper");

test('executing query', function() {

  test("queing query", function() {

    test('when connection is ready', function() {
      var client = helper.client();
      assert.empty(client.connection.queries);
      client.connection.emit('readyForQuery');
      client.query('yes');
      assert.length(client.connection.queries, 1);
      assert.equal(client.connection.queries, 'yes');
    });

    test('when connection is not ready', function() {
      var client = helper.client();

      test('query is not sent', function() {
        client.query('boom');
        assert.empty(client.connection.queries);
      });

      test('sends query to connection once ready', function() {
        assert.ok(client.connection.emit('readyForQuery'));
        assert.length(client.connection.queries, 1);
        assert.equal(client.connection.queries[0], "boom");
      });

    });

    test("multiple in the queue", function() {
      var client = helper.client();
      var connection = client.connection;
      var queries = connection.queries;
      client.query('one');
      client.query('two');
      client.query('three');
      assert.empty(queries);

      test("after one ready for query",function() {
        connection.emit('readyForQuery');
        assert.length(queries, 1);
        assert.equal(queries[0], "one");
      });

      test('after two ready for query', function() {
        connection.emit('readyForQuery');
        assert.length(queries, 2);
      });

      test("after a bunch more", function() {
        connection.emit('readyForQuery');
        connection.emit('readyForQuery');
        connection.emit('readyForQuery');
        assert.length(queries, 3);
        assert.equal(queries[0], "one");
        assert.equal(queries[1], 'two');
        assert.equal(queries[2], 'three');
      });
    });
  });

  test("query event binding and flow", function() {
    var client = helper.client();
    var con = client.connection;
    var query = client.query('whatever');

    test("has no queries sent before ready", function() {
      assert.empty(con.queries);
    });

    test('sends query on readyForQuery event', function() {
      con.emit('readyForQuery');
      assert.length(con.queries, 1);
      assert.equal(con.queries[0], 'whatever');
    });

    test('handles rowDescription message', function() {
      var handled = con.emit('rowDescription',{});
      assert.ok(handled, "should have handlded rowDescritpion");
    });

    test('handles dataRow messages', function() {
      assert.raises(query, 'row', function(row) {
        assert.equal(row.fields[0], "hi");
      });
      var handled = con.emit('dataRow', { fields: ["hi"] });
      assert.ok(handled, "should have handled first data row message");

      assert.raises(query, 'row', function(row) {
        assert.equal(row.fields[0], "bye");
      });
      var handledAgain = con.emit('dataRow', { fields: ["bye"] });
      assert.ok(handledAgain, "should have handled seciond data row message");

    });

    test('removes itself after another readyForQuery message', function() {
      assert.raises(query, "end");
      con.emit("readyForQuery");
      //this would never actually happen
      ['dataRow','rowDescritpion', 'commandComplete'].forEach(function(msg) {
        assert.equal(con.emit(msg), false, "Should no longer be picking up '"+ msg +"' messages");
      });
    });

  });

});