|
| 1 | +const assert = require('assert') |
| 2 | +const Cursor = require('../') |
| 3 | +const pg = require('pg') |
| 4 | + |
| 5 | +describe('transactions', () => { |
| 6 | + it('can execute multiple statements in a transaction', async () => { |
| 7 | + const client = new pg.Client() |
| 8 | + await client.connect() |
| 9 | + await client.query('begin') |
| 10 | + await client.query('CREATE TEMP TABLE foobar(id SERIAL PRIMARY KEY)') |
| 11 | + const cursor = client.query(new Cursor('SELECT * FROM foobar')) |
| 12 | + const rows = await new Promise((resolve, reject) => { |
| 13 | + cursor.read(10, (err, rows) => (err ? reject(err) : resolve(rows))) |
| 14 | + }) |
| 15 | + assert.equal(rows.length, 0) |
| 16 | + await client.query('ALTER TABLE foobar ADD COLUMN name TEXT') |
| 17 | + await client.end() |
| 18 | + }) |
| 19 | + |
| 20 | + it('can execute multiple statements in a transaction if ending cursor early', async () => { |
| 21 | + const client = new pg.Client() |
| 22 | + await client.connect() |
| 23 | + await client.query('begin') |
| 24 | + await client.query('CREATE TEMP TABLE foobar(id SERIAL PRIMARY KEY)') |
| 25 | + const cursor = client.query(new Cursor('SELECT * FROM foobar')) |
| 26 | + await new Promise(resolve => cursor.close(resolve)) |
| 27 | + await client.query('ALTER TABLE foobar ADD COLUMN name TEXT') |
| 28 | + await client.end() |
| 29 | + }) |
| 30 | + |
| 31 | + it.only('can execute multiple statements in a transaction if no data', async () => { |
| 32 | + const client = new pg.Client() |
| 33 | + await client.connect() |
| 34 | + await client.query('begin') |
| 35 | + // create a cursor that has no data response |
| 36 | + const createText = 'CREATE TEMP TABLE foobar(id SERIAL PRIMARY KEY)' |
| 37 | + const cursor = client.query(new Cursor(createText)) |
| 38 | + const err = await new Promise(resolve => cursor.read(100, resolve)) |
| 39 | + assert.ifError(err) |
| 40 | + await client.query('ALTER TABLE foobar ADD COLUMN name TEXT') |
| 41 | + await client.end() |
| 42 | + }) |
| 43 | +}) |
0 commit comments