ruby rspec:如何让存根在每次调用时都有不同的行为

RSpec.describe "Stubbing with different behaviors" do
  let(:my_object) { double("MyObject") }

  it "returns different values on each call" do
    allow(my_object).to receive(:my_method) do |arg|
      # 这里可以根据传入的参数或其他条件返回不同的值
      case arg
      when 1 then "first"
      when 2 then "second"
      else "unknown"
      end
    end

    expect(my_object.my_method(1)).to eq("first")
    expect(my_object.my_method(2)).to eq("second")
    expect(my_object.my_method(3)).to eq("unknown")
  end
end

RSpec.describe "Stubbing with yield" do
  let(:my_object) { double("MyObject") }

  it "yields different blocks on each call" do
    allow(my_object).to receive(:my_method) do |&block|
      # 这里可以执行传入的块，并根据需要返回不同的结果
      block.call
    end

    expect(my_object.my_method { "first" }).to eq("first")
    expect(my_object.my_method { "second" }).to eq("second")
  end
end