loopback-connector-mysql/test/connection.test.js

require('./init.js');
var assert = require('assert');

var db, DummyModel, odb;

describe('connections', function () {

  before(function () {
    require('./init.js');

    odb = getDataSource({collation: 'utf8_general_ci', createDatabase: true});
    db = odb;
  });

  it('should use utf8 charset', function (done) {

    var test_set = /utf8/;
    var test_collo = /utf8_general_ci/;
    var test_set_str = 'utf8';
    var test_set_collo = 'utf8_general_ci';
    charsetTest(test_set, test_collo, test_set_str, test_set_collo, done);

  });

  it('should disconnect first db', function (done) {
    db.disconnect(function () {
      odb = getDataSource();
      done();
    });
  });

  it('should use latin1 charset', function (done) {

    var test_set = /latin1/;
    var test_collo = /latin1_general_ci/;
    var test_set_str = 'latin1';
    var test_set_collo = 'latin1_general_ci';
    charsetTest(test_set, test_collo, test_set_str, test_set_collo, done);

  });

  it('should drop db and disconnect all', function (done) {
    db.connector.execute('DROP DATABASE IF EXISTS ' + db.settings.database, function (err) {
      db.disconnect(function () {
        done();
      });
    });
  });
});

function charsetTest(test_set, test_collo, test_set_str, test_set_collo, done) {

  query('DROP DATABASE IF EXISTS ' + odb.settings.database, function (err) {
    assert.ok(!err);
    odb.disconnect(function () {

      db = getDataSource({collation: test_set_collo, createDatabase: true});
      DummyModel = db.define('DummyModel', {string: String});
      db.automigrate(function () {
        var q = 'SELECT DEFAULT_COLLATION_NAME' +
          ' FROM information_schema.SCHEMATA WHERE SCHEMA_NAME = ' +
          db.driver.escape(db.settings.database) + ' LIMIT 1';
        db.connector.execute(q, function (err, r) {
          assert.ok(!err);
          assert.ok(r[0].DEFAULT_COLLATION_NAME.match(test_collo));
          db.connector.execute('SHOW VARIABLES LIKE "character_set%"', function (err, r) {
            assert.ok(!err);
            var hit_all = 0;
            for (var result in r) {
              hit_all += matchResult(r[result], 'character_set_connection', test_set);
              hit_all += matchResult(r[result], 'character_set_database', test_set);
              hit_all += matchResult(r[result], 'character_set_results', test_set);
              hit_all += matchResult(r[result], 'character_set_client', test_set);
            }
            assert.equal(hit_all, 4);
          });
          db.connector.execute('SHOW VARIABLES LIKE "collation%"', function (err, r) {
            assert.ok(!err);
            var hit_all = 0;
            for (var result in r) {
              hit_all += matchResult(r[result], 'collation_connection', test_set);
              hit_all += matchResult(r[result], 'collation_database', test_set);
            }
            assert.equal(hit_all, 2);
            done();
          });
        });
      });
    });
  });

}

function matchResult(result, variable_name, match) {
  if (result.Variable_name === variable_name) {
    assert.ok(result.Value.match(match));
    return 1;
  }
  return 0;
}

var query = function (sql, cb) {
  odb.connector.execute(sql, cb);
};