Question

我有一个通过用户和站点之间的关系具有has_many的模型。在sites_users连接表is_default上还有一个额外的数据，其类型为boolean，目的是允许每个用户从该用户的相关站点列表中拥有一个默认站点。

用户模型

class User < ApplicationRecord
  has_many :sites_users
  has_many :sites, through: :sites_users
  accepts_nested_attributes_for :sites_users, allow_destroy: true
  ...
end

用户工厂

factory :user do
  sequence(:email) { |n| "user_#{n}@example.com" }
  role { Role.find_by(title: 'Marketing') }
  image { Rack::Test::UploadedFile.new(Rails.root.join('spec', 'support', 'fixtures', 'user.jpg'), 'image/jpeg') }

  factory :super_admin do
    role { Role.find_by(title: "Super Admin") }
    admin true
  end

  before :create do |u|
    u.sites_users.build(site: site, is_default: true)
  end

结束

备用用户工厂方法

在用户工厂上，我也尝试了下面包含的此方法，但是找不到使用此语法包含is_default: true的方法。因此，我最终放弃了此方法，转而支持上面的before_create调用。

factory :user do
  ...
  site { site }
  ...
end

我非常感谢任何人都可以提供的任何帮助。谢谢！

模式信息

表：用户

t.string "email", default: "", null: false
t.boolean "admin", default: false
t.integer "role_id"
t.string "first_name"
t.string "last_name"

表：网站

t.string "domain", default: "", null: false
t.string "name", default: "", null: false
t.string "logo"
t.string "logo_mark"

表：sites_users

t.bigint "site_id", null: false
t.bigint "user_id", null: false
t.boolean "is_default", default: false

Answer 1

为：site_user创建工厂

factory :site_user, class: SiteUser do 
  site { site } # you could delete this line and add the site in factory :user
  is_default { false } # as specified in your DB
end

不要在site工厂内创建:user，而要使用漂亮的语法创建其关系：

factory :user do 
  ...
  sites_users { [FactoryBot.build(:site_user, is_default: true)] }
  ...
end

应该可以解决问题！

Answer 2

因此，当我处理联接表上的额外字段时，我会创建自定义方法来建立那些联接关系，而不是尝试依赖于内置方法中的rails。我已经测试了这种方法，并且可以与factory_bot一起使用。

用户模型

class User < ApplicationRecord
  ...

  def set_site(site, default = false)
    SiteUser.find_or_create_by(
      user_id: id,
      site_id: site.id
    ).update_attribute(:is_default, default)
  end

end

注意：这段代码是我使用的一个块，因此我可以通过相同的方法创建新的站点关系并更新默认值。您可以将其简化为只创建而不需要检查是否存在。

用户工厂

factory :user do
  ...

  # to relate to new site with default true
  after(:create) do |u|
    u.set_site(create(:site), true)
  end

  # to relate to existing site with default true
  after(:create) do |u|
    u.set_site(site_name, true)
  end
end

请告诉我这是否有帮助！（或者，如果有人有一个更好的默认Railsish方式也能正常工作，我很想听听它！）

Answer 3

您可能要考虑通过架构更改来解决此问题。您可以在用户表中添加default_site_id列，并将默认网站作为用户模型的单独关联进行管理。

在迁移中：

add_foreign_key :users, :sites, column: :default_site_id

在用户类别中：

class User < ApplicationRecord
  ...
  belongs_to :default_site, class_name: 'Site'
  ...

  # validate that the default site has an association to this user
  validate :default_site_id, inclusion: {in: sites.map(&:id)}, if: Proc.new {default_site_id.present?}
end

这将简化关联并确保没有用户将拥有is_default为true的多个site_users记录。在工厂中设置默认站点应该很简单。

如何添加额外的数据以加入factory_bot factory中的表

用户模型

用户工厂

备用用户工厂方法

模式信息

3 个答案:

用户模型

用户工厂