使用 Rspec on Rails 在回调中测试关联对象的创建

Question

试图围绕 rspec 和适当的测试，并且很难正确地执行以下操作

假设我们有三个类

Class User
    belongs_to :company
    has_and_belongs_to_many :roles
end

Class Company
    has_many :users
end

Class Role
    has_and_belongs_to_many :users
end

在 User 类中，如果用户是第一个与公司关联的用户，我有 before_create 回调分配用户默认的“company_admin”角色

def check_for_initial_company_admin_role
  if self.company.users.count == 0
    self.roles << Role.find_by_name("company_admin")
  end
end

如果用户是与公司关联的第一个用户，如何在我的模型规范中正确测试该用户被分配了“company_admin”角色？

---

更新 解决方案

describe "#check_for_initial_company_admin_role" do
  it "sets the first user created to be the administrator" do
    Factory(:role)
    user = Factory(:user)

    user.roles.count.should be > 0
    user.roles.should include Role.find_by_name("company_admin")
  end
end


Factory.define :user do |f|
  f.email { Factory.next(:email) }
  f.password "secret"
  f.password_confirmation "secret"
  f.association :company
end 

Factory.define :role do |f|
  f.name "company_admin"
end

Factory.define :company do |f|
  f.name "foobar"
  f.vat_id "1234"
end

Answer 1

我会这样处理它：

describe "#check_for_initial_company_admin_role" do
  it "sets the first user created to be the administrator" do
    company = Factory(:company)
    user = Factory(:user)
    company.users << user


    user.roles.count.should > 0
    user.roles.should include Role.find_by_name("company_admin")
  end
end

这里可能不正确的假设是您在测试框架中使用了 Factory Girl。如果没有，那并不会真正改变这个测试的“内容”……只是你创建公司和用户的第一行。

您还可以选择从公司方面检查用户，但老实说，这感觉就像一个完全不同的测试——测试这些模型之间的关联。

我会采取的方法是，由于这实际上是一个模型测试，您需要创建和更改真实的模型对象，而不是模拟这些对象。如果这是一个控制器测试，我会模拟模型并积极地存根模型。

我希望这有助于解决您的问题。如果没有，请告诉我我在哪里，我会再做一遍:) 我只有大约一年的时间进入 rspec，但我发现一旦我开始思考如何测试模型与模型。控制器我已经爱上它了。

Answer 2

在不更改现有逻辑的情况下，我将在 user_spec 中测试此逻辑，如下所示：

describe User do
  let!(:admin_role) { Role.create!(name: 'company_admin') }
  let!(:company) { Company.create! }

  it 'should be added to the default role when created' do
    user = company.users.create!(name: 'Joe', email: 'joe@email.com')

    user.should have(1).roles
    user.roles.first.should == admin_role
  end
end

注意：我会使用 FactoryGirl 之类的东西作为管理员角色和公司对象，以使它们可重用。

您使用角色名称来指示行为并不理想。它可能会导致整个应用程序中出现大量分散的逻辑，您可以通过角色名称查找角色并使用 if/else 或 case 语句检查名称。我建议使用 Rail 的单表继承并将所有管理员角色逻辑移至单独的类。它将保持模型的逻辑更清晰，并使测试更容易。