通过模拟测试两种不同的期望

Question

我最近刚刚将 Devise 添加到我的第一个 Rails3 应用程序中，但在控制器测试方面遇到了一些麻烦。

我正在测试 User 控制器类，它与 Devise 使用的模型相同。因此，在我的规范的开头，我有这样的内容：

before(:each) do
  sign_in @user = Factory.create(:user)
end

现在我可以在不使用模拟或存根的情况下通过测试，如下所示：

describe "GET edit" do
  it "assigns the requested user as @user" do
    user = Factory(:user)
    get :edit, :id => user.id
    assigns(:user).should eql(user)
  end
end

但出于教育目的，我想知道如何让它与模拟和存根一起工作，通常是完全简单，但 Devise 似乎在控制器操作之前调用 User.find ，并使测试失败。

describe "GET edit" do
  it "assigns the requested user as @user" do
    user = Factory(:user)
    User.expects(:find).with(:first, :conditions => {:id => 37}).returns(user)
    get :edit, :id => '37'
    assigns(:user).should be(user)
  end
end

此外，通过在期望中添加两次也会失败，因为第一次调用 find 与我设置期望的调用不同。

任何见解将不胜感激。

Answer 1

您可以使用 stubs 或 expects 指定多个返回值，如下所示：

require 'test/unit'
require 'mocha'

class Foo
end

class FooTest < Test::Unit::TestCase

  # This passes!
  def test_multiple_returns_with_stubs
    Foo.stubs(:find).returns('a').returns('b').returns('c')

    assert_equal('a', Foo.find('z'))
    assert_equal('b', Foo.find('y'))
    assert_equal('c', Foo.find('x'))
  end

  # This passes too!
  def test_multiple_returns_with_expects
    Foo.expects(:find).times(1..3).returns('a').returns('b').returns('c')

    assert_equal('a', Foo.find('z'))
    assert_equal('b', Foo.find('y'))
    assert_equal('c', Foo.find('x'))
  end
end

显然，区别在于 expects 需要知道它是多少次将会被召唤。如果您没有指定任何内容，它会假设once，并且会在后续调用中抱怨。 存根不在乎。