mirror of
https://github.com/brianc/node-postgres.git
synced 2025-12-08 20:16:25 +00:00
779803fbce
* Add ref/unref noop to native client * Use promise.catch in test * Make partition test not flake on old node * Fix test flake on old node
52 lines
1.2 KiB
JavaScript
52 lines
1.2 KiB
JavaScript
const assert = require('assert')
|
|
const Cursor = require('../')
|
|
const pg = require('pg')
|
|
|
|
const text = 'SELECT generate_series as num FROM generate_series(0, 5)'
|
|
|
|
describe('cursor using promises', function () {
|
|
beforeEach(function (done) {
|
|
const client = (this.client = new pg.Client())
|
|
client.connect(done)
|
|
|
|
this.pgCursor = function (text, values) {
|
|
return client.query(new Cursor(text, values || []))
|
|
}
|
|
})
|
|
|
|
afterEach(function () {
|
|
this.client.end()
|
|
})
|
|
|
|
it('resolve with result', async function () {
|
|
const cursor = this.pgCursor(text)
|
|
const res = await cursor.read(6)
|
|
assert.strictEqual(res.length, 6)
|
|
})
|
|
|
|
it('reject with error', function (done) {
|
|
const cursor = this.pgCursor('select asdfasdf')
|
|
cursor.read(1).catch((err) => {
|
|
assert(err)
|
|
done()
|
|
})
|
|
})
|
|
|
|
it('read multiple times', async function () {
|
|
const cursor = this.pgCursor(text)
|
|
let res
|
|
|
|
res = await cursor.read(2)
|
|
assert.strictEqual(res.length, 2)
|
|
|
|
res = await cursor.read(3)
|
|
assert.strictEqual(res.length, 3)
|
|
|
|
res = await cursor.read(1)
|
|
assert.strictEqual(res.length, 1)
|
|
|
|
res = await cursor.read(1)
|
|
assert.strictEqual(res.length, 0)
|
|
})
|
|
})
|