How to make fun of an instance method of an already mocked object?

Question

I need to make fun of the following:

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

So, in my user tests, in order to access facebook user information, I need to call user.facebook.me.infoto get the information. If I want to mock this, I am currently using the following:

@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"]

This works, but it seems a little cumbersome, is there a better way to do this?

[edit] I use mocha as a mockery of the framework

Answer 1

You can try something like this: -

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

If you really want to verify that all of these methods have been called (which I suspect you didn’t), you can do the following: -

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 , .