如何使用重新连接模拟请求模块?

时间:2016-07-23 11:06:03

标签: node.js unit-testing request mocking

我有一个自定义模块 tokens.js ,其功能通过 npm请求模块发出请求。它看起来像这样:

'use strict';

let request = require('request');

module.exports.getToken = function(code, cb) {
  let url = 'some_url';

  request(url, function (err, response, body) {
    if (err) {
      return cb(err);
    } else if (response.statusCode !== 200) {
      return cb('err');
    }

    parseGetResponse(body, function (err, token) {
      if (err) {
        return cb(err);
      }

      return cb(null, token);
  });
});

我想编写单元测试来涵盖 getToken()函数,但我在使用模拟请求模块时遇到了一些麻烦。

我试过了:

let mocha = require('mocha');
let rewire = require('rewire');
let should = require('should');

let requestMock = {
  request: function (url, cb) {
      // return cb(); etc
  }
}

let tokens = rewire('services/tokens.js');
tokens.__set__('request', requestMock);

但这种方法不起作用:

TypeError: request is not a function
  at Object.module.exports.getToken (services/tokens.js)

实际上,它引出了另一个问题: 请求模块如何在不直接调用导出函数(request.request(url,cb))的情况下工作?我该如何使用重新布线?

2 个答案:

答案 0 :(得分:1)

另一种选择是重构代码,以便轻松配置其依赖关系。

function TokenGetter(request) {
   this.request = request || require('request');
   this.getToken = function(code, cb) { ...
}
module.exports.TokenGetter = TokenGetter;

// production code
var tokenGetter = new TokenGetter();

测试代码

// test code can configure a mock request for your test, no 
// 3rd party libraries
// configure mockObject with assertions/return values
var mockRequest = function(url, cb) { .... ;

var testTokenGetter = new TokenGetter(mockRequest);

有许多策略可以使您的类可配置,并允许有机会注入虚假对象进行测试。我相信努力编写可测试性代码应该可以消除为项目添加更多第三方依赖项的需要。

答案 1 :(得分:0)

实际上,我找到了一种模拟请求模块的方法。我只是让requestMock成为一个函数。另外我发现在请求模块中有一条线,这使得它成为可能: module.exports = request