RSpec 存根方法可以按顺序返回不同的值吗？

Question

我有一个模型系列，其方法 location 合并了其他对象 Members 的 location 输出。 （成员与家庭相关联，但这在这里并不重要。）

例如，给定

• member_1 具有 location == '圣地亚哥（旅行，5 月 15 日返回）'
• member_2 具有 location == '圣地亚哥'

Family.location 可能会返回 '圣地亚哥（member_1 旅行，5 月 15 日返回）' 具体细节并不重要。

为了简化 Family.location 的测试，我想存根 Member.location。但是，我需要它返回两个不同的（指定的）值，如上面的示例所示。理想情况下，这些将基于 member 的属性，但只需按序列返回不同的值就可以了。有没有办法在 RSpec 中做到这一点？

可以在每个测试示例中覆盖 Member.location 方法，例如，

it "when residence is the same" do 
  class Member
    def location
      return {:residence=>'Home', :work=>'his_work'} if self.male?
      return {:residence=>'Home', :work=>'her_work'}
    end
  end
  @family.location[:residence].should == 'Home'
end

但我怀疑这是一个好的做法。无论如何，当 RSpec 运行一系列示例时，它不会恢复原始类，因此这种覆盖会“毒害”后续示例。

那么，有没有办法让存根方法在每次调用时返回不同的指定值？

Answer 1

您可以对方法进行存根处理，以在每次调用该方法时返回不同的值；

allow(@family).to receive(:location).and_return('first', 'second', 'other')

因此，第一次调用 @family.location 时，它将返回“first”，第二次调用时，它将返回“second”，后续所有调用它的时候，它将返回“other”。

Answer 2

RSpec 3 语法：

allow(@family).to receive(:location).and_return("abcdefg", "bcdefgh")

Answer 3

仅当您有特定数量的调用并且需要特定的数据序列时，才应使用已接受的解决方案。但是如果您不知道将进行的调用数量，或者不关心数据的顺序，只是每次都不同，该怎么办？正如OP所说：

简单地按序列返回不同的值就可以了

and_return 的问题是返回值被记忆。这意味着即使您返回动态的东西，您也将始终得到相同的结果。

例如

allow(mock).to receive(:method).and_return(SecureRandom.hex)
mock.method # => 7c01419e102238c6c1bd6cc5a1e25e1b
mock.method # => 7c01419e102238c6c1bd6cc5a1e25e1b

，或者一个实际的例子是使用工厂并获取相同的 ID：

allow(Person).to receive(:create).and_return(build_stubbed(:person))
Person.create # => Person(id: 1)
Person.create # => Person(id: 1)

在这些情况下，您可以存根方法主体以使代码每次执行：

allow(Member).to receive(:location) do
  { residence: Faker::Address.city }
end
Member.location # => { residence: 'New York' }
Member.location # => { residence: 'Budapest' }

请注意，您无法通过以下方式访问 Member 对象： self 在此上下文中，但可以使用测试上下文中的变量。

例如

member = build(:member)
allow(member).to receive(:location) do
  { residence: Faker::Address.city, work: member.male? 'his_work' : 'her_work' }
end

Answer 4

如果由于某种原因您想使用旧语法，您仍然可以：

@family.stub(:location).and_return('foo', 'bar')

Answer 5

我已经尝试过上面的解决方案大纲，但它对我不起作用。我通过使用替代实现进行存根解决了这个问题。

像这样的东西：

@family.stub(:location) { rand.to_s }

Answer 6

我遇到的问题是，我对一个方法进行了多次调用，但顺序并不总是相同。在这种情况下，我建议使用 .with 来区分实例，并使用方法的参数。

例如，这可能是您的“默认”返回值：

allow(@family).to receive(:location).and_return('her_work')

但是，如果 location 收到像“male”这样的参数，您可以添加：

allow(@family).to receive(:location).with("male").and_return('his_work')

有很多不同的匹配参数类型可以与 .with< 一起使用/code>:

https:// relishapp.com/rspec/rspec-mocks/v/3-2/docs/setting-constraints/matching-arguments

Answer 7

可以帮助您的模式：

allow(MyClass).to receive(:my_method) do |kwargs|
    # define the response for "my_method" (kwargs is the hash of arguments given to the method)
    "my return value"
end