使用回调存储函数

时间:2015-07-06 08:47:36

标签: javascript sinon stubbing

通常我使用此代码:

  aws_stub.S3 = function(){};
  var fake_aws_listObjects = function(params, func){func("failure", null)};
  var fake_aws_listObjects_stub = sinon.spy(fake_aws_listObjects);
  aws_stub.S3.prototype.listObjects = fake_aws_listObjects_stub;

将以下函数存根:

new AWS.S3().listObjects(that.build_params(), function(err, data) {
  if(err){
    that.make_failure_callback();
  }
  else{
    that.add_keys(data);

    if(data.IsTruncated){
      that.make_request(data.NextMarker);
    }else{
      that.make_success_callback(that.keys);
    }
  }
});

问题这个存根是每个请求 返回相同的

现在我想做的是每次通话都有不同的存根:

  aws_stub.S3 = function(){};
  var fake_aws_truncated = function(params, func){func(null, {
    Contents: [{Key: "3fb252ba-0724-438c-93b6-8cd0fd972a8e/image/1:::2.jpg"}],
    IsTruncated: true,
    NextMarker: "nextMarker"
    })};
  var fake_aws_listObjects = function(params, func){func(null, {Contents: [{
    Key: "3fb252ba-0724-438c-93b6-8cd0fd972a8e/image/2:::3.jpg"
  }]})};

  var fake_aws_listObjects_stub = sinon.stub();
     fake_aws_listObjects_stub.onCall(0).returns(fake_aws_truncated);
      fake_aws_listObjects_stub.onCall(1).returns(fake_aws_listObjects);

  aws_stub.S3.prototype.listObjects = fake_aws_listObjects_stub;

问题似乎是 returns,它没有执行该功能!!

我也不能这样写:

fake_aws_listObjects_stub.onCall(0) = fake_aws_truncated;

因为这是一个错误的手工作业!

我需要改变什么?以下是sinon文档:http://sinonjs.org/docs/

谢谢!

1 个答案:

答案 0 :(得分:2)

我可能会采取另一种方式。

如果您想要存根AWS.S3.prototype.listObjects,我会这样做:

var stub = sinon.stub(AWS.S3.prototype, 'listObjects');

要使用各种值调用回调,请使用stub.yields()

stub.onCall(0).yields(null, {
  Contents    : [{Key: "3fb252ba-0724-438c-93b6-8cd0fd972a8e/image/1:::2.jpg"}],
  IsTruncated : true,
  NextMarker  : "nextMarker"
});

stub.onCall(1).yields(null, {
  Contents : [{Key: "3fb252ba-0724-438c-93b6-8cd0fd972a8e/image/2:::3.jpg"}]
});

要测试您的代码,您只需像以前一样致电listObjects

var s3 = new AWS.S3();
s3.listObjects(params, function(err, value) {
  ...`value` is now one of the fixtures you declared above...
});

要恢复到该方法的原始版本,请使用以下方法之一:

stub.restore();
// OR:
AWS.S3.prototype.listObjects.restore();