FactoryGirl与嵌套的belongs_to关联

时间:2013-06-27 15:00:48

标签: ruby factory-bot

我有两个型号,我们称之为模型Flower和模型Bee。蜜蜂总是坚持同一朵花,因此他们belong_to。一朵花可以有多只蜜蜂,但是有一只蜂王,我们想要在花朵记录中保留一个指向主蜜蜂的“指针”。这样做的原因是我们希望能够灵活地将花点指向新的蜜蜂,而无需修改蜜蜂模型。我们还想避免单独查询以找到合适的蜂王。

class Flower < ActiveRecord::Base
  belongs_to :queen_bee, class_name: 'Bee'
  has_many :bees

  after_create :create_queen_bee
end

class Bee < ActiveRecord::Base
  belongs_to :flower
end

我有Flower的FactoryGirl工厂,我需要确保创建了蜂王记录。

这样做的正确方法是什么?我的想法是我应该能够做到

flower = FactoryGirl.create(:flower)
puts flower.queen_bee.name
puts flower.queen_bee.flower.name

1 个答案:

答案 0 :(得分:0)

如果你是通过after_create回调创建的,那么应该自动创建它(假设你正确地写了create_queen_be)。

但是,如果您想通过FactoryGirl,在工厂内,您可以指定您的queen_bee;像这样:

factory :flower do
  queen_bee { FactoryGirl.create(:bee) }
end