ruby-on-rails - 工厂少女协会自引用父模型

Question

在我的应用程序中，一个帐户可以有一个所有者(用户)和多个用户。

在我的测试中我这样做:

# account_factory_static.rb
FactoryGirl.define do
  factory :account do
    name 'demoaccount'
    association :owner, :factory => :user
  end
end

# user_factory_static.rb
FactoryGirl.define do
  factory :user do
    email 'demo@example.com'
    first_name 'Jon'
    last_name 'Doe'
    password 'password'
  end
end

并像下面这样使用它们:

let(:account) { FactoryGirl.create(:account) }

问题是现在account.users.count等于 0因为我没有办法做类似 @account.users << @account.owner 的事情就像我在用户注册时在我的 Controller 中所做的那样。

问题是如何将关联帐户的 id 添加到 FactoryGirl 中用户的 account_id 属性中？

换句话说，你如何在 FactoryGirl 中做到这一点？

谢谢。

Answer 1

你可以使用 after :create block :

FactoryGirl.define do
  factory :account do
    name 'demoaccount'
    association :owner, :factory => :user

    after :create do |account|
      account.users << account.owner
    end
  end
end