thinkjs/test/Lib/Behavior/DenyIpBehavior.js

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

global.APP_PATH = path.normalize(__dirname + '/../../App');
global.RESOURCE_PATH = path.normalize(__dirname + '/../../www')
process.execArgv.push('--no-app');
require(path.normalize(__dirname + '/../../../index.js'));


var Http = thinkRequire('Http');
var http = require('http');
var req = new http.IncomingMessage();
req.headers = { 
  'x-real-ip': '127.0.0.1',
  'x-forwarded-for': '127.0.0.1',
  'host': 'meinv.ueapp.com',
  'x-nginx-proxy': 'true',
  'connection': 'close',
  'cache-control': 'max-age=0',
  'accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8',
  'user-agent': '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',
  'accept-encoding': 'gzip,deflate,sdch',
  'accept-language': 'zh-CN,zh;q=0.8,en;q=0.6,ja;q=0.4,nl;q=0.2,zh-TW;q=0.2'
};
req.method = 'GET';
req.httpVersion = '1.1';
req.url = '/index/index/name/welefen?test=welefen&value=1111';
var res = new http.ServerResponse(req);
var instance = Http(req, res).run();

describe('DenyIpBehavior', function(){
  var httpInstance = null;
  var promise = function(){
    return instance.then(function(http){
      httpInstance = http;
      return B('DenyIp', http);
    })
  }
  it('empty', function(done){
    C('deny_ip', []);
    promise().then(function(data){
      assert.equal(data, true)
      done();
    })
  })
  it('local ip', function(done){
    C('deny_ip', ['127.0.0.100']);
    promise().then(function(data){
      assert.equal(data, true)
      done();
    })
  })
  it('local ip hit', function(done){
    C('deny_ip', ['127.0.0.1']);
    httpInstance.res.end = function(){
      done();
    }
    promise();
  })
  it('local ip star', function(done){
    C('deny_ip', ['127.0.*']);
    httpInstance.res.end = function(){
      done();
    }
    promise();
  })
  it('local ip star, not hit', function(done){
    C('deny_ip', ['127.1.*']);
    promise().then(function(data){
      assert.equal(data, true);
      done();
    })
  })
})