thinkjs/test/controller/base.js

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 _http = require('../_http.js');

function getHttp(config, options){
  think.APP_PATH = path.dirname(__dirname) + think.sep + 'testApp';
  var instance = _http.createReqRes();
  var req = instance.req;
  var res = instance.res;
  return think.http(req, res).then(function(http){
    if(config){
      http._config = config;
    }
    if(options){
      for(var key in options){
        http[key] = options[key];
      }
    }
    return http;
  })
}
var Controller = think.lookClass('', 'controller');

function getInstance(options){
  return getHttp().then(function(http){
    if(options){
      for(var key in options){
        http[key] = options[key];
      }
    }
    return new Controller(http);
  })
}

describe('controller/base.js', function(){
  it('ip', function(done){
    getInstance().then(function(instance){
      var ip = instance.ip();
      assert.equal(ip, '127.0.0.1');
      done();
    })
  })
  it('view', function(done){
    getInstance().then(function(instance){
      var view = instance.view();
      assert.equal(view instanceof think.require('view'), true);
      done();
    })
  })
  it('method', function(done){
    getInstance().then(function(instance){
      var method = instance.method();
      assert.equal(method, 'get');
      done();
    })
  })
  it('isGet', function(done){
    getInstance().then(function(instance){
      var data = instance.isGet();
      assert.equal(data, true);
      done();
    })
  })
  it('isPost', function(done){
    getInstance().then(function(instance){
      var data = instance.isPost();
      assert.equal(data, false);
      done();
    })
  })
  it('isMethod', function(done){
    getInstance().then(function(instance){
      var data = instance.isMethod('get');
      assert.equal(data, true);
      done();
    })
  })
  it('isAjax', function(done){
    getInstance().then(function(instance){
      var data = instance.isAjax();
      assert.equal(data, false);
      done();
    })
  })
  it('isWebSocket', function(done){
    getInstance().then(function(instance){
      var data = instance.isWebSocket();
      assert.equal(data, false);
      done();
    })
  })
  it('isCli', function(done){
    think.cli = false;
    getInstance().then(function(instance){
      var data = instance.isCli();
      assert.equal(data, false);
      done();
    })
  })
  it('isJsonp', function(done){
    getInstance().then(function(instance){
      var data = instance.isJsonp();
      assert.equal(data, false);
      done();
    })
  })
  it('get all get data', function(done){
    getInstance().then(function(instance){
      var data = instance.get();
      assert.deepEqual(data, { test: 'welefen', value: '1111' });
      done();
    })
  })
  it('get item get data', function(done){
    getInstance().then(function(instance){
      var data = instance.get('test');
      assert.deepEqual(data, 'welefen');
      done();
    })
  })
  it('get item get data, empty', function(done){
    getInstance().then(function(instance){
      var data = instance.get('test111');
      assert.deepEqual(data, '');
      done();
    })
  })
  it('get all post data', function(done){
    getInstance().then(function(instance){
      var data = instance.post();
      assert.deepEqual(data, { });
      done();
    })
  })
  it('get post data 2', function(done){
    getInstance().then(function(instance){
      instance.http._post.key2 = ['EXP', 2]
      var data = instance.post('key2');
      delete instance.http._post.key2;
      assert.deepEqual(data, ['EXP ', 2]);
      done();
    })
  })
  it('get post data 3', function(done){
    getInstance().then(function(instance){
      instance.http._post.key2 = ['EXP2', 2]
      var data = instance.post('key2');
      delete instance.http._post.key2;
      assert.deepEqual(data, ['EXP2', 2]);
      done();
    })
  })
  it('get post data 4', function(done){
    getInstance().then(function(instance){
      instance.http._post.key2 = {EXP: 2}
      var data = instance.post('key2');
      delete instance.http._post.key2;
      assert.deepEqual(data, {'EXP ': 2});
      done();
    })
  })
  it('get post data 4.1', function(done){
    getInstance().then(function(instance){
      instance.http._post.key2 = {exp: 2}
      var data = instance.post('key2');
      delete instance.http._post.key2;
      assert.deepEqual(data, {'exp ': 2});
      done();
    })
  })
  it('get post data 5', function(done){
    getInstance().then(function(instance){
      instance.http._post.key2 = {EXP2: 2}
      var data = instance.post('key2');
      delete instance.http._post.key2;
      assert.deepEqual(data, {'EXP2': 2});
      done();
    })
  })


  it('get all param data', function(done){
    getInstance().then(function(instance){
      var data = instance.param();
      assert.deepEqual(data, { test: 'welefen', value: '1111' });
      done();
    })
  })
  it('get all file data', function(done){
    getInstance().then(function(instance){
      var data = instance.file();
      assert.deepEqual(data, {});
      done();
    })
  })
  it('get all header data', function(done){
    getInstance().then(function(instance){
      var data = instance.header();
      data = Object.keys(data).sort();
      assert.deepEqual(data, [ 'accept', 'accept-encoding', 'accept-language', 'cache-control', 'connection', 'host', 'user-agent', 'x-forwarded-for', 'x-nginx-proxy', 'x-real-ip' ]);
      done();
    })
  })
  it('get usrAgent', function(done){
    getInstance().then(function(instance){
      var data = instance.userAgent();
      assert.deepEqual(data, 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_0) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/35.0.1916.153 Safari/537.36')
      done();
    })
  })
  it('get referrer', function(done){
    getInstance().then(function(instance){
      var data = instance.referrer();
      assert.deepEqual(data, '')
      done();
    })
  })
  it('get referer', function(done){
    getInstance().then(function(instance){
      var data = instance.referer();
      assert.deepEqual(data, '')
      done();
    })
  })
  it('get cookie', function(done){
    getInstance().then(function(instance){
      var data = instance.cookie();
      assert.deepEqual(data, {})
      done();
    })
  })
  it('send time', function(done){
    muk(think, 'log', function(){})
    getInstance().then(function(instance){
      var data = instance.sendTime();
      muk.restore();
      done();
    })
  })
  it('error', function(done){
    muk(think, 'log', function(){})
    getInstance().then(function(instance){
      instance.error().catch(function(err){
        assert.equal(think.isPrevent(err), true);
        muk.restore();
        done();
      });
      
    })
  })
  it('fail', function(done){
    muk(think, 'log', function(){})
    getInstance().then(function(instance){
      var data = instance.fail().catch(function(err){
        assert.equal(think.isPrevent(err), true);
        muk.restore();
        done();
      });
    })
  })
  it('success', function(done){
    muk(think, 'log', function(){})
    getInstance().then(function(instance){
      var data = instance.success().catch(function(err){
        assert.equal(think.isPrevent(err), true);
        muk.restore();
        done();
      });
    })
  })
  it('type', function(done){
    muk(think, 'log', function(){})
    getInstance().then(function(instance){
      var data = instance.type();
      muk.restore();
      done();
    })
  })
  it('end', function(done){
    muk(think, 'log', function(){})
    getInstance().then(function(instance){
      var data = instance.end().catch(function(err){
        assert.equal(think.isPrevent(err), true);
        muk.restore();
        done();
      });
    })
  })
  it('send', function(done){
    muk(think, 'log', function(){})
    getInstance().then(function(instance){
      var data = instance.send().catch(function(err){
        assert.equal(think.isPrevent(err), true);
        muk.restore();
        done();
      });
    })
  })
  it('expires', function(done){
    muk(think, 'log', function(){})
    getInstance().then(function(instance){
      var flag = false;
      instance.http.expires = function(){
        flag = true;
      }
      var data = instance.expires();
      muk.restore();
      assert.equal(flag, true)
      done();
    })
  })
  it('write', function(done){
    muk(think, 'log', function(){})
    getInstance().then(function(instance){
      var data = instance.write();
      muk.restore();
      done();
    })
  })
  it('deny', function(done){
    muk(think, 'log', function(){})
    getInstance().then(function(instance){
      var data = instance.deny().catch(function(err){
        assert.equal(think.isPrevent(err), true);
        muk.restore();
        done();
      });
    })
  })
  it('deny 404', function(done){
    muk(think, 'log', function(){})
    getInstance({
      status: function(status){
        assert.equal(status, 404);
      }
    }).then(function(instance){
      instance.deny(404).catch(function(err){
        assert.equal(think.isPrevent(err), true);
        muk.restore();
        done();
      });
    })
  })
  it('status', function(done){
    muk(think, 'log', function(){})
    getInstance().then(function(instance){
      var data = instance.status();
      assert.equal(data, instance);
      muk.restore();
      done();
    })
  })
  it('json', function(done){
    muk(think, 'log', function(){})
    getInstance().then(function(instance){
      var data = instance.json().catch(function(err){
        assert.equal(think.isPrevent(err), true);
        muk.restore();
        done();
      });
    })
  })
  it('jsonp', function(done){
    muk(think, 'log', function(){})
    getInstance().then(function(instance){
      var data = instance.jsonp().catch(function(err){
        assert.equal(think.isPrevent(err), true);
        muk.restore();
        done();
      });
    })
  })
  it('redirect', function(done){
    muk(think, 'log', function(){})
    getInstance().then(function(instance){
      var data = instance.redirect().catch(function(err){
        assert.equal(think.isPrevent(err), true);
        muk.restore();
        done();
      });
    })
  })
  it('lang use cookie', function(done){
    getInstance().then(function(instance){
      var data = instance.lang();
      assert.deepEqual(data, 'zh-cn')
      done();
    }).catch(function(err){
      console.log(err.stack)
    })
  })
  it('lang use cookie 2', function(done){
    getInstance({
      _cookie: {
        think_locale: 'en'
      }
    }).then(function(instance){
      var data = instance.lang();
      assert.deepEqual(data, 'en')
      done();
    })
  })
  it('locale, not exist', function(done){
    getInstance().then(function(instance){
      var data = instance.locale('welefen');
      assert.deepEqual(data, 'welefen')
      done();
    })
  })
  it('locale', function(done){
    getInstance({
      _config: think.config()
    }).then(function(instance){
      var data = instance.locale('CONTROLLER_NOT_FOUND');
      assert.deepEqual(data, 'controller `%s` not found. url is `%s`.')
      done();
    })
  })
  it('locale with data', function(done){
    getInstance({
      _config: think.config()
    }).then(function(instance){
      var data = instance.locale('CONTROLLER_NOT_FOUND', 'test');
      assert.deepEqual(data, 'controller `test` not found. url is `%s`.')
      done();
    })
  })
  it('locale', function(done){
    getInstance({
      _cookie: {
        think_locale: 'dddd'
      },
      _config: think.config()
    }).then(function(instance){
      var data = instance.locale('CONTROLLER_NOT_FOUND');
      assert.deepEqual(data, 'controller `%s` not found. url is `%s`.')
      done();
    })
  })
  it('fetch', function(done){
    getInstance({
    }).then(function(instance){
      return instance.fetch(__filename)
    }).then(function(content){
      assert.equal(content.indexOf('describe') > -1, true);
      done();
    })
  })
  it('display', function(done){
    muk(think, 'log', function(){})
    getInstance({
    }).then(function(instance){
      return instance.display(__filename)
    }).catch(function(err){
      assert.equal(think.isPrevent(err), true);
      muk.restore();
      done();
    })
  })
  it('render', function(done){
    muk(think, 'log', function(){})
    getInstance({
    }).then(function(instance){
      return instance.render(__filename)
    }).catch(function(err){
      assert.equal(think.isPrevent(err), true);
      muk.restore();
      done();
    })
  })
  it('download', function(done){
    getInstance({
      header: function(type, value){
        if(type === 'Content-Type'){
          assert.equal(value, 'application/javascript')
        }else if(type === 'Content-Disposition'){
          assert.equal(value, 'attachment; filename=base.js')
        }
      },
      end: function(){}
    }).then(function(instance){
      return instance.download(__filename)
    }).catch(function(err){
      assert.equal(think.isPrevent(err), true);
      done();
    })
  })
  it('download, with content-type', function(done){
    getInstance({
      header: function(type, value){
        //console.log(type, value)
        if(type === 'Content-Type'){
          assert.equal(value, 'text/html')
        }else if(type === 'Content-Disposition'){
          assert.equal(value, 'attachment; filename=base.js')
        }
        this._contentTypeIsSend = true;
      },
      end: function(){}
    }).then(function(instance){
      return instance.download(__filename, 'text/html')
    }).catch(function(err){
      assert.equal(think.isPrevent(err), true);
      done();
    })
  })
  it('download, with content-type, with filename', function(done){
    getInstance({
      header: function(type, value){
        if(type === 'Content-Type'){
          assert.equal(value, 'text/html')
        }else if(type === 'Content-Disposition'){
          assert.equal(value, 'attachment; filename=a.html')
        }
        this._contentTypeIsSend = true;
      },
      end: function(){}
    }).then(function(instance){
      return instance.download(__filename, 'text/html', 'a.html')
    }).catch(function(err){
      assert.equal(think.isPrevent(err), true);
      done();
    })
  })
  it('download,  with filename', function(done){
    getInstance({
      header: function(type, value){
        if(type === 'Content-Type'){
          assert.equal(value, 'application/javascript')
        }else if(type === 'Content-Disposition'){
          assert.equal(value, 'attachment; filename=a.html')
        }
        this._contentTypeIsSend = true;
      },
      end: function(){}
    }).then(function(instance){
      return instance.download(__filename, 'a.html')
    }).catch(function(err){
      assert.equal(think.isPrevent(err), true);
      done();
    })
  })
  it('get session', function(done){
    getInstance({
    }).then(function(instance){
      return instance.session('welefen')
    }).then(function(data){
      assert.equal(data, undefined)
      done();
    })
  })
  it('set session', function(done){
    var ins;
    getInstance({
    }).then(function(instance){
      ins = instance;
      return ins.session('welefen', 'suredy')
    }).then(function(){
      return ins.session('welefen');
    }).then(function(data){
      assert.equal(data, 'suredy')
      done();
    })
  })
  it('delete session', function(done){
    var ins;
    getInstance({
    }).then(function(instance){
      ins = instance;
      return ins.session('welefen', 'suredy')
    }).then(function(){
      return ins.session();
    }).then(function(){
      return ins.session('welefen');
    }).then(function(data){
      assert.equal(data, undefined)
      done();
    })
  })
  it('emit error', function(done){
    var ins;
    getInstance({
    }).then(function(instance){
      try{
        instance.emit('event', 'data');
        assert.equal(1, 2)
      }catch(e){}
      done();
    })
  })
  it('emit correct', function(done){
    var ins;
    getInstance({
    }).then(function(instance){
      var flag = false;
      instance.http.socketEmit = function(event, data){
        assert.equal(event, 'event');
        assert.equal(data, 'data');
        flag = true;
      }
      instance.http.socket = {}
      try{
        instance.emit('event', 'data');
      }catch(e){ 
        assert.equal(1, 2)
      }
      assert.equal(flag, true);
      done();
    })
  })
  it('broadcast error', function(done){
    var ins;
    getInstance({
    }).then(function(instance){
      try{
        instance.broadcast('event', 'data');
        assert.equal(1, 2)
      }catch(e){}
      done();
    })
  })
  it('broadcast correct', function(done){
    var ins;
    getInstance({
    }).then(function(instance){
      var flag = false;
      instance.http.socketBroadcast = function(event, data){
        assert.equal(event, 'event');
        assert.equal(data, 'data');
        flag = true;
      }
      instance.http.socket = {}
      try{
        instance.broadcast('event', 'data');
      }catch(e){ 
        assert.equal(1, 2)
      }
      assert.equal(flag, true);
      done();
    })
  })

})