authentication_test.js

var mongodb = process.env['TEST_NATIVE'] != null ? require('../lib/mongodb').native() : require('../lib/mongodb').pure();
var useSSL = process.env['USE_SSL'] != null ? true : false;

var testCase = require('nodeunit').testCase,
  debug = require('util').debug,
  inspect = require('util').inspect,
  nodeunit = require('nodeunit'),
  gleak = require('../dev/tools/gleak'),
  Db = mongodb.Db,
  Cursor = mongodb.Cursor,
  Collection = mongodb.Collection,
  Server = mongodb.Server;

var MONGODB = 'integration_tests';
var client = null;

/**
 * Retrieve the server information for the current
 * instance of the db client
 * 
 * @ignore
 */
exports.setUp = function(callback) {
  var self = exports;  
  client = new Db(MONGODB, new Server("127.0.0.1", 27017, {auto_reconnect: true, poolSize: 4, ssl:useSSL}), {native_parser: (process.env['TEST_NATIVE'] != null)});
  client.open(function(err, db_p) {
    if(numberOfTestsRun == (Object.keys(self).length)) {
      // If first test drop the db
      client.dropDatabase(function(err, done) {
        callback();
      });
    } else {
      return callback();
    }
  });
}

/**
 * Retrieve the server information for the current
 * instance of the db client
 * 
 * @ignore
 */
exports.tearDown = function(callback) {
  var self = this;
  numberOfTestsRun = numberOfTestsRun - 1;
  // Close connection
  client.close();
  callback();
}

/**
 * Test the authentication method for the user
 * 
 * @ignore
 */
exports.shouldCorrectlyAuthenticate = function(test) {
  var user_name = 'spongebob';
  var password = 'squarepants';

  client.authenticate('admin', 'admin', function(err, replies) {      
    test.ok(err instanceof Error);
    test.ok(!replies);

    // Add a user
    client.addUser(user_name, password, function(err, result) {
      client.authenticate(user_name, password, function(err, replies) {
        test.ok(!(err instanceof Error));
        test.ok(replies);
        test.done();
      });
    });
  });    
}

/**
 * Test the authentication method for the user
 * 
 * @ignore
 */
exports.shouldCorrectlyReAuthorizeReconnectedConnections = function(test) {
  var user_name = 'spongebob2';
  var password = 'password';

  var p_client = new Db(MONGODB, new Server("127.0.0.1", 27017, {auto_reconnect: true, poolSize:3, ssl:useSSL}), {native_parser: (process.env['TEST_NATIVE'] != null)});
  p_client.open(function(err, automatic_connect_client) {
    p_client.authenticate('admin', 'admin', function(err, replies) {
      test.ok(err instanceof Error);
      // Add a user
      p_client.addUser(user_name, password, function(err, result) {
        // Execute authentication
        p_client.authenticate(user_name, password, function(err, replies) {
          test.ok(err == null);
                  
          // Kill a connection to force a reconnect
          p_client.serverConfig.close();
                
          p_client.createCollection('shouldCorrectlyReAuthorizeReconnectedConnections', function(err, collection) {
            collection.insert({a:1}, {safe:true}, function(err, r) {
              collection.insert({a:2}, {safe:true}, function(err, r) {
                collection.insert({a:3}, {safe:true}, function(err, r) {                                        
                  collection.count(function(err, count) {
                    test.equal(3, count);
                    p_client.close();
                    test.done();
                  })
                })
              })
            })
          });
        });            
      });
    });
  });    
}

exports.shouldCorrectlyAddAndRemoveUser = function(test) {
  var user_name = 'spongebob2';
  var password = 'password';

  var p_client = new Db(MONGODB, new Server("127.0.0.1", 27017, {auto_reconnect: true, ssl:useSSL}), {native_parser: (process.env['TEST_NATIVE'] != null)});
  p_client.open(function(err, automatic_connect_client) {
    p_client.authenticate('admin', 'admin', function(err, replies) {
      test.ok(err instanceof Error);

      // Add a user
      p_client.addUser(user_name, password, function(err, result) {
        p_client.authenticate(user_name, password, function(err, replies) {
          test.ok(replies);

          // Remove the user and try to authenticate again
          p_client.removeUser(user_name, function(err, result) {
            p_client.authenticate(user_name, password, function(err, replies) {
              test.ok(err instanceof Error);

              test.done();
              p_client.close();
            });
          });
        });
      });
    });
  });    
}

// run this last
exports.noGlobalsLeaked = function(test) {
  var leaks = gleak.detectNew();
  test.equal(0, leaks.length, "global var leak detected: " + leaks.join(', '));
  test.done();
}

/**
 * Retrieve the server information for the current
 * instance of the db client
 * 
 * @ignore
 */
exports.noGlobalsLeaked = function(test) {
  var leaks = gleak.detectNew();
  test.equal(0, leaks.length, "global var leak detected: " + leaks.join(', '));
  test.done();
}

/**
 * Retrieve the server information for the current
 * instance of the db client
 * 
 * @ignore
 */
var numberOfTestsRun = Object.keys(this).length - 2;