如何在 Factory Girl 中创建与 has_many :through 关系的关联？

Question

在我的模型中，我有以下设置：

class User < ActiveRecord::Base
  has_many :assignments
  has_many :roles, :through => :assignments
end


class Role < ActiveRecord::Base
  has_many :assignments
  has_many :users,  :through => :assignments
end

class Assignment < ActiveRecord::Base
  belongs_to :user
  belongs_to :role

  attr_accessible :role_id, :user_id
end

在我的factory.rb 文件中，我有：

FactoryGirl.define do
  factory :user do
    sequence(:username) { |n| "user#{n}" }
    email { "#{username}@example.com" }
    password 'secret'
    password_confirmation 'secret'

    factory :admin  do
      ...
    end
  end

  factory :role do
    name 'Normal'
    value 'normal'
  end

  factory :assignment do
    ...
  end
end

我正在努力弄清楚如何添加角色， :name => “管理员”，：值=> “admin”，到“user”块内的“admin”工厂，这样我就可以调用

create(:admin)

我的测试并让用户具有管理员角色。

谢谢您的关注。

Answer 1

对于这样的工厂，您需要使用工厂女孩的回调。
试试这个：

FactoryGirl.define do
  factory :user do
    ...
  end

  factory :admin, :parent => :user do 
    after_create {|u| Factory(:assignment, :role => Factory(:role, :name => 'admin', :value => 'admin'), :user => u)}
  end

  factory :role do
    ...
  end

  factory :assignment do
    user {|a| a.association(:user)}
    role {|a| a.association(:role)}
  end
end

Answer 2

@kshil 是正确的，但您可以稍微收紧代码并使其更加模块化。

为管理员用户创建第二个 :role 工厂。

factory :role do
  name 'Normal'
  value 'normal'

  factory :admin_role do
    name  'admin'
    value  'admin'
  end
end

此外，如果工厂名称与关联名称相同，则可以省略工厂名称。 :assignment 工厂变为：

factory :assignment do
  user
  role
end

在 :user 工厂内定义 :admin_user 工厂，并且不必指定父工厂。您还可能添加两个工厂来定义普通用户和管理员用户。

factory :user do
  sequence(:username) { |n| "user#{n}" }
  email { "#{username}@example.com" }
  password 'secret'
  password_confirmation 'secret'

  factory :normal_user do
    after_create {|u| Factory(:assignment, :user => u)}
  end

  factory :admin_user do
    after_create {|u| Factory(:assignment, :role => Factory(:admin_role), :user => u)}
  end
end