query-error-handling-tests.js

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

test('error during query execution', function () {
  var client = new Client(helper.args)
  client.connect(
    assert.success(function () {
      var queryText = 'select pg_sleep(10)'
      var sleepQuery = new Query(queryText)
      var pidColName = 'procpid'
      var queryColName = 'current_query'
      helper.versionGTE(
        client,
        90200,
        assert.success(function (isGreater) {
          if (isGreater) {
            pidColName = 'pid'
            queryColName = 'query'
          }
          var query1 = client.query(
            sleepQuery,
            assert.calls(function (err, result) {
              assert(err)
              client.end()
            })
          )
          //ensure query1 does not emit an 'end' event
          //because it was killed and received an error
          //https://github.com/brianc/node-postgres/issues/547
          query1.on('end', function () {
            assert.fail('Query with an error should not emit "end" event')
          })
          setTimeout(function () {
            var client2 = new Client(helper.args)
            client2.connect(
              assert.success(function () {
                var killIdleQuery = `SELECT ${pidColName}, (SELECT pg_cancel_backend(${pidColName})) AS killed FROM pg_stat_activity WHERE ${queryColName} LIKE $1`
                client2.query(
                  killIdleQuery,
                  [queryText],
                  assert.calls(function (err, res) {
                    assert.ifError(err)
                    assert(res.rows.length > 0)
                    client2.end()
                    assert.emits(client2, 'end')
                  })
                )
              })
            )
          }, 300)
        })
      )
    })
  )
})

if (helper.config.native) {
  return
}

test('9.3 column error fields', function () {
  var client = new Client(helper.args)
  client.connect(
    assert.success(function () {
      helper.versionGTE(
        client,
        90300,
        assert.success(function (isGreater) {
          if (!isGreater) {
            return client.end()
          }

          client.query('CREATE TEMP TABLE column_err_test(a int NOT NULL)')
          client.query('INSERT INTO column_err_test(a) VALUES (NULL)', function (err) {
            if (!helper.config.native) {
              assert(err instanceof DatabaseError)
            }
            assert.equal(err.severity, 'ERROR')
            assert.equal(err.code, '23502')
            assert.equal(err.table, 'column_err_test')
            assert.equal(err.column, 'a')
            return client.end()
          })
        })
      )
    })
  )
})

test('9.3 constraint error fields', function () {
  var client = new Client(helper.args)
  client.connect(
    assert.success(function () {
      helper.versionGTE(
        client,
        90300,
        assert.success(function (isGreater) {
          if (!isGreater) {
            console.log('skip 9.3 error field on older versions of postgres')
            return client.end()
          }

          client.query('CREATE TEMP TABLE constraint_err_test(a int PRIMARY KEY)')
          client.query('INSERT INTO constraint_err_test(a) VALUES (1)')
          client.query('INSERT INTO constraint_err_test(a) VALUES (1)', function (err) {
            if (!helper.config.native) {
              assert(err instanceof DatabaseError)
            }
            assert.equal(err.severity, 'ERROR')
            assert.equal(err.code, '23505')
            assert.equal(err.table, 'constraint_err_test')
            assert.equal(err.constraint, 'constraint_err_test_pkey')
            return client.end()
          })
        })
      )
    })
  )
})