Stubbing Date.now()和Math.random()

时间:2013-06-09 16:36:26

标签: javascript node.js unit-testing dependency-injection sinon

我正在使用MochaSinon对我的node.js模块进行单元测试。我已经成功地模拟了其他依赖项(我编写过的其他模块),但是我遇到了存在非纯函数的问题(例如Math.random()Date.now())。我已经尝试了以下(简化以便这个问题没有如此本地化),但由于明显的范围问题,Math.random()没有被删除。 Math的实例在测试文件和mymodule.js之间是独立的。

test.js

var sinon    = require('sinon'),
    mymodule = require('./mymodule.js'),
    other    = require('./other.js');

describe('MyModule', function() {
    describe('funcThatDependsOnRandom', function() {
        it('should call other.otherFunc with a random num when no num provided', function() {
            sinon.mock(other).expects('otherFunc').withArgs(0.5).once();
            sinon.stub(Math, 'random').returns(0.5);

            funcThatDependsOnRandom(); // called with no args, so should call
                                       // other.otherFunc with random num

            other.verify(); // ensure expectation has been met
        });
    });
});

所以在这个人为的例子中,functThatDependsOnRandom()看起来像是:

mymodule.js

var other = require('./other.js');

function funcThatDependsOnRandom(num) {
    if(typeof num === 'undefined') num = Math.random();

    return other.otherFunc(num);
}

使用Sinon可以在此场景中存根Math.random()吗?

3 个答案:

答案 0 :(得分:8)

是的,这是一个老问题,但它是有效的。这是一个有效的答案,但我很想听听如何让它变得更好的建议。

我在浏览器中处理此问题的方法是创建一个代理对象。例如,您无法在浏览器中存根窗口对象,因此您可以创建名为windowProxy的代理对象。当您想要获取位置时,您在windowProxy中创建一个名为location的方法,该方法返回或设置windowLocation。然后,在测试时,您模拟windowProxy.location。

你可以用Node.js做同样的事情,但它不能简单地工作。简单的版本是一个模块不能弄乱另一个模块的私有命名空间。

解决方案是使用mockery模块。在初始化mockery之后,如果使用与mock告诉mock相匹配的参数调用require(),它将允许您覆盖require语句并返回自己的属性。

更新:我已经创建了一个功能齐全的代码示例。它在Github at newz2000/dice-tddavailable via npm上。 / END UPDATE

文档非常好,所以我建议阅读它们,但这是一个例子:

使用以下内容创建文件randomHelper.js

module.exports.random = function() {
  return Math.random();
}

然后在需要随机数的代码中,您:

var randomHelper = require('./randomHelper');

console.log('A random number: ' + randomHelper.random() );

一切都应该像平常一样工作。您的代理对象的行为与Math.random相同。

请务必注意,require语句接受单个参数'./randomHelper'。我们需要注意的是。

现在在你的测试中,(我正在使用mocha和chai):

var sinon = require('sinon');
var mockery = require('mockery')
var yourModule; // note that we didn't require() your module, we just declare it here

describe('Testing my module', function() {

  var randomStub; // just declaring this for now

  before(function() {
    mockery.enable({
      warnOnReplace: false,
      warnOnUnregistered: false
    });

    randomStub = sinon.stub().returns(0.99999);

    mockery.registerMock('./randomHelper', randomStub)
    // note that I used the same parameter that I sent in to requirein the module
    // it is important that these match precisely

    yourmodule = require('../yourmodule');
    // note that we're requiring your module here, after mockery is setup
  }

  after(function() {
    mockery.disable();
  }

  it('Should use a random number', function() {
    callCount = randomStub.callCount;

    yourmodule.whatever(); // this is the code that will use Math.random()

    expect(randomStub.callCount).to.equal(callCount + 1);
  }
}

就是这样。在这种情况下,我们的存根将始终返回0.0.99999;你当然可以改变它。

答案 1 :(得分:0)

你确定没有嘲笑Math是问题。似乎这条线没有多大意义:

sinon.mock(other).expects('otherFunc').withArgs(0.5).once();

你在一个模块中模拟others但在另一个模块中使用它。我不认为你会在mymodule.js中获得模拟版本。另一方面,存在Math.random应该有效,因为这对所有模块都是全局的。

另请参阅此SO以了解nodeJS测试中的依赖项。

答案 2 :(得分:0)

使用Fake timers很容易将Date.now()sinon存根:

  

伪定时器提供时钟对象来传递时间,也可用于控制通过新的Date()创建的Date对象;或Date.now(); (如果浏览器支持)。

// Arrange
const now = new Date();
const clock = sinon.useFakeTimers(now.getTime());

// Act
// Call you function ...

// Assert
// Make some assertions ...

// Teardown
clock.restore();