69 lines
1.6 KiB
JavaScript
69 lines
1.6 KiB
JavaScript
|
var mysql = require('../../../index.js');
|
||
|
|
var auth = require('../../../lib/auth_41.js');
|
||
|
|
var assert = require('assert');
|
||
|
|
|
||
|
|
var server;
|
||
|
|
|
||
|
|
function authenticate(params, cb) {
|
||
|
|
var doubleSha = auth.doubleSha1('testpassword');
|
||
|
|
var isValid = auth.verifyToken(params.authPluginData1, params.authPluginData2, params.authToken, doubleSha);
|
||
|
|
assert(isValid);
|
||
|
|
cb(null);
|
||
|
|
}
|
||
|
|
|
||
|
|
var queryCalls = 0;
|
||
|
|
|
||
|
|
var server = mysql.createServer();
|
||
|
|
server.listen(3307);
|
||
|
|
server.on('connection', function(conn) {
|
||
|
|
conn.serverHandshake({
|
||
|
|
protocolVersion: 10,
|
||
|
|
serverVersion: 'node.js rocks',
|
||
|
|
connectionId: 1234,
|
||
|
|
statusFlags: 2,
|
||
|
|
characterSet: 8,
|
||
|
|
capabilityFlags: 0xffffff,
|
||
|
|
authCallback: authenticate
|
||
|
|
});
|
||
|
|
conn.on('query', function(sql) {
|
||
|
|
assert.equal(sql, 'select 1+1');
|
||
|
|
queryCalls++;
|
||
|
|
conn.close();
|
||
|
|
});
|
||
|
|
});
|
||
|
|
|
||
|
|
var connection = mysql.createConnection({
|
||
|
|
port: 3307,
|
||
|
|
user: 'testuser',
|
||
|
|
database: 'testdatabase',
|
||
|
|
passwordSha1: Buffer('8bb6118f8fd6935ad0876a3be34a717d32708ffd', 'hex')
|
||
|
|
});
|
||
|
|
|
||
|
|
connection.on('error', function(err) {
|
||
|
|
assert.equal(err.code, 'PROTOCOL_CONNECTION_LOST');
|
||
|
|
});
|
||
|
|
|
||
|
|
connection.query('select 1+1', function(err) {
|
||
|
|
assert.equal(err.code, 'PROTOCOL_CONNECTION_LOST');
|
||
|
|
server._server.close();
|
||
|
|
});
|
||
|
|
|
||
|
|
var _1_2 = false;
|
||
|
|
var _1_3 = false;
|
||
|
|
|
||
|
|
connection.query('select 1+2', function(err) {
|
||
|
|
assert.equal(err.code, 'PROTOCOL_CONNECTION_LOST');
|
||
|
|
_1_2 = true;
|
||
|
|
});
|
||
|
|
|
||
|
|
connection.query('select 1+3', function(err) {
|
||
|
|
assert.equal(err.code, 'PROTOCOL_CONNECTION_LOST');
|
||
|
|
_1_3 = true;
|
||
|
|
});
|
||
|
|
|
||
|
|
process.on('exit', function() {
|
||
|
|
assert.equal(queryCalls, 1);
|
||
|
|
assert.equal(_1_2, true);
|
||
|
|
assert.equal(_1_3, true);
|
||
|
|
});
|