Как создать ассоциацию с отношениями has_many: through в Factory Girl? - PullRequest
Новая
       20

Как создать ассоциацию с отношениями has_many: through в Factory Girl?

2 голосов
/ 21 декабря 2011

В моих моделях у меня есть следующие настройки: 

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",: value => "admin" для фабрики "admin" внутри блока "user", чтобы я мог вызывать 

create(:admin)

в моих тестах и ​​иметьпользователь с ролью администратора.

Спасибо, что заглянули.

Ответы [ 2 ]

5 голосов
/ 21 декабря 2011

Для такой фабрики нужно использовать обратные вызовы фабричной девушки. Попробуйте это: 

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
3 голосов
/ 21 декабря 2011

@ 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

Определите фабрику :admin_user внутри фабрики :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
...