| 6 | const text = 'SELECT generate_series as num FROM generate_series(0, 50)' |
| 7 | |
| 8 | function poolQueryPromise(pool, readRowCount) { |
| 9 | return new Promise((resolve, reject) => { |
| 10 | pool.connect((err, client, done) => { |
| 11 | if (err) { |
| 12 | done(err) |
| 13 | return reject(err) |
| 14 | } |
| 15 | const cursor = client.query(new Cursor(text)) |
| 16 | cursor.read(readRowCount, (err) => { |
| 17 | if (err) { |
| 18 | done(err) |
| 19 | return reject(err) |
| 20 | } |
| 21 | cursor.close((err) => { |
| 22 | if (err) { |
| 23 | done(err) |
| 24 | return reject(err) |
| 25 | } |
| 26 | done() |
| 27 | resolve() |
| 28 | }) |
| 29 | }) |
| 30 | }) |
| 31 | }) |
| 32 | } |
| 33 | |
| 34 | describe('pool', function () { |
| 35 | beforeEach(function () { |