FactoryBot 一个与许多属性关联的记录

Question

我有User、Assignment和Department模型。 我的Assignment模型与User模型有 2 个belongs_to关联，即requestor和assignee。

User和Assignment模式反过来也belongs_to一个部门。

我想对我的Assignment运行验证，以便assignment、requestor和assignee都属于同一个部门。

这是我的模型代码

# app/models/assignment.rb
class Assignment < ApplicationRecord
belongs_to :requestor, class_name: "User", foreign_key: "requested_by"
belongs_to :assignee, class_name: "User", foreign_key: "assigned_to"
belongs_to :department
end
# app/models/user.rb
class User < ApplicationRecord
has_many :user_departments, class_name: "UserDepartment"
has_many :departments, through: :user_departments
belongs_to :department
end
# app/models/department.rb
class Department < ApplicationRecord
has_many :user_departments, class_name: "UserDepartment"
has_many :users, through: :user_departments
has_many :assignments
has_many :users
end

这是我测试的代码

# spec/factories/assignment.rb
FactoryBot.define do
sequence(:title) { |n| "Title #{n}" }
sequence(:description) { |n| "Description #{n}" }
factory :assignment do
title
description
image { Rack::Test::UploadedFile.new(Rails.root.join('spec/support/Floyd.jpeg'), 'image/jpeg') }
release_date { DateTime.now >> 1 }
department { create(:department) }
requestor create(:user, :bm, department: "I want to use the same department as created in earlier line here")
assignee create(:user, :mu, department: "I want to use the same department as created in earlier line here")
end
end

我应该在哪里定义部门，以便我可以在所有三个关联中使用相同的部门。

Answer 1

你试过self.department吗？它在我的代码中起作用：

# spec/factories/assignment.rb
FactoryBot.define do
sequence(:title) { |n| "Title #{n}" }
sequence(:description) { |n| "Description #{n}" }
factory :assignment do
title
description
image { Rack::Test::UploadedFile.new(Rails.root.join('spec/support/Floyd.jpeg'), 'image/jpeg') }
release_date { DateTime.now >> 1 }
department { create(:department) }
requestor { create(:user, :bm, department: self.department) }
assignee { create(:user, :mu, department: self.department) }
end
end

Answer 2

希望我答对了。您希望所有模型具有相同的部门。 这是您在规范中实际执行的操作。

describe Foo::Bar do
before(:each) do
@department = create(:deparment)
@assignment = create(:assignment, department: @department)
@user = create(:user, department: department)
# ....
end

但这始终取决于您的整体结构如何。当assignment是中心模型并且始终有一个部门时，您可以做类似的事情

describe Foo::Bar do
before(:each) do
@assignment = create(:assignment)
@user = create(:user, department: @assignment.department)
# ....
end