node-postgres/test/index.js

var expect = require('expect.js')
var Client = require('pg').Client
var co = require('co')
var Promise = require('bluebird')
var _ = require('lodash')

var Pool = require('../')

describe('pool', function() {

  describe('with callbacks', function() {
    it('works totally unconfigured', function(done) {
      const pool = new Pool()
      pool.connect(function(err, client, release) {
        if (err) return done(err)
        client.query('SELECT NOW()', function(err, res) {
          release()
          if (err) return done(err)
          expect(res.rows).to.have.length(1)
          pool.end(done)
        })
      })
    })

    it('passes props to clients', function(done) {
      const pool = new Pool({ binary: true })
      pool.connect(function(err, client, release) {
        release()
        expect(client.binary).to.eql(true)
        pool.end(done)
      })
    })

    it('removes client if it errors in background', function(done) {
      const pool = new Pool()
      pool.connect(function(err, client, release) {
        release()
        client.testString = 'foo'
        setTimeout(function() {
          client.emit('error', new Error('on purpose'))
        }, 10)
      })
      pool.on('error', function(err) {
        expect(err.message).to.be('on purpose')
        expect(err.client).to.not.be(undefined)
        expect(err.client.testString).to.be('foo')
        err.client.connection.stream.on('end', function() {
          pool.end(done)
        })
      })
    })
  })

  describe('with promises', function() {
    it('connects and disconnects', co.wrap(function*() {
      var pool = new Pool()
      var client = yield pool.connect()
      expect(pool.pool.availableObjectsCount()).to.be(0)
      var res = yield client.queryAsync('select $1::text as name', ['hi'])
      expect(res.rows).to.eql([{ name: 'hi' }])
      client.release()
      expect(pool.pool.getPoolSize()).to.be(1)
      expect(pool.pool.availableObjectsCount()).to.be(1)
      return yield pool.end()
    }))

    it('properly pools clients', co.wrap(function*() {
      var pool = new Pool({ poolSize: 9 })
      var count = 0
      while (count < 30) {
        count++
        pool.connect().then(function(client) {
          client.queryAsync('select $1::text as name', ['hi']).then(function(res) {
            client.release()
          })
        })
      }
      yield Promise.delay(100)
      expect(pool.pool.getPoolSize()).to.be(9)
      return yield pool.end()
    }))

    it('supports just running queries', co.wrap(function*() {
      var pool = new Pool({ poolSize: 9 })
      var count = 0
      var queries = _.times(30).map(function() {
        return pool.query('SELECT $1::text as name', ['hi'])
      })
      console.log('executing')
      yield queries
      expect(pool.pool.getPoolSize()).to.be(9)
      expect(pool.pool.availableObjectsCount()).to.be(9)
      return yield pool.end()
    }))

    it('recovers from all errors', co.wrap(function*() {
      var pool = new Pool({ poolSize: 9 })
      var count = 0

      while(count++ < 30) {
        try {
          yield pool.query('SELECT lksjdfd')
        } catch(e) {
        }
      }
      var res = yield pool.query('SELECT $1::text as name', ['hi'])
      expect(res.rows).to.eql([{ name: 'hi' }])
      return yield pool.end()
    }))
  })
})