如何使用FactoryBot定义嵌套工厂?

时间:2012-08-24 23:16:32

标签: ruby ruby-on-rails-3 rspec factory-bot

给定两个模型,Alert和Zipcode,其中一个Alert必须有1个或更多Zipcodes:

class Alert < ActiveRecord::Base
  attr_accessible :descr, :zipcode

  has_many :zipcode
  validates :zipcode, :length => { :minimum => 1 }
end

class Zipcode < ActiveRecord::Base
  attr_accessible :zip
  belongs_to :alert
end 

如何编写FactoryBot工厂,以便:

  • Zipcode工厂在自己的文件中定义
  • 警报工厂在其自己的文件中定义
  • 警报可以依赖Zipcode定义的工厂吗?

我读过的所有文档和示例都希望您在父工厂文件中定义包含的类,将它们全部组合在一起,或者做出其他一些妥协或解决方法。是不是有一种干净的方法可以让规范工厂分开?

1 个答案:

答案 0 :(得分:12)

诀窍是确保容器类(即定义中带有has_many语句的容器类)在FactoryBot中将包含的类创建为数组。例如:

在您的spec / factories / zipcodes.rb中:

FactoryBot.define do
  factory :zipcode do
    zip { 78701 + rand(99) } 
  end
end

在spec / factories / alerts.rb中:

FactoryBot.define do
  factory :alert do
    zipcode { Array.new(3) { FactoryBot.build(:zipcode) } }
  end
end