143 lines
4.5 KiB
JavaScript
143 lines
4.5 KiB
JavaScript
var mysql = require('../../common').createConnection({multipleStatements: true});
|
|
var assert = require('assert');
|
|
mysql.query('CREATE TEMPORARY TABLE no_rows (test int)');
|
|
mysql.query('CREATE TEMPORARY TABLE some_rows (test int)');
|
|
mysql.query('INSERT INTO some_rows values(0)');
|
|
mysql.query('INSERT INTO some_rows values(42)');
|
|
mysql.query('INSERT INTO some_rows values(314149)');
|
|
|
|
var clone = function(obj) { return JSON.parse(JSON.stringify(obj)); };
|
|
|
|
var rs1 = {
|
|
affectedRows: 0,
|
|
fieldCount: 0,
|
|
insertId: 0,
|
|
serverStatus: 10,
|
|
warningStatus: 0
|
|
};
|
|
var rs2 = clone(rs1);
|
|
rs2.serverStatus = 2;
|
|
|
|
var twoInsertResult = [[rs1, rs2], [undefined, undefined], 2];
|
|
var select1 = [{"1":"1"}];
|
|
var select2 = [{"2":"2"}];
|
|
var fields1 = [{
|
|
catalog: "def",
|
|
characterSet: 63,
|
|
columnLength: 1,
|
|
columnType: 8,
|
|
decimals: 0,
|
|
flags: 129,
|
|
name: "1",
|
|
orgName: "",
|
|
orgTable: "",
|
|
schema: "",
|
|
table: ""
|
|
}];
|
|
var nr_fields = [{
|
|
catalog: "def",
|
|
characterSet: 63,
|
|
columnLength: 11,
|
|
columnType: 3,
|
|
decimals: 0,
|
|
flags: 0,
|
|
name: "test",
|
|
orgName: "test",
|
|
orgTable: "no_rows",
|
|
schema: "test",
|
|
table: "no_rows"
|
|
}];
|
|
var sr_fields = clone(nr_fields);
|
|
sr_fields[0].orgTable = "some_rows";
|
|
sr_fields[0].table = "some_rows";
|
|
var select3 = [{"test":0},{"test":42},{"test":314149}];
|
|
|
|
var fields2 = clone(fields1);
|
|
fields2[0].name = "2";
|
|
|
|
var tests = [
|
|
["select * from some_rows", [select3,sr_fields,1]], // select 3 rows
|
|
["SET @OLD_CHARACTER_SET_CLIENT=@@CHARACTER_SET_CLIENT; SET @OLD_CHARACTER_SET_RESULTS=@@CHARACTER_SET_RESULTS;", twoInsertResult],
|
|
["/*!40101 SET @OLD_CHARACTER_SET_CLIENT=@@CHARACTER_SET_CLIENT */;/*!40101 SET @OLD_CHARACTER_SET_RESULTS=@@CHARACTER_SET_RESULTS */;", twoInsertResult], // issue #26
|
|
["set @a = 1", [rs2, undefined, 1]], // one insert result
|
|
["set @a = 1; set @b = 2", twoInsertResult],
|
|
["select 1; select 2", [[select1,select2],[fields1,fields2], 2]],
|
|
["set @a = 1; select 1", [[rs1, select1], [undefined, fields1], 2]],
|
|
["select 1; set @a = 1", [[select1, rs2], [fields1, undefined], 2]],
|
|
["select * from no_rows", [[], nr_fields, 1]], // select 0 rows"
|
|
["set @a = 1; select * from no_rows", [[rs1, []], [undefined, nr_fields], 2]], // insert + select 0 rows
|
|
["select * from no_rows; set @a = 1", [[[], rs2], [nr_fields, undefined], 2]], // select 0 rows + insert
|
|
["set @a = 1; select * from some_rows", [[rs1, select3],[undefined,sr_fields],2]], // insert + select 3 rows
|
|
["select * from some_rows; set @a = 1", [[select3,rs2],[sr_fields,undefined],2]] // select 3 rows + insert
|
|
];
|
|
|
|
// TODO: tests with error in the query with different index
|
|
// TODO: multiple results from single query
|
|
|
|
function do_test(testIndex) {
|
|
var entry = tests[testIndex];
|
|
var sql = entry[0];
|
|
var expectation = entry[1];
|
|
mysql.query(sql, function(err, _rows, _columns) {
|
|
var _numResults = 0;
|
|
if (_rows.constructor.name == 'ResultSetHeader')
|
|
_numResults = 1;
|
|
else if (_rows.length === 0) {
|
|
// empty select
|
|
_numResults = 1;
|
|
}
|
|
else if (_rows.length > 0) {
|
|
if (_rows.constructor.name == 'Array' && _rows[0].constructor.name == 'TextRow')
|
|
_numResults = 1;
|
|
if (_rows.constructor.name == 'Array' &&
|
|
(_rows[0].constructor.name == 'Array' || _rows[0].constructor.name =='ResultSetHeader'))
|
|
_numResults = _rows.length
|
|
}
|
|
if (err) {
|
|
console.log(err);
|
|
process.exit(-1);
|
|
}
|
|
assert.deepEqual(expectation, [_rows, _columns, _numResults]);
|
|
var q = mysql.query(sql);
|
|
var resIndex = 0;
|
|
var rowIndex = 0;
|
|
function checkRow(row, index) {
|
|
if (_numResults == 1) {
|
|
assert.equal(index, 0);
|
|
if (row.constructor.name == 'ResultSetHeader')
|
|
assert.deepEqual(_rows, row);
|
|
else
|
|
assert.deepEqual(_rows[rowIndex], row);
|
|
} else {
|
|
if (resIndex != index) {
|
|
rowIndex = 0;
|
|
resIndex = index;
|
|
}
|
|
if (row.constructor.name == 'ResultSetHeader')
|
|
assert.deepEqual(_rows[index], row);
|
|
else
|
|
assert.deepEqual(_rows[index][rowIndex], row);
|
|
}
|
|
rowIndex++;
|
|
}
|
|
function checkFields(fields, index) {
|
|
if (_numResults == 1) {
|
|
assert.equal(index, 0);
|
|
assert.deepEqual(_columns, fields);
|
|
}
|
|
else
|
|
assert.deepEqual(_columns[index], fields);
|
|
}
|
|
q.on('result', checkRow);
|
|
q.on('fields', checkFields);
|
|
q.on('end', function() {
|
|
if (testIndex + 1 < tests.length)
|
|
do_test(testIndex + 1);
|
|
else {
|
|
mysql.end();
|
|
}
|
|
});
|
|
});
|
|
}
|
|
do_test(0);
|