如何指定与factory_bot的关联？

Question

例如,我有两个模型,一个用户和一个帖子.帖子属于用户,用户有很多帖子

#spec/factories/post.rb
FactoryBot.define do
  factory :post do
    user
    body Faker::Movie.quote
    posted_at "2018-04-03 13:33:05"
  end
end

#spec/factories/user.rb
FactoryBot.define do 
  factory :user do 
    first_name 'Jake'
  end
end

在测试中使用Rspec我想这样做:

user = create(:user, first_name: 'Barry') #id 1
post = create(:post, user: user)

我希望post的user_id为1但是它先创建另一个用户,user_id为2.

如何使用factory_bot/factory_girl创建对象时指定关联？

Answer 1

您应该使用explicit关联而不是关联implicit:

#spec/factories/post.rb
FactoryBot.define do
  factory :post do
    association :user    # <<<--- here the change
    body Faker::Movie.quote
    posted_at "2018-04-03 13:33:05"
  end
end

#spec/factories/user.rb
FactoryBot.define do 
  factory :user do 
    first_name 'Jake'
  end
end

https://github.com/thoughtbot/factory_bot/blob/master/GETTING_STARTED.md

Answer 2

如果您的关联名称和工厂名称不同，我们还有另一种解决方案，您可以遵循以下语法。

#spec/factories/post.rb
FactoryBot.define do
  factory :post do
    association :author, factory: :user
    body Faker::Movie.quote
    posted_at "2018-04-03 13:33:05"
  end
end

如果您的工厂名称是作者，型号名称是用户，那么您可以使用上面的语法

Answer 3

另一种选择是#trait在父级中使用方法。

FactoryBot.define do
  factory :post do
    user nil
    body Faker::Movie.quote
    posted_at "2018-04-03 13:33:05"
  end
end

FactoryBot.define do 
  factory :user do 
    first_name 'Jake'
  end

  trait :with_post do
    after(:create) do |user|
      create(:post, user_id: user.id)
    end
  end
end

FactoryBot.create(:user, :with_post)