如何在 Sinon 中模拟链式函数调用

Question

你如何使用这样的诗乃模拟？

const data = await getData(); 
const res= await data.collection('myCollection').deleteOne({ id: 12 });

Answer 1

您可能正在寻找.returnsThis()，它只返回函数的this，就像这样的流式API 所做的那样，并且专门用于存根此类API。

你可以这样使用它：

const db = require('...')

async function test() {
  // Setup
  const fakeResponse = {
    collection: Sinon.stub().returnsThis(),
    deleteOne: Sinon.stub().returnsThis()
  }
  Sinon.stub(db, 'getData').resolves(fakeResponse)
  
  // Call your code that does stuff
  const data = await db.getData(); 
  const res = await data.collection('myCollection').deleteOne({ id: 12 })
  
  // Assert
  Sinon.assert.calledWith(fakeResponse.collection, 'myCollection')
  Sinon.assert.calledWith(fakeResponse.deleteOne, { id: 12 })
}

我在这里假设getData 是某个对象的方法，比如db。如果没有，您将不得不想出其他方法来存根 getData()。