88 lines
2.3 KiB
JavaScript
88 lines
2.3 KiB
JavaScript
var expect = require('expect.js')
|
|
var EventEmitter = require('events').EventEmitter
|
|
var describe = require('mocha').describe
|
|
var it = require('mocha').it
|
|
var objectAssign = require('object-assign')
|
|
var Pool = require('../')
|
|
|
|
describe('events', function () {
|
|
it('emits connect before callback', function (done) {
|
|
var pool = new Pool()
|
|
var emittedClient = false
|
|
pool.on('connect', function (client) {
|
|
emittedClient = client
|
|
})
|
|
|
|
pool.connect(function (err, client, release) {
|
|
if (err) return done(err)
|
|
release()
|
|
pool.end()
|
|
expect(client).to.be(emittedClient)
|
|
done()
|
|
})
|
|
})
|
|
|
|
it('emits "connect" only with a successful connection', function (done) {
|
|
var pool = new Pool({
|
|
// This client will always fail to connect
|
|
Client: mockClient({
|
|
connect: function (cb) {
|
|
process.nextTick(function () { cb(new Error('bad news')) })
|
|
}
|
|
})
|
|
})
|
|
pool.on('connect', function () {
|
|
throw new Error('should never get here')
|
|
})
|
|
pool._create(function (err) {
|
|
if (err) done()
|
|
else done(new Error('expected failure'))
|
|
})
|
|
})
|
|
|
|
it('emits acquire every time a client is acquired', function (done) {
|
|
var pool = new Pool()
|
|
var acquireCount = 0
|
|
pool.on('acquire', function (client) {
|
|
expect(client).to.be.ok()
|
|
acquireCount++
|
|
})
|
|
for (var i = 0; i < 10; i++) {
|
|
pool.connect(function (err, client, release) {
|
|
err ? done(err) : release()
|
|
release()
|
|
if (err) return done(err)
|
|
})
|
|
pool.query('SELECT now()')
|
|
}
|
|
setTimeout(function () {
|
|
expect(acquireCount).to.be(20)
|
|
pool.end(done)
|
|
}, 40)
|
|
})
|
|
|
|
it('emits error and client if an idle client in the pool hits an error', function (done) {
|
|
var pool = new Pool()
|
|
pool.connect(function (err, client) {
|
|
expect(err).to.equal(null)
|
|
client.release()
|
|
setImmediate(function () {
|
|
client.emit('error', new Error('problem'))
|
|
})
|
|
pool.once('error', function (err, errClient) {
|
|
expect(err.message).to.equal('problem')
|
|
expect(errClient).to.equal(client)
|
|
done()
|
|
})
|
|
})
|
|
})
|
|
})
|
|
|
|
function mockClient (methods) {
|
|
return function () {
|
|
var client = new EventEmitter()
|
|
objectAssign(client, methods)
|
|
return client
|
|
}
|
|
}
|