监视函数sinon返回的函数

时间:2016-06-26 16:48:38

标签: javascript testing sinon

我对Sinon有点新,并且我不仅需要监视一个函数,而且需要监视函数返回的函数。具体来说,我正在尝试模拟Azure存储SDK,并确保在创建队列服务后,还会调用队列服务返回的方法。这是一个例子:

// test.js

// Setup a Sinon sandbox for each test
test.beforeEach(async t => {

    sandbox = Sinon.sandbox.create();
});

// Restore the sandbox after each test
test.afterEach(async t => {

    sandbox.restore();
});

test.only('creates a message in the queue for the payment summary email', async t => {

    // Create a spy on the mock
    const queueServiceSpy = sandbox.spy(AzureStorageMock, 'createQueueService');

    // Replace the library with the mock
    const EmailUtil = Proxyquire('../../lib/email', {
        'azure-storage': AzureStorageMock,
        '@noCallThru': true
    });

    await EmailUtil.sendBusinessPaymentSummary();

    // Expect that the `createQueueService` method was called
    t.true(queueServiceSpy.calledOnce); // true

    // Expect that the `createMessage` method returned by
    // `createQueueService` is called
    t.true(queueServiceSpy.createMessage.calledOnce); // undefined
});

这是模拟:

const Sinon = require('sinon');

module.exports =  {

    createQueueService: () => {

        return {

            createQueueIfNotExists: (queueName) => {

                return Promise.resolve(Sinon.spy());
            },

            createMessage: (queueName, message) => {

                return Promise.resolve(Sinon.spy());
            }
        };
    }
};

我能够确认queueServiceSpy被调用一次,但我无法确定该方法返回的方法是否被调用(createMessage)。

有没有更好的方法来设置它或我只是遗漏了什么?

谢谢!

1 个答案:

答案 0 :(得分:2)

您需要做的是保留您的服务功能以返回间谍,然后您可以跟踪其他地方的来电。你可以任意深入地嵌套(虽然我强烈反对这一点)。

类似的东西:

const cb = sandbox.spy();
const queueServiceSpy = sandbox.stub(AzureStorageMock, 'createQueueService')
    .returns({createMessage() {return cb;}}});

const EmailUtil = Proxyquire('../../lib/email', {
    'azure-storage': AzureStorageMock,
    '@noCallThru': true
});

await EmailUtil.sendBusinessPaymentSummary();

// Expect that the `createQueueService` method was called
t.true(queueServiceSpy.calledOnce); // true

// Expect that the `createMessage` method returned by
// `createQueueService` is called
t.true(cb.calledOnce);