Cách tạo has_and_belongs_to_many liên kết trong Factory girl

Cho những điều sau đây

class User < ActiveRecord::Base
  has_and_belongs_to_many :companies
end

class Company < ActiveRecord::Base
  has_and_belongs_to_many :users
end

làm thế nào để bạn xác định nhà máy cho các công ty và người dùng bao gồm cả liên kết hai chiều? Đây là nỗ lực của tôi

Factory.define :company do |f|
  f.users{ |users| [users.association :company]}
end

Factory.define :user do |f|
  f.companies{ |companies| [companies.association :user]}
end

bây giờ tôi thử

Factory :user

Có lẽ không có gì đáng ngạc nhiên khi điều này dẫn đến một vòng lặp vô hạn khi các nhà máy sử dụng đệ quy lẫn nhau để xác định chính chúng.

Đáng ngạc nhiên hơn là tôi không tìm thấy đề cập đến cách thực hiện điều này ở bất cứ đâu, có một khuôn mẫu nào để xác định các nhà máy cần thiết hay tôi đang làm điều gì đó sai về cơ bản?

Đây là giải pháp phù hợp với tôi.

FactoryGirl.define do

  factory :company do
    #company attributes
  end

  factory :user do
   companies {[FactoryGirl.create(:company)]}
   #user attributes
  end

end

nếu bạn cần một công ty cụ thể, bạn có thể sử dụng nhà máy theo cách này

company = FactoryGirl.create(:company, #{company attributes})
user = FactoryGirl.create(:user, :companies => [company])

Hy vọng điều này sẽ hữu ích cho ai đó.

Factorygirl kể từ đó đã được cập nhật và hiện bao gồm các lệnh gọi lại để giải quyết vấn đề này. Hãy xem tại http://robots.thoughtbot.com/post/254496652/aint-no-calla-back-girl để biết thêm thông tin.

Theo tôi, chỉ cần tạo hai nhà máy khác nhau như:

 Factory.define: user,: class => User do | u |
  # Chỉ khởi tạo thuộc tính bình thường
 kết thúc

 Factory.define: company,: class => Company do | u |
  # Chỉ khởi tạo thuộc tính bình thường
 kết thúc

Khi bạn viết test-case cho người dùng thì chỉ cần viết như thế này

 Nhà máy (: người dùng,: công ty => [Nhà máy (: công ty)])

Hy vọng nó sẽ hoạt động.

Tôi không thể tìm thấy một ví dụ cho trường hợp được đề cập ở trên trên trang web được cung cấp. (Chỉ có 1: N và giả thiết đa hình, nhưng không có habtm). Tôi đã gặp một trường hợp tương tự và mã của tôi trông như thế này:

Factory.define :user do |user|
 user.name "Foo Bar"
 user.after_create { |u| Factory(:company, :users => [u]) }
end

Factory.define :company do |c|
 c.name "Acme"
end

Điều làm việc cho tôi là thiết lập sự liên kết khi sử dụng nhà máy. Sử dụng ví dụ của bạn:

user = Factory(:user)
company = Factory(:company)

company.users << user 
company.save! 

Tìm thấy cách này tốt và dài:

FactoryGirl.define do
  factory :foo do
    name "Foo" 
  end

  factory :bar do
    name "Bar"
    foos { |a| [a.association(:foo)] }
  end
end

  factory :company_with_users, parent: :company do

    ignore do
      users_count 20
    end

    after_create do |company, evaluator|
      FactoryGirl.create_list(:user, evaluator.users_count, users: [user])
    end

  end

Cảnh báo: Thay đổi người dùng: [user] thành: users => [user] cho ruby ​​1.8.x

Trước hết, tôi thực sự khuyến khích bạn sử dụng has_many: through thay vì habtm (thêm về điều này ở đây ), vì vậy bạn sẽ kết thúc với một cái gì đó như:

Employment belongs_to :users
Employment belongs_to :companies

User has_many :employments
User has_many :companies, :through => :employments 

Company has_many :employments
Company has_many :users, :through => :employments

Sau đó, bạn sẽ có has_many liên kết ở cả hai bên và có thể gán cho chúng trong factory_girl theo cách bạn đã làm.

Cập nhật cho Rails 5:

Thay vì sử dụng has_and_belongs_to_manyliên kết, bạn nên xem xét: has_many :throughliên kết.

Nhà máy người dùng cho liên kết này trông giống như sau:

FactoryBot.define do
  factory :user do
    # user attributes

    factory :user_with_companies do
      transient do
        companies_count 10 # default number
      end

      after(:create) do |user, evaluator|
         create_list(:companies, evaluator.companies_count, user: user)
      end
    end
  end
end

Bạn có thể tạo nhà máy công ty theo cách tương tự.

Khi cả hai nhà máy được thiết lập, bạn có thể tạo user_with_companiesnhà máy với companies_count option. Tại đây, bạn có thể chỉ định số lượng công ty mà người dùng thuộc về:create(:user_with_companies, companies_count: 15)

Bạn có thể tìm thấy lời giải thích chi tiết về các hiệp hội nữ công xưởng tại đây .

Đối với HABTM, tôi đã sử dụng các đặc điểm và lệnh gọi lại .

Giả sử bạn có các mô hình sau:

class Catalog < ApplicationRecord
  has_and_belongs_to_many :courses
  …
end
class Course < ApplicationRecord
  …
end

Bạn có thể xác định Nhà máy ở trên :

FactoryBot.define do
  factory :catalog do
    description "Catalog description"
    …

    trait :with_courses do
      after :create do |catalog|
        courses = FactoryBot.create_list :course, 2

        catalog.courses << courses
        catalog.save
      end
    end
  end
end

Bạn có thể xác định nhà máy mới và sử dụng lệnh gọi lại after (: create) để tạo danh sách các liên kết. Hãy xem cách làm điều đó trong ví dụ này:

FactoryBot.define do

  # user factory without associated companies
  factory :user do
    # user attributes

    factory :user_with_companies do
      transient do
        companies_count 10
      end

      after(:create) do |user, evaluator|
        create_list(:companies, evaluator.companies_count, user: user)
      end
    end
  end
end

Thuộc tính company_count là một thuộc tính tạm thời và có sẵn trong các thuộc tính của nhà máy và trong lệnh gọi lại thông qua trình đánh giá. Bây giờ, bạn có thể tạo người dùng với các công ty với tùy chọn chỉ định số lượng công ty bạn muốn:

create(:user_with_companies).companies.length # 10
create(:user_with_companies, companies_count: 15).companies.length # 15