mirror of
https://github.com/brianc/node-postgres.git
synced 2026-01-18 15:55:05 +00:00
83a0e3e90e
When enabling this rule, it's recommended to also *disable* the standard `no-unused-vars` rule. Although `no-unused-vars` is not currently enabled, it seems helpful to explicitly disable it here. See: https://typescript-eslint.io/rules/no-unused-vars/ Co-authored-by: alxndrsn <alxndrsn>
104 lines
2.6 KiB
JavaScript
104 lines
2.6 KiB
JavaScript
'use strict'
|
|
const assert = require('assert')
|
|
const co = require('co')
|
|
|
|
const helper = require('./test-helper')
|
|
|
|
const suite = new helper.Suite('multiple result sets')
|
|
|
|
suite.test(
|
|
'two select results work',
|
|
co.wrap(function* () {
|
|
const client = new helper.Client()
|
|
yield client.connect()
|
|
|
|
const results = yield client.query(`SELECT 'foo'::text as name; SELECT 'bar'::text as baz`)
|
|
assert(Array.isArray(results))
|
|
|
|
assert.equal(results[0].fields[0].name, 'name')
|
|
assert.deepEqual(results[0].rows, [{ name: 'foo' }])
|
|
|
|
assert.equal(results[1].fields[0].name, 'baz')
|
|
assert.deepEqual(results[1].rows, [{ baz: 'bar' }])
|
|
|
|
return client.end()
|
|
})
|
|
)
|
|
|
|
suite.test(
|
|
'throws if queryMode set to "extended"',
|
|
co.wrap(function* () {
|
|
const client = new helper.Client()
|
|
yield client.connect()
|
|
|
|
// TODO should be text or sql?
|
|
try {
|
|
yield client.query({
|
|
text: `SELECT 'foo'::text as name; SELECT 'bar'::text as baz`,
|
|
queryMode: 'extended',
|
|
})
|
|
assert.fail('Should have thrown')
|
|
} catch (err) {
|
|
if (err instanceof assert.AssertionError) throw err
|
|
|
|
assert.equal(err.severity, 'ERROR')
|
|
assert.equal(err.code, '42601')
|
|
assert.equal(err.message, 'cannot insert multiple commands into a prepared statement')
|
|
}
|
|
|
|
return client.end()
|
|
})
|
|
)
|
|
|
|
suite.test(
|
|
'multiple selects work',
|
|
co.wrap(function* () {
|
|
const client = new helper.Client()
|
|
yield client.connect()
|
|
|
|
const text = `
|
|
SELECT * FROM generate_series(2, 4) as foo;
|
|
SELECT * FROM generate_series(8, 10) as bar;
|
|
SELECT * FROM generate_series(20, 22) as baz;
|
|
`
|
|
|
|
const results = yield client.query(text)
|
|
assert(Array.isArray(results))
|
|
|
|
assert.equal(results[0].fields[0].name, 'foo')
|
|
assert.deepEqual(results[0].rows, [{ foo: 2 }, { foo: 3 }, { foo: 4 }])
|
|
|
|
assert.equal(results[1].fields[0].name, 'bar')
|
|
assert.deepEqual(results[1].rows, [{ bar: 8 }, { bar: 9 }, { bar: 10 }])
|
|
|
|
assert.equal(results[2].fields[0].name, 'baz')
|
|
assert.deepEqual(results[2].rows, [{ baz: 20 }, { baz: 21 }, { baz: 22 }])
|
|
|
|
assert.equal(results.length, 3)
|
|
|
|
return client.end()
|
|
})
|
|
)
|
|
|
|
suite.test(
|
|
'mixed queries and statements',
|
|
co.wrap(function* () {
|
|
const client = new helper.Client()
|
|
yield client.connect()
|
|
|
|
const text = `
|
|
CREATE TEMP TABLE weather(type text);
|
|
INSERT INTO weather(type) VALUES ('rain');
|
|
SELECT * FROM weather;
|
|
`
|
|
|
|
const results = yield client.query(text)
|
|
assert(Array.isArray(results))
|
|
assert.equal(results[0].command, 'CREATE')
|
|
assert.equal(results[1].command, 'INSERT')
|
|
assert.equal(results[2].command, 'SELECT')
|
|
|
|
return client.end()
|
|
})
|
|
)
|