333

schangxiang@126.com

2025-09-19 18966e02fb573c7e2bb0c6426ed792b38b910940

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
'use strict'
const co = require('co')
const expect = require('expect.js')
 
const describe = require('mocha').describe
const it = require('mocha').it
const BluebirdPromise = require('bluebird')
 
const Pool = require('../')
 
const checkType = promise => {
  expect(promise).to.be.a(BluebirdPromise)
  return promise.catch(e => undefined)
}
 
describe('Bring your own promise', function () {
  it('uses supplied promise for operations', co.wrap(function * () {
    const pool = new Pool({ Promise: BluebirdPromise })
    const client1 = yield checkType(pool.connect())
    client1.release()
    yield checkType(pool.query('SELECT NOW()'))
    const client2 = yield checkType(pool.connect())
    // TODO - make sure pg supports BYOP as well
    client2.release()
    yield checkType(pool.end())
  }))
 
  it('uses promises in errors', co.wrap(function * () {
    const pool = new Pool({ Promise: BluebirdPromise, port: 48484 })
    yield checkType(pool.connect())
    yield checkType(pool.end())
    yield checkType(pool.connect())
    yield checkType(pool.query())
    yield checkType(pool.end())
  }))
})