node-task/node_modules/mysql2/test/integration/connection/test-multiple-results.js

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);