|
| 1 | +var assert = require('assert') |
| 2 | +var helper = require('./helper') |
| 3 | +var QueryStream = require('../') |
| 4 | + |
| 5 | +helper('passing options', function(client) { |
| 6 | + it('passes row mode array', function(done) { |
| 7 | + var stream = new QueryStream('SELECT * FROM generate_series(0, 10) num', [], { rowMode: 'array' }) |
| 8 | + var query = client.query(stream) |
| 9 | + var result = [] |
| 10 | + query.on('data', datum => { |
| 11 | + result.push(datum) |
| 12 | + }) |
| 13 | + query.on('end', () => { |
| 14 | + const expected = new Array(11).fill(0).map((_, i) => [i]) |
| 15 | + assert.deepEqual(result, expected) |
| 16 | + done() |
| 17 | + }) |
| 18 | + }) |
| 19 | + |
| 20 | + it('passes custom types', function(done) { |
| 21 | + const types = { |
| 22 | + getTypeParser: () => string => string, |
| 23 | + } |
| 24 | + var stream = new QueryStream('SELECT * FROM generate_series(0, 10) num', [], { types }) |
| 25 | + var query = client.query(stream) |
| 26 | + var result = [] |
| 27 | + query.on('data', datum => { |
| 28 | + result.push(datum) |
| 29 | + }) |
| 30 | + query.on('end', () => { |
| 31 | + const expected = new Array(11).fill(0).map((_, i) => ({ |
| 32 | + num: i.toString(), |
| 33 | + })) |
| 34 | + assert.deepEqual(result, expected) |
| 35 | + done() |
| 36 | + }) |
| 37 | + }) |
| 38 | +}) |
0 commit comments