如何模拟已模拟对象的实例方法？

Question

我需要模拟以下内容：

Class User
  def facebook
    #returns an instance of a facebook gem
  end
end

因此，在我的用户测试中，要访问用户的 facebook 信息，我需要调用 user.facebook.me.info 来检索其信息。如果我想嘲笑这个，我目前正在使用以下内容：

@user = Factory(:user)
facebook = mock()
me = mock()
me.expects(:info).returns({"name" => "John Doe"})
facebook.expects(:me).returns(me)
@user.expects(:facebook).returns(facebook)
assert_equal "John Doe", @user.facebook.me.info["name"]

这有效，但似乎有点笨拙，有更好的方法吗？

[编辑] 我使用 mocha 作为模拟框架

Answer 1

你可以尝试这样的事情：-

user = Factory(:user)
user.stubs(:facebook => stub(:me => stub(:info => {:name => "John Doe"})))

如果你真的想检查所有这些方法是否被调用（我怀疑你没有），你可以执行以下操作：-

user = Factory(:user)
user.expects(:facebook => mock(:me => mock(:info => {:name => "John Doe"})))

它有点冗长，但通常值得给每个模拟反对一个名字：-

user = Factory(:user)
user.stubs(:facebook => stub('facebook', :me => stub('me', :info => {:name => "John Doe"})))

我希望有帮助。

Answer 2

如果您不想检查是否调用了所有方法，您还可以使用不同的模拟替代方法。例如，您可以使用 OpenStruct。

@user = Factory(:user)
facebook = mock()
me = mock()
me.expects(:info).returns({"name" => "John Doe"})
facebook.expects(:me).returns(me)
@user.expects(:facebook).returns(facebook)
assert_equal "John Doe", @user.facebook.me.info["name"]

这个解决方案还为您提供了

@user = Factory(:user)
@facebook = OpenStruct.new(:me => OpenStruct.new(:info => {:name => "John Doe"}))
@user.expects(:facebook).returns(@facebook)

一个优势，如果您需要测试不同的条件，您可以在测试中更改@facebook 属性。