node-postgres/test/integration/client/transaction-tests.js

"use strict";
var helper = require('./test-helper');
const suite = new helper.Suite()
const pg = helper.pg

const client = new pg.Client()
client.connect(assert.success(function () {

  client.query('begin');

  var getZed = {
    text: 'SELECT * FROM person WHERE name = $1',
    values: ['Zed']
  };

  suite.test('name should not exist in the database', function (done) {
    client.query(getZed, assert.calls(function (err, result) {
      assert(!err);
      assert.empty(result.rows);
      done()
    }))
  })

  suite.test('can insert name', (done) => {
    client.query("INSERT INTO person(name, age) VALUES($1, $2)", ['Zed', 270], assert.calls(function (err, result) {
      assert(!err)
      done()
    }));
  })

  suite.test('name should exist in the database', function (done) {
    client.query(getZed, assert.calls(function (err, result) {
      assert(!err);
      assert.equal(result.rows[0].name, 'Zed');
      done()
    }))
  })

  suite.test('rollback', (done) => {
    client.query('rollback', done);
  })

  suite.test('name should not exist in the database', function (done) {
    client.query(getZed, assert.calls(function (err, result) {
      assert(!err);
      assert.empty(result.rows);
      client.end(done)
    }))
  })
}))

suite.test('gh#36', function (cb) {
  const pool = new pg.Pool()
  pool.connect(assert.success(function (client, done) {
    client.query("BEGIN");
    client.query({
      name: 'X',
      text: "SELECT $1::INTEGER",
      values: [0]
    }, assert.calls(function (err, result) {
      if (err) throw err;
      assert.equal(result.rows.length, 1);
    }))
    client.query({
      name: 'X',
      text: "SELECT $1::INTEGER",
      values: [0]
    }, assert.calls(function (err, result) {
      if (err) throw err;
      assert.equal(result.rows.length, 1);
    }))
    client.query("COMMIT", function () {
      done();
      pool.end(cb)
    })
  }));
})