jsharmony-db-pgsql
Version:
jsHarmony Database Connector for PostgreSQL
541 lines (530 loc) • 22.7 kB
JavaScript
/*
Copyright 2017 apHarmony
This file is part of jsHarmony.
jsHarmony is free software: you can redistribute it and/or modify
it under the terms of the GNU Lesser General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
jsHarmony is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public License
along with this package. If not, see <http://www.gnu.org/licenses/>.
*/
var JSHpgsql = require('../index');
var JSHdb = require('jsharmony-db');
var assert = require('assert');
var path = require('path');
var fs = require('fs');
var _ = require('lodash');
var os = require('os');
var moment = require('moment');
var dbconfig = { };
var path_TestDBConfig = path.join(os.homedir(),'jsharmony/testdb_pgsql.json');
if(fs.existsSync(path_TestDBConfig)){
dbconfig = JSON.parse(fs.readFileSync(path_TestDBConfig,'utf8'));
console.log('\r\n==== Loading test database config ====\r\n'+JSON.stringify(dbconfig,null,4)+'\r\n');
}
var tempTable = 'create temp table temp_c(c_id bigint); insert into temp_c(c_id) values (1);insert into temp_c(c_id) values (2);insert into temp_c(c_id) values (3);';
dbconfig = _.extend({_driver: new JSHpgsql(), host: "server.domain.com", database: "DBNAME", user: "DBUSER", password: "DBPASS", options: { pooled: true } },dbconfig);
var db = new JSHdb(dbconfig);
describe('Basic',function(){
before('Create temp table', function (done) {
//Connect to database and get data
db.Scalar('',tempTable,[],{},function(err,rslt){
assert(!err,'Success');
return done();
});
});
it('Select Parameter', function (done) {
//Connect to database and get data
var c_id = '1';
db.Recordset('','select @c_id c_id',[JSHdb.types.BigInt],{'c_id': c_id},function(err,rslt){
assert(!err,'Success');
assert((rslt && rslt.length && (rslt[0].c_id==c_id)),'Parameter returned correctly');
return done();
});
});
it('Scalar', function (done) {
//Connect to database and get data
db.Scalar('','select count(*) from temp_c',[],{},function(err,rslt){
assert(!err,'Success');
assert(rslt==3,'Scalar correct');
return done();
});
});
it('Row', function (done) {
//Connect to database and get data
var C_ID = '1';
db.Row('','select * from temp_c where c_id=@C_ID;',[JSHdb.types.BigInt],{'C_ID': C_ID},function(err,rslt){
assert(!err,'Success');
assert(rslt && (rslt.c_id==C_ID),'Recordset correct');
return done();
});
});
it('Recordset', function (done) {
//Connect to database and get data
db.Recordset('','select * from temp_c;',[],{},function(err,rslt){
assert(!err,'Success');
assert(rslt && rslt.length && (rslt.length==3) && (rslt[0].c_id==1),'Recordset correct');
return done();
});
});
it('MultiRecordset', function (done) {
//Connect to database and get data
db.MultiRecordset('',"set sessionvars.testparam to 'abc';select * from temp_c;select count(*) cnt from temp_c;",[],{},function(err,rslt){
assert(!err,'Success');
assert(rslt && rslt.length && (rslt.length==2),'Multiple recordsets returned');
assert(rslt[0] && (rslt[0].length==3) && (rslt[0][0].c_id==1),'Recordset 1 correct');
assert(rslt[1] && (rslt[1].length==1) && (rslt[1][0].cnt==3),'Recordset 2 correct');
return done();
});
});
it('Error', function (done) {
//Connect to database and get data
db.Command('','select b;',[],{},function(err,rslt){
assert(err,'Success');
return done();
});
});
it('With Statement', function (done) {
//Connect to database and get data
db.Scalar('',"with xrslt as (\
insert into temp_c(c_id) values (4)\
returning c_id as \"c_id\"\
) select * from xrslt;",[],{},function(err,rslt){
assert(!err,'Success');
assert(rslt==4,'Result correct');
return done();
});
});
it('Delete Statement', function (done) {
//Connect to database and get data
db.Command('','delete from temp_c where c_id=4;',[],{},function(err,rslt){
assert(!err,'Success');
return done();
});
});
it('Transact-SQL', function (done) {
//Connect to database and get data
db.Scalar('',"do $$ \
declare a bigint := 1;\
BEGIN\
a := a + 1;\
a := a + 1;\
a := a + 1;\
insert into temp_c(c_id) values (a);\
end$$;\
select c_id from temp_c order by c_id desc limit 1;\
delete from temp_c where c_id=4;",[],{},function(err,rslt){
assert(!err,'Success');
assert(rslt==4,'Result correct');
return done();
});
});
it('Application Error', function (done) {
//Connect to database and get data
db.Scalar('',"do $$ BEGIN RAISE EXCEPTION 'Application Error - Test Error'; end$$;",[],{},function(err,rslt){
assert(err,'Exception raised');
assert(err.message=='Application Error - Test Error','Application Error raised');
return done();
});
});
it('Application Warning', function (done) {
//Connect to database and get data
db.Scalar('',"do $$ BEGIN RAISE WARNING 'Test warning'; end$$;",[],{},function(err,rslt,stats){
assert(!err, 'Success');
assert(stats.warnings && stats.warnings.length,'Warning generated');
assert(stats.notices && !stats.notices.length,'No notice generated');
assert((stats.warnings[0].message=='Test warning') && (stats.warnings[0].severity=='WARNING'),'Warning valid');
return done();
});
});
it('Application Notice', function (done) {
//Connect to database and get data
db.Scalar('',"do $$ BEGIN RAISE NOTICE 'Test notice'; end$$;",[],{},function(err,rslt,stats){
assert(!err, 'Success');
assert(stats.notices && stats.notices.length,'Notice generated');
assert(stats.notices && !stats.warnings.length,'No warnings generated');
assert((stats.notices[0].message=='Test notice') && (stats.notices[0].severity=='NOTICE'),'Notice valid');
return done();
});
});
it('Context', function (done) {
//Connect to database and get data
db.Scalar('CONTEXT',"select current_setting('sessionvars.appuser');",[],{},function(err,rslt){
assert(rslt && (rslt.toString().substr(0,7)=='CONTEXT'),'Context found');
return done();
});
});
it('Bad Transaction', function (done) {
//Connect to database and get data
db.ExecTransTasks({
task1: function(dbtrans, callback, transtbl){
db.Command('','insert into temp_c(c_id) values(4);',[],{},dbtrans,function(err,rslt){ callback(err, rslt); });
},
task2: function(dbtrans, callback, transtbl){
db.Recordset('','select * from temp_c',[],{},dbtrans,function(err,rslt){ assert(rslt && (rslt.length==4),'Row count correct'); callback(err, rslt); });
},
task3: function(dbtrans, callback, transtbl){
db.Recordset('',"do $$ BEGIN RAISE EXCEPTION 'Application Error - Test Error'; end$$;",[],{},dbtrans,function(err,rslt){ callback(err, rslt); });
},
},function(err,rslt){
assert(err,'Rollback generated an error');
assert(err.message=='Application Error - Test Error','Application Error raised');
return done();
});
});
it('Transaction Rolled back', function (done) {
//Connect to database and get data
db.Scalar('','select count(*) from temp_c',[],{},function(err,rslt){
assert(!err,'Success');
assert(rslt==3,'Row count correct');
return done();
});
});
it('Good Transaction', function (done) {
//Connect to database and get data
db.ExecTransTasks({
task1: function(dbtrans, callback, transtbl){
db.Command('','insert into temp_c(c_id) values(4);',[],{},dbtrans,function(err,rslt,stats){ callback(err, rslt, stats); });
},
task2: function(dbtrans, callback, transtbl){
db.Command('','insert into temp_c(c_id) values(5);',[],{},dbtrans,function(err,rslt,stats){ callback(err, rslt, stats); });
},
task3: function(dbtrans, callback, transtbl){
db.Command('',"do $$ BEGIN RAISE WARNING 'Test warning'; end$$;",[],{},dbtrans,function(err,rslt,stats){ callback(err, rslt, stats); });
},
task4: function(dbtrans, callback, transtbl){
db.Command('',"do $$ BEGIN RAISE NOTICE 'Test notice'; end$$;",[],{},dbtrans,function(err,rslt,stats){ callback(err, rslt, stats); });
},
task5: function(dbtrans, callback, transtbl){
db.Recordset('',"select count(*) count from temp_c",[],{},dbtrans,function(err,rslt,stats){ callback(err, rslt, stats); });
},
},function(err,rslt,stats){
assert(!err,'Success');
assert((rslt.task5.length==1)&&(rslt.task5[0].count==5),'Correct result');
assert((stats.task3.warnings[0].message=='Test warning'),'Warning generated');
assert((stats.task4.notices[0].message=='Test notice'),'Notice generated');
return done();
});
});
it('Transaction Committed', function (done) {
//Connect to database and get data
db.Scalar('','select count(*) from temp_c',[],{},function(err,rslt){
assert(!err,'Success');
assert(rslt==5,'Row count correct');
return done();
});
});
it('Drop temp table', function (done) {
//Connect to database and get data
db.Scalar('','drop table temp_c;',[],{},function(err,rslt){
assert(!err,'Success');
return done();
});
});
it('ExecTasks - One item', function (done) {
//Connect to database and get data
db.ExecTasks([
function(callback){
db.Recordset('','select 1 a;',[],{},undefined,function(err,rslt,stats){ callback(err, rslt, stats); });
}
],function(err,rslt,stats){
assert(!err,'Success');
assert(rslt&&(rslt.length==1)&&(rslt[0].length==1)&&(rslt[0][0].a==1),'Correct result');
return done();
});
});
it('ExecTasks - Parallel', function (done) {
//Connect to database and get data
db.ExecTasks({
task1: function(callback){
db.Recordset('','select 1 a;',[],{},undefined,function(err,rslt,stats){ callback(err, rslt, stats); });
},
task2: function(callback){
db.Recordset('','select 2 b;',[],{},undefined,function(err,rslt,stats){ callback(err, rslt, stats); });
},
task3: function(callback){
db.Command('',"do $$ BEGIN RAISE WARNING 'Test warning'; end$$;",[],{},undefined,function(err,rslt,stats){ callback(err, rslt, stats); });
},
task4: function(callback){
db.Command('',"do $$ BEGIN RAISE NOTICE 'Test notice'; end$$;",[],{},undefined,function(err,rslt,stats){ callback(err, rslt, stats); });
},
},function(err,rslt,stats){
assert(!err,'Success');
assert((rslt.task1.length==1)&&(rslt.task1[0].a==1),'Correct result');
assert((stats.task3.warnings[0].message=='Test warning'),'Warning generated');
assert((stats.task4.notices[0].message=='Test notice'),'Notice generated');
return done();
});
});
it('ExecTasks - Serial & Parallel', function (done) {
//Connect to database and get data
var dbtasks = [{}, {}];
dbtasks[0] = {
task11: function(callback){
db.Recordset('','select 1 a;',[],{},undefined,function(err,rslt,stats){ callback(err, rslt, stats); });
},
task12: function(callback){
db.Recordset('','select 2 b;',[],{},undefined,function(err,rslt,stats){ callback(err, rslt, stats); });
},
task13: function(callback){
db.Command('',"do $$ BEGIN RAISE WARNING 'Test warning'; end$$;",[],{},undefined,function(err,rslt,stats){ callback(err, rslt, stats); });
},
task14: function(callback){
db.Command('',"do $$ BEGIN RAISE NOTICE 'Test notice'; end$$;",[],{},undefined,function(err,rslt,stats){ callback(err, rslt, stats); });
},
};
dbtasks[1] = {
task21: function(callback,dbrslt){
assert(dbrslt.task11 && dbrslt.task11[0] && (dbrslt.task11[0].a==1),'Series execution worked');
db.Recordset('','select 1 a;',[],{},undefined,function(err,rslt,stats){ callback(err, rslt, stats); });
},
task22: function(callback){
db.Recordset('','select 2 b;',[],{},undefined,function(err,rslt,stats){ callback(err, rslt, stats); });
},
task23: function(callback){
db.Command('',"do $$ BEGIN RAISE WARNING 'Test warning2'; end$$;",[],{},undefined,function(err,rslt,stats){ callback(err, rslt, stats); });
},
task24: function(callback){
db.Command('',"do $$ BEGIN RAISE NOTICE 'Test notice2'; end$$;",[],{},undefined,function(err,rslt,stats){ callback(err, rslt, stats); });
},
};
db.ExecTasks(dbtasks, function(err,rslt,stats){
assert(!err,'Success');
assert((rslt.task11.length==1)&&(rslt.task11[0].a==1),'Correct result');
assert((rslt.task21.length==1)&&(rslt.task21[0].a==1),'Correct result');
assert((stats.task13.warnings[0].message=='Test warning'),'Warning generated');
assert((stats.task14.notices[0].message=='Test notice'),'Notice generated');
assert((stats.task23.warnings[0].message=='Test warning2'),'Warning2 generated');
assert((stats.task24.notices[0].message=='Test notice2'),'Notice2 generated');
return done();
});
});
it('ExecTasks - Serial & Parallel Array', function (done) {
//Connect to database and get data
var dbtasks = [{}, {}];
dbtasks[0] = [
function(callback){
db.Recordset('','select 1 a;',[],{},undefined,function(err,rslt,stats){ callback(err, rslt, stats); });
},
function(callback){
db.Recordset('','select 2 b;',[],{},undefined,function(err,rslt,stats){ callback(err, rslt, stats); });
},
function(callback){
db.Command('',"do $$ BEGIN RAISE WARNING 'Test warning'; end$$;",[],{},undefined,function(err,rslt,stats){ callback(err, rslt, stats); });
},
function(callback){
db.Command('',"do $$ BEGIN RAISE NOTICE 'Test notice'; end$$;",[],{},undefined,function(err,rslt,stats){ callback(err, rslt, stats); });
},
];
dbtasks[1] = [
function(callback,dbrslt){
assert(dbrslt[0] && dbrslt[0][0] && (dbrslt[0][0].a==1),'Series execution worked');
db.Recordset('','select 1 a;',[],{},undefined,function(err,rslt,stats){ callback(err, rslt, stats); });
},
function(callback){
db.Recordset('','select 2 b;',[],{},undefined,function(err,rslt,stats){ callback(err, rslt, stats); });
},
function(callback){
db.Command('',"do $$ BEGIN RAISE WARNING 'Test warning2'; end$$;",[],{},undefined,function(err,rslt,stats){ callback(err, rslt, stats); });
},
function(callback){
db.Command('',"do $$ BEGIN RAISE NOTICE 'Test notice2'; end$$;",[],{},undefined,function(err,rslt,stats){ callback(err, rslt, stats); });
},
];
db.ExecTasks(dbtasks, function(err,rslt,stats){
assert(!err,'Success');
assert((rslt[0].length==1)&&(rslt[0][0].a==1),'Correct result');
assert((rslt[4].length==1)&&(rslt[4][0].a==1),'Correct result');
assert((stats[2].warnings[0].message=='Test warning'),'Warning generated');
assert((stats[3].notices[0].message=='Test notice'),'Notice generated');
assert((stats[6].warnings[0].message=='Test warning2'),'Warning2 generated');
assert((stats[7].notices[0].message=='Test notice2'),'Notice2 generated');
return done();
});
});
it('DB Script Notices', function (done) {
db.SQLExt.Scripts['test'] = {};
db.SQLExt.Scripts['test']['dropfakedb'] = ["drop database if exists fakedbthatdoesnotexist"];
db.RunScripts(db.platform, ['test','dropfakedb'],{},function(err,rslt,stats){
assert(!err,'Success');
assert(stats[0].notices[0].message.indexOf('skipping')>=0,'Script generated notice that we are skipping the drop');
return done();
});
});
it('Date passthru', function (done) {
//Connect to database and get data
db.Scalar('',"select to_char(@dt::date,'MM/DD/YYYY')",[JSHdb.types.Date],{'dt': moment('2018-12-03').toDate()},function(err,rslt){
assert(!err,'Success');
assert(rslt=='12/03/2018','Date passthru');
return done();
});
});
it('DateTime passthru', function (done) {
//Connect to database and get data
db.Scalar('',"select to_char(@dt::timestamp,'MM/DD/YYYY')",[JSHdb.types.DateTime(7)],{'dt': moment('2018-12-03').toDate()},function(err,rslt){
assert(!err,'Success');
assert(rslt=='12/03/2018','Date passthru');
return done();
});
});
it('ExecTasks - Stream Empty', function (done) {
//Connect to database and get data
var rslt = [];
db.StreamRecordset('','select 1 where 0=1;',[],{},{
onRow: function(row, row_cb){
rslt.push(row);
row_cb();
},
onComplete: function(err){
assert(!err,'Success');
assert(rslt&&(rslt.length==0),'Correct result');
return done();
}
});
});
it('ExecTasks - Stream Multiple', function (done) {
//Connect to database and get data
var rslt = [];
db.StreamRecordset('','select 1 a union select 2 a union select 3 a;',[],{},{
onRow: function(row, row_cb){
rslt.push(row);
row_cb();
},
onComplete: function(err){
assert(!err,'Success');
assert(rslt&&(rslt.length==3)&&(rslt[0].a==1)&&(rslt[1].a==2)&&(rslt[2].a==3),'Correct result');
return done();
}
});
});
it('ExecTasks - Stream Error', function (done) {
//Connect to database and get data
var rslt = [];
db.StreamRecordset('','select a;',[],{},{
onRow: function(row, row_cb){
rslt.push(row);
row_cb();
},
onComplete: function(err){
assert(err,'Has Error');
assert(rslt&&(rslt.length==0),'No results');
return done();
}
});
});
it('ExecTasks - Stream Notice', function (done) {
//Connect to database and get data
var rslt = [];
var notices = [];
var warnings = [];
db.StreamRecordset('',"do $$ BEGIN RAISE NOTICE 'Test notice'; end$$;",[],{},{
onRow: function(row, row_cb){
rslt.push(row);
row_cb();
},
onNotice: function(notice){
notices.push(notice);
},
onWarning: function(warning){
warnings.push(warning);
},
onComplete: function(err){
assert(!err,'Success');
assert(rslt&&(rslt.length==0),'No results');
assert(notices.length && (notices[0].message=='Test notice') && (notices[0].severity=='NOTICE'),'Notice valid');
assert(warnings&&(warnings.length==0),'No warnings');
return done();
}
});
});
it('ExecTasks - Stream Warning', function (done) {
//Connect to database and get data
var rslt = [];
var notices = [];
var warnings = [];
db.StreamRecordset('',"do $$ BEGIN RAISE WARNING 'Test warning'; end$$;",[],{},{
onRow: function(row, row_cb){
rslt.push(row);
row_cb();
},
onNotice: function(notice){
notices.push(notice);
},
onWarning: function(warning){
warnings.push(warning);
},
onComplete: function(err){
assert(!err,'Success');
assert(rslt&&(rslt.length==0),'No results');
assert(notices&&(notices.length==0),'No warnings');
assert(warnings.length && (warnings[0].message=='Test warning') && (warnings[0].severity=='WARNING'),'Warning valid');
return done();
}
});
});
it('ExecTasks - Stream Pause', function (done) {
//Connect to database and get data
var rslt = [];
db.StreamRecordset('','select 1 a union select 2 a union select 3 a;',[],{},{
onRow: function(row, row_cb){
setTimeout(function(){
rslt.push(row);
row_cb();
}, 100);
},
onDrained: function(err){
assert(!err,'Success');
assert(rslt&&(rslt.length==3)&&(rslt[0].a==1)&&(rslt[1].a==2)&&(rslt[2].a==3),'Correct result');
return done();
}
});
});
it('ExecTasks - Stream Error', function (done) {
//Connect to database and get data
var rslt = [];
db.StreamRecordset('',"select 1 a union select 2 a union select 3 a;",[],{},{
onRow: function(row, row_cb){
setTimeout(function(){
rslt.push(row);
row_cb((rslt.length == 2) && (new Error('Test write error')));
}, 200);
},
onDrained: function(err){
assert(err,'Error');
assert(rslt&&(rslt.length==2)&&(rslt[0].a==1)&&(rslt[1].a==2),'Correct result');
return done();
}
});
});
it('ExecTasks - Multiline Statement for Recordset', function (done) {
//Connect to database and get data
db.Recordset('',"set sessionvars.appuser to 'task';set sessionvars.last_trigger_source to '';select 1 a union select 2 a union select 3 a;",[],{},function(err,rslt){
assert(!err,'Success');
assert(rslt&&(rslt.length==3)&&(rslt[0].a==1)&&(rslt[1].a==2)&&(rslt[2].a==3),'Correct result');
return done();
});
});
it('ExecTasks - Error on StreamRecordset w/Context (Multiline StreamRecordset not supported w/PostgreSQL)', function (done) {
var err = false;
try{
db.StreamRecordset('abc',"select 1 a union select 2 a union select 3 a;",[],{},{});
}
catch(ex){
err = true;
}
assert(err,'Error');
return done();
});
after(function(done){
assert(db.dbconfig._driver.pool.length==1,'Pool exists');
assert(db.dbconfig._driver.pool[0].con,'Pool connected');
db.Close(function(){
assert(!db.dbconfig._driver.pool[0].con,'Pool closed');
return done();
});
});
});