如何根据参数区分对同一方法的两次调用?

Question

我在NodeJS应用程序中使用sinon来存根Google Drive的功能。在相同的测试下，我对create方法进行了两次不同的调用(我不能在调用之间进行恢复):

// Call 1:
drive.files.create({ 'requestBody': requestBody, 'media': media });
// Call 2:
drive.files.create({ 'resource': resource });

为了存根，我可以这样做:

const stub = sinon.stub(drive.files, 'create').returns({
'status': 200,
'data': {
'files': [{ 'id': id }]
}
});

这个存根，实际上存根了两个调用，但是如果我想要第一次调用成功，第二次调用失败怎么办?如何根据参数来区分这两个调用(可能)?

Answer 1

可以使用stub.withArgs(arg1[, arg2, ...]);API

创建一个stub也很有用，它可以对不同的参数做出不同的响应。

。

const sinon = require('sinon');
describe('72234931', () => {
it('should pass', () => {
const drive = {
files: {
create(opts) {},
},
};
const stub = sinon.stub(drive.files, 'create');
stub.withArgs({ requestBody: 'requestBody', media: 'media' }).returns({
status: 200,
data: {
files: [{ id: 1 }],
},
});
stub.withArgs({ resource: 'resource' }).returns({ status: 500 });
const r1 = drive.files.create({ requestBody: 'requestBody', media: 'media' });
sinon.assert.match(r1, { status: 200, data: { files: [{ id: 1 }] } });
const r2 = drive.files.create({ resource: 'resource' });
sinon.assert.match(r2, { status: 500 });
});
});