测试接收到的数组大小是否正确

Question

我想测试一个类的函数是否被特定长度的数组调用。

下面的示例我想验证函数 s3upload 是否使用数组大小​​为 2 的数组参数调用。

• ATest_spec.rb 是测试类。 AwsUploader.s3upload 是从 letsUpload 调用的

Project/lib/First/version.rb
  module Documentz
    class AwsUploader
      def s3upload(event_id:nil, docs: nil)
        puts("uploded")
      end
    end
  end
Project/lib/First.rb
  module Exporter
    class AnExporter
      def letsUpload
        Documentz::Uploader::AwsUploader.new.s3upload( docs :[1,2])
      end
    end
  end
ATest_spec.rb
  it 'helps in mocking a class' do
    exp=Exporter::AnExporter.new
    exp.letsUpLoad
    allow_any_instance_of(Documentz::Uploader::AwsUploader).to receive(:s3upload).with( {:docs=>[1,2]})
    ## how to check if the array size (:docs)==2
  end

正如您在ATest_spec.rb 中注意到的那样，我可以测试参数是否为 [1,2] 但我实际上想验证数组的大小（接收到的参数）实际上是 2。

你能建议怎么做吗？

Answer 1

我将使用new 方法而不是allow_any_instance_of 存根，并返回一个instance_double，我在它上面监视预期的方法调用。为确保参数具有特定结构，请使用可以根据需要复杂的自定义匹配器，例如：

RSpec::Matchers.define :expected_data_structure do
  match { |actual| actual.is_a?(Hash)         &&
                   actual[:docs].is_a?(Array) && 
                   actual[:docs].size == 2    &&
                   actual[:docs].all?(Integer) 
  }
end

subject(:exporter) { Exporter::AnExporter.new }
let(:spy) { instance_double('Documentz::Uploader::AwsUploader') }

before do 
  allow(Documentz::Uploader::AwsUploader).to receive(:new).and_return(spy) 
end

it 'calls `s3upload` with the expected arguments' do
  exporter.letsUpLoad

  expect(spy).to have_received(:s3upload).with(expected_data_structure)
end

在 RSpec 文档中了解 custom matches。

顺便说一句。在 Ruby 中，按照惯例，方法名称是用下划线而不是驼峰写的。根据该规则，您的方法应命名为 lets_up_load（或只是 upload）而不是 letsUpLoad

Answer 2

在这里，您已经模拟了一个接收特定参数的方法，此外，您需要返回一个文档 ID 数组（假设这是您的 s3upload 方法将返回的内容）

allow_any_instance_of(Documentz::Uploader::AwsUploader).to receive(:s3upload).with( {:docs=>[1,2]}).and_return([1,2])
    
expect(exp.letsUpLoad.length).to eq 2