tests: add basic auth switch helpers to unit test server

Author: dougwilson

test/FakeServer.js

@@ -71,6 +71,10 @@ function FakeConnection(socket) {
   socket.on('data', this._handleData.bind(this));
 }
 
+FakeConnection.prototype.authSwitchRequest = function authSwitchRequest(options) {
+  this._sendPacket(new Packets.AuthSwitchRequestPacket(options));
+};
+
 FakeConnection.prototype.deny = function deny(message, errno) {
   message = message || 'Access Denied';
   errno   = errno || Errors.ER_ACCESS_DENIED_ERROR;
@@ -280,6 +284,11 @@ FakeConnection.prototype._parsePacket = function() {
   packet.parse(this._parser);
 
   switch (Packet) {
+    case Packets.AuthSwitchResponsePacket:
+      if (!this.emit('authSwitchResponse', packet)) {
+        this.deny('No auth response handler');
+      }
+      break;
     case Packets.ClientAuthenticationPacket:
       this.database = (packet.database || null);
       this.user     = (packet.user || null);

test/common.js

@@ -18,6 +18,7 @@ common.fakeServerSocket = __dirname + '/fake_server.sock';
 common.testDatabase = process.env.MYSQL_DATABASE || 'test';
 
 // Export common modules
+common.Auth             = require(common.lib + '/protocol/Auth');
 common.Charsets         = require(common.lib + '/protocol/constants/charsets');
 common.ClientConstants  = require(common.lib + '/protocol/constants/client');
 common.Connection       = require(common.lib + '/Connection');

test/unit/connection/test-auth-switch-native.js

@@ -5,7 +5,6 @@ var connection = common.createConnection({
   port     : common.fakeServerPort,
   password : 'authswitch'
 });
-var Auth       = require(common.lib + '/protocol/Auth');
 
 var random = Crypto.pseudoRandomBytes || Crypto.randomBytes; // Depends on node.js version
 var server = common.createFakeServer();
@@ -28,15 +27,15 @@ server.on('connection', function(incomingConnection) {
   random(20, function (err, scramble) {
     assert.ifError(err);
 
+    incomingConnection.on('authSwitchResponse', function (packet) {
+      this._sendAuthResponse(packet.data, common.Auth.token('authswitch', scramble));
+    });
+
     incomingConnection.on('clientAuthentication', function () {
-      this._sendPacket(new common.Packets.AuthSwitchRequestPacket({
+      this.authSwitchRequest({
         authMethodName : 'mysql_native_password',
         authMethodData : scramble
-      }));
-    });
-
-    incomingConnection.on('AuthSwitchResponsePacket', function (packet) {
-      this._sendAuthResponse(packet.data, Auth.token('authswitch', scramble));
+      });
     });
 
     incomingConnection.handshake();

test/unit/connection/test-auth-switch-unknown.js

@@ -24,10 +24,10 @@ server.listen(common.fakeServerPort, function (err) {
 
 server.on('connection', function(incomingConnection) {
   incomingConnection.on('clientAuthentication', function () {
-    this._sendPacket(new common.Packets.AuthSwitchRequestPacket({
+    this.authSwitchRequest({
       authMethodName : 'foo_plugin_password',
       authMethodData : Buffer.alloc(0)
-    }));
+    });
   });
 
   incomingConnection.handshake();