mirror of
https://github.com/brianc/node-postgres.git
synced 2025-12-08 20:16:25 +00:00
79 lines
2.0 KiB
JavaScript
79 lines
2.0 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(
|
|
'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()
|
|
})
|
|
)
|