You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
105 lines
3.0 KiB
105 lines
3.0 KiB
'use strict';
|
|
|
|
var assert = require('assert')
|
|
var gonna = require('gonna')
|
|
|
|
var concat = require('concat-stream')
|
|
var _ = require('lodash')
|
|
var pg = require('pg')
|
|
|
|
var copy = require('../').from
|
|
|
|
var client = function() {
|
|
var client = new pg.Client()
|
|
client.connect()
|
|
return client
|
|
}
|
|
|
|
var testConstruction = function() {
|
|
var highWaterMark = 10
|
|
var stream = copy('COPY numbers FROM STDIN', {highWaterMark: 10, objectMode: true})
|
|
for(var i = 0; i < highWaterMark * 1.5; i++) {
|
|
stream.write('1\t2\n')
|
|
}
|
|
assert(!stream.write('1\t2\n'), 'Should correctly set highWaterMark.')
|
|
}
|
|
|
|
testConstruction()
|
|
|
|
var testRange = function(top) {
|
|
var fromClient = client()
|
|
fromClient.query('CREATE TEMP TABLE numbers(num int, bigger_num int)')
|
|
|
|
var txt = 'COPY numbers FROM STDIN'
|
|
var stream = fromClient.query(copy(txt))
|
|
for(var i = 0; i < top; i++) {
|
|
stream.write(Buffer.from('' + i + '\t' + i*10 + '\n'))
|
|
}
|
|
stream.end()
|
|
var countDone = gonna('have correct count')
|
|
stream.on('end', function() {
|
|
fromClient.query('SELECT COUNT(*) FROM numbers', function(err, res) {
|
|
assert.ifError(err)
|
|
assert.equal(res.rows[0].count, top, 'expected ' + top + ' rows but got ' + res.rows[0].count)
|
|
assert.equal(stream.rowCount, top, 'expected ' + top + ' rows but db count is ' + stream.rowCount)
|
|
//console.log('found ', res.rows.length, 'rows')
|
|
countDone()
|
|
var firstRowDone = gonna('have correct result')
|
|
assert.equal(stream.rowCount, top, 'should have rowCount ' + top + ' ')
|
|
fromClient.query('SELECT (max(num)) AS num FROM numbers', function(err, res) {
|
|
assert.ifError(err)
|
|
assert.equal(res.rows[0].num, top-1)
|
|
firstRowDone()
|
|
fromClient.end()
|
|
})
|
|
})
|
|
})
|
|
}
|
|
|
|
testRange(1000)
|
|
|
|
var testSingleEnd = function() {
|
|
var fromClient = client()
|
|
fromClient.query('CREATE TEMP TABLE numbers(num int)')
|
|
var txt = 'COPY numbers FROM STDIN';
|
|
var stream = fromClient.query(copy(txt))
|
|
var count = 0;
|
|
stream.on('end', function() {
|
|
count++;
|
|
assert(count==1, '`end` Event was triggered ' + count + ' times');
|
|
if (count == 1) fromClient.end();
|
|
})
|
|
stream.end(Buffer.from('1\n'))
|
|
|
|
}
|
|
testSingleEnd()
|
|
|
|
var testClientReuse = function() {
|
|
var fromClient = client()
|
|
fromClient.query('CREATE TEMP TABLE numbers(num int)')
|
|
var txt = 'COPY numbers FROM STDIN';
|
|
var count = 0;
|
|
var countMax = 2;
|
|
var card = 100000;
|
|
var runStream = function() {
|
|
var stream = fromClient.query(copy(txt))
|
|
stream.on('end', function() {
|
|
count++;
|
|
if (count<countMax) {
|
|
runStream()
|
|
} else {
|
|
fromClient.query('SELECT sum(num) AS s FROM numbers', function(err, res) {
|
|
var total = countMax * card * (card+1)
|
|
assert.equal(res.rows[0].s, total, 'copy-from.ClientReuse wrong total')
|
|
fromClient.end()
|
|
})
|
|
}
|
|
})
|
|
stream.write(Buffer.from(_.range(0, card+1).join('\n') + '\n'))
|
|
stream.end(Buffer.from(_.range(0, card+1).join('\n') + '\n'))
|
|
}
|
|
runStream();
|
|
}
|
|
testClientReuse()
|
|
|