可以在单个测试中两次存根方法以返回不同的结果吗？

Question

我想重新存根someHandler.getStatus，但我收到了TypeError: Attempted to wrap getStatus which is already wrapped ..

it('is a test', function() {

  sandbox.stub(someHandler, 'getStatus', function(callback) {
    callback(null, {
      value: 1
    });
  });

  sandbox.stub(someOtherHandler, 'doSomething', function(callback) {
    callback(null);
  });

  sandbox.stub(someHandler, 'getStatus', function(callback) {
    callback(null, {
      value: 0
    });
  });
});

Answer 1

Sinon 有一个很好的 API 可以处理多个调用 (stub.onCall(n);) 到同一个存根方法。

来自stub api doc的示例：

"test should stub method differently on consecutive calls": function () {
    var callback = sinon.stub();
    callback.onCall(0).returns(1);
    callback.onCall(1).returns(2);
    callback.returns(3);

    callback(); // Returns 1
    callback(); // Returns 2
    callback(); // All following calls return 3
}

您也可以使用onFirstCall()、onSecondCall() 和onThirdCall()。

我们在测试中广泛使用这种方法。

Answer 2

尝试使用外部变量跟踪调用。第一次调用时，返回第一个值并递增计数器。第二次调用（以及所有后续调用），返回第二个值。

it('is a test', function() {

  let callCount = 0

  sandbox.stub(someHandler, 'getStatus', function(callback) {
    callback(null, {
      value: callCount++ == 0 ? 1 : 0
    });
  });

  sandbox.stub(someOtherHandler, 'doSomething', function(callback) {
    callback(null);
  });    
});

这是一个处理 2 个调用的简单示例，但如果需要，您可以将其扩展为处理多个调用。