thinkjs/test/adapter/socket/mongo.js

'use strict';

var assert = require('assert');
var path = require('path');
var fs = require('fs');
var muk = require('muk');




for(var filepath in require.cache){
  delete require.cache[filepath];
}
var Index = require('../../../lib/index.js');
var instance = new Index();
instance.load();

think.APP_PATH = path.dirname(__dirname) + think.sep + 'testApp';

var mongoSocket = think.adapter('socket', 'mongo');

describe('adapter/socket/mongo', function(){
  it('get instance', function(){
    var instance = new mongoSocket();
    assert.deepEqual(instance.config, {host: '127.0.0.1', port: 27017});
  })
  it('get connection', function(done){
    var npm = think.npm;
    think.npm = function(){
      return {
        MongoClient: {
          connect: function(url, config, callback){
            assert.equal(url, 'mongodb://127.0.0.1:27017/test');
            assert.deepEqual(config, { host: '127.0.0.1', port: 27017, database: 'test' });
            callback && callback(null, {on: function(){}});
          }
        }
      }
    }
    var instance = new mongoSocket({
      database: 'test'
    });
    instance.getConnection().then(function(connection){
      think.npm = npm;

      done();
    })
  })
  it('get connection, user', function(done){
    var npm = think.npm;
    think.npm = function(){
      return {
        MongoClient: {
          connect: function(url, config, callback){
            assert.equal(url, 'mongodb://welefen:suredy@127.0.0.1:27017/test');
            assert.deepEqual(config, { host: '127.0.0.1', port: 27017, user: 'welefen', password: 'suredy', database: 'test' });
            callback && callback(null, {on: function(){}});
          }
        }
      }
    }
    var instance = new mongoSocket({
      user: 'welefen',
      password: 'suredy',
      database: 'test'
    });
    instance.getConnection().then(function(connection){
      think.npm = npm;

      done();
    })
  })
  it('get connection, options', function(done){
    var npm = think.npm;
    think.npm = function(){
      return {
        MongoClient: {
          connect: function(url, config, callback){
            assert.equal(url, 'mongodb://welefen:suredy@127.0.0.1:27017/test?slaveOk=true');
            //assert.deepEqual(config, { host: '127.0.0.1', port: 27017, user: 'welefen', password: 'suredy', database: 'test' });
            callback && callback(null, {on: function(){}});
          }
        }
      }
    }
    var instance = new mongoSocket({
      user: 'welefen',
      password: 'suredy',
      database: 'test',
      options: {
        slaveOk: true
      }
    });
    instance.getConnection().then(function(connection){
      think.npm = npm;

      done();
    })
  })
  it('get connection, change host', function(done){
    var npm = think.npm;
    think.npm = function(){
      return {
        MongoClient: {
          connect: function(url, config, callback){
            assert.equal(url, 'mongodb://welefen:suredy@welefen.com:1234/test?slaveOk=true');
            //assert.deepEqual(config, { host: '127.0.0.1', port: 27017, user: 'welefen', password: 'suredy', database: 'test' });
            callback && callback(null, {on: function(){}});
          }
        }
      }
    }
    var instance = new mongoSocket({
      host: 'welefen.com',
      port: 1234,
      user: 'welefen',
      password: 'suredy',
      database: 'test',
      options: {
        slaveOk: true
      }
    });
    instance.getConnection().then(function(connection){
      think.npm = npm;

      done();
    })
  })
  it('get connection, many host', function(done){
    var npm = think.npm;
    think.npm = function(){
      return {
        MongoClient: {
          connect: function(url, config, callback){
            assert.equal(url, 'mongodb://welefen:suredy@welefen.com:6350,suredy.com:1234/test?slaveOk=true');
            //assert.deepEqual(config, { host: '127.0.0.1', port: 27017, user: 'welefen', password: 'suredy', database: 'test' });
            callback && callback(null, {on: function(){}});
          }
        }
      }
    }
    var instance = new mongoSocket({
      host: ['welefen.com', 'suredy.com'],
      port: [6350, 1234],
      user: 'welefen',
      password: 'suredy',
      database: 'test',
      options: {
        slaveOk: true
      }
    });
    instance.getConnection().then(function(connection){
      think.npm = npm;

      done();
    })
  })
  it('get connection, log level', function(done){
    var npm = think.npm;
    think.npm = function(){
      return {
        MongoClient: {
          connect: function(url, config, callback){
            assert.equal(url, 'mongodb://welefen:suredy@127.0.0.1:27017/test?slaveOk=true');
            callback && callback(null, {on: function(){}});
          }
        },
        Logger: {
          setLevel: function(level){
            assert.equal(level, 'welefen')
          }
        }
      }
    }
    var instance = new mongoSocket({
      user: 'welefen',
      password: 'suredy',
      database: 'test',
      log_level: 'welefen',
      options: {
        slaveOk: true
      }
    });
    instance.getConnection().then(function(connection){
      think.npm = npm;

      done();
    })
  })
  it('get connection, error', function(done){
    var npm = think.npm;
    var reject = think.reject;
    think.npm = function(){
      return {
        MongoClient: {
          connect: function(url, config, callback){
            assert.equal(url, 'mongodb://welefen:suredy@127.0.0.1:27017/test?slaveOk=true');
            callback && callback(new Error('welefen'));
          }
        }
      }
    };
    think.reject = function(err){
      return Promise.reject(err);
    };
    var instance = new mongoSocket({
      user: 'welefen',
      password: 'suredy',
      database: 'test',
      log_level: 'welefen',
      options: {
        slaveOk: true
      }
    });
    instance.getConnection().catch(function(err){
      assert.equal(err.message, 'welefen')
      think.npm = npm;
      think.reject = reject;
      done();
    })
  })
  it('get connection, error 2', function(done){
    var npm = think.npm;
    var reject = think.reject;
    think.npm = function(){
      return {
        MongoClient: {
          connect: function(url, config, callback){
            assert.equal(url, 'mongodb://welefen:suredy@127.0.0.1:27017/test?slaveOk=true');
            callback && callback(new Error('EADDRNOTAVAIL'));
          }
        }
      }
    };
    think.reject = function(err){
      return Promise.reject(err);
    };
    var instance = new mongoSocket({
      user: 'welefen',
      password: 'suredy',
      database: 'test',
      log_level: 'welefen',
      options: {
        slaveOk: true
      }
    });
    instance.getConnection().catch(function(err){
      assert.equal(err.message, 'Address not available, mongodb://welefen:suredy@127.0.0.1:27017/test?slaveOk=true. http://www.thinkjs.org/doc/error_message.html#eaddrnotavail')
      think.npm = npm;
      think.reject = reject;
      done();
    })
  })
  it('get connection, exist', function(done){
    var npm = think.npm;
    think.npm = function(){
      return {
        MongoClient: {
          connect: function(url, config, callback){
            assert.equal(url, 'mongodb://welefen:suredy@127.0.0.1:27017/test?slaveOk=true');
            callback && callback(null, {on: function(){}});
          }
        }
      }
    }
    var instance = new mongoSocket({
      user: 'welefen',
      password: 'suredy',
      database: 'test',
      options: {
        slaveOk: true
      }
    });
    instance.getConnection().then(function(connection){
      return instance.getConnection();
    }).then(function(connection){
      //assert.deepEqual(connection, {})
      think.npm = npm;
      done();
    })
  })
  it('close, connection not exist', function(done){
    var instance = new mongoSocket({
      user: 'welefen',
      password: 'suredy',
      database: 'test',
      options: {
        slaveOk: true
      }
    });
    instance.close();
    done();
  })
  it('get connection, exist', function(done){
    var npm = think.npm;
    think.npm = function(){
      return {
        MongoClient: {
          connect: function(url, config, callback){
            assert.equal(url, 'mongodb://welefen:suredy@127.0.0.1:27017/test?slaveOk=true');
            callback && callback(null, {
              on: function(){},
              close: function(){

              }
            });
          }
        }
      }
    }
    var instance = new mongoSocket({
      user: 'welefen',
      password: 'suredy',
      database: 'test',
      options: {
        slaveOk: true
      }
    });
    instance.getConnection().then(function(connection){
      instance.close();
      think.npm = npm;
      done();
    })
  })
})