Sinon存根不替换功能

Question

我正在尝试使用sinon存根替换一个可能需要很长时间的函数。但是当我运行测试时，测试代码似乎没有使用sinon存根。

这是我要测试的代码。

function takeTooLong() {
    return  returnSomething();
}

function returnSomething() {
    return new Promise((resolve) => {
        setTimeout(() => {
          resolve('ok')
        }, 1500)
    })
}

module.exports = {
  takeTooLong,
  returnSomething
}

这是测试代码。

const chai = require('chai')
chai.use(require('chai-string'))
chai.use(require('chai-as-promised'))
const expect = chai.expect
chai.should()
const db = require('./database')
const sinon = require('sinon')
require('sinon-as-promised')

describe('Mock the DB connection', function () {

it('should use stubs for db connection for takeTooLong', function (done) {

    const stubbed = sinon.stub(db, 'returnSomething').returns(new Promise((res) => res('kk')));
    const result = db.takeTooLong()

    result.then((res) => {

        expect(res).to.equal('kk')
        sinon.assert.calledOnce(stubbed);
        stubbed.restore()
        done()
    }).catch((err) => done(err))

})

我得到一个断言错误

 AssertionError: expected 'ok' to equal 'kk'
      + expected - actual

  -ok
  +kk

我做错了什么？为什么不使用存根？摩卡的测试框架。

Answer 1

您可能想看看直接导出的stub/spy函数的Proxyquire。https://www.npmjs.com/package/proxyquire/

Answer 2

Sinon存根对象的属性，而不是函数本身。

在本例中，您是在对象中导出该函数。

module.exports = {
  takeTooLong,
  returnSomething
}

因此，为了从对象中正确调用函数，您需要将函数调用替换为对导出对象的引用，例如：

function takeTooLong() {
    return module.exports.returnSomething();
}

当然，根据您的代码，您可以随时对其进行重构：

var exports = module.exports = {

    takeTooLong: function() { return exports.returnSomething() }

    returnSomething: function() { /* .. */ }

}