query-error-handling-prepared-statement-tests.js

'use strict'
var helper = require('./test-helper')
var Query = helper.pg.Query
var util = require('util')

var suite = new helper.Suite()

suite.test('client end during query execution of prepared statement', function (done) {
  var client = new Client()
  client.connect(assert.success(function () {
    var sleepQuery = 'select pg_sleep($1)'

    var queryConfig = {
      name: 'sleep query',
      text: sleepQuery,
      values: [5]
    }

    var queryInstance = new Query(queryConfig, assert.calls(function (err, result) {
      assert.equal(err.message, 'Connection terminated')
      done()
    }))

    var query1 = client.query(queryInstance)

    query1.on('error', function (err) {
      assert.fail('Prepared statement should not emit error')
    })

    query1.on('row', function (row) {
      assert.fail('Prepared statement should not emit row')
    })

    query1.on('end', function (err) {
      assert.fail('Prepared statement when executed should not return before being killed')
    })

    client.end()
  }))
})

function killIdleQuery (targetQuery, cb) {
  var client2 = new Client(helper.args)
  var pidColName = 'procpid'
  var queryColName = 'current_query'
  client2.connect(assert.success(function () {
    helper.versionGTE(client2, 90200, assert.success(function (isGreater) {
      if (isGreater) {
        pidColName = 'pid'
        queryColName = 'query'
      }
      var killIdleQuery = 'SELECT ' + pidColName + ', (SELECT pg_terminate_backend(' + pidColName + ')) AS killed FROM pg_stat_activity WHERE ' + queryColName + ' = $1'
      client2.query(killIdleQuery, [targetQuery], assert.calls(function (err, res) {
        assert.ifError(err)
        assert.equal(res.rows.length, 1)
        client2.end(cb)
        assert.emits(client2, 'end')
      }))
    }))
  }))
}

suite.test('query killed during query execution of prepared statement', function (done) {
  if (helper.args.native) {
    return done()
  }
  var client = new Client(helper.args)
  client.connect(assert.success(function () {
    var sleepQuery = 'select pg_sleep($1)'

    const queryConfig = {
      name: 'sleep query',
      text: sleepQuery,
      values: [5]
    }

    // client should emit an error because it is unexpectedly disconnected
    assert.emits(client, 'error')

    var query1 = client.query(new Query(queryConfig), assert.calls(function (err, result) {
      assert.equal(err.message, 'terminating connection due to administrator command')
    }))

    query1.on('error', function (err) {
      assert.fail('Prepared statement should not emit error')
    })

    query1.on('row', function (row) {
      assert.fail('Prepared statement should not emit row')
    })

    query1.on('end', function (err) {
      assert.fail('Prepared statement when executed should not return before being killed')
    })

    killIdleQuery(sleepQuery, done)
  }))
})