函数调用上的Javascript模拟第三方节点库

Question

我正在使用sinon，我想做一些非常简单的事情，经过大量的搜索，我找不到任何解决方案。

我的类MessageService类如下所示：

import sb from "@azure/service-bus";

    export class MessageService {

        async handleNewMessage() {

            //some code ....

            await sb.delay(5000);

           //some code ....
        }

    }

我想测试handleNewMessage方法，但是里面有对第三方库sb.delay(5000)的调用(这实际上是睡眠5s)。

我想替换这个函数，使它成为，不做任何。

在我的messageService.spec.js文件中，我试图导入并替换它，但是它不能工作。

import sb from "@azure/service-bus";

describe('message service', () => {

 beforeEach(() => {
        sinon.stub(sb, 'delay').returns(()=>{});
    });

 afterEach(() => {
        sinon.restore();
    });

it("handleNewMessage should work", async () => {

        //This is call to the real sb.delay function
        await messageServiceMock.handleNewMessage();

    });

})

有没有办法取代原来的第三方函数调用？

Answer 1

从这个评论：https://github.com/sinonjs/sinon/issues/1711#issuecomment-369220753

我们不允许修改使用导入导入的模块。看起来进口是关于出口的只读视图。

因此，sinon.stub无法修改导入的模块。

看看sb.delay实现：

export function delay<T>(t: number, value?: T): Promise<T> {
  return new Promise(resolve => setTimeout(() => resolve(value), t));
}
//...
export { delay }

以下是解决办法：

MessageService.ts

import * as sb from "@azure/service-bus";

export class MessageService {
  async handleNewMessage() {
    await sb.delay(5000);
  }
}

MessageService.test.ts

import sinon from "sinon";
import proxyquire from "proxyquire";

describe("message service", () => {
  afterEach(() => {
    sinon.restore();
  });

  it("handleNewMessage should work", async () => {
    const delayStub = sinon.stub().resolves({});
    const { MessageService } = proxyquire("./MessageService.ts", {
      "@azure/service-bus": {
        delay: delayStub,
      },
    });
    const messageServiceMock = new MessageService();
    await messageServiceMock.handleNewMessage();
    sinon.assert.calledWith(delayStub, 5000);
  });
});

100%覆盖范围的单元测试结果：

 message service
    ✓ handleNewMessage should work (1343ms)


  1 passing (1s)

------------------------|----------|----------|----------|----------|-------------------|
File                    |  % Stmts | % Branch |  % Funcs |  % Lines | Uncovered Line #s |
------------------------|----------|----------|----------|----------|-------------------|
All files               |      100 |      100 |      100 |      100 |                   |
 MessageService.test.ts |      100 |      100 |      100 |      100 |                   |
 MessageService.ts      |      100 |      100 |      100 |      100 |                   |
------------------------|----------|----------|----------|----------|-------------------|

源代码：https://github.com/mrdulin/mocha-chai-sinon-codelab/tree/master/src/stackoverflow/59234255