如何在ruby on rails中的factorybot的“回調”函數中訪問子類而不是父類？

Question

我在 ROR 中實施了 STI。 請看下面的代碼：

class Category < ApplicationRecord
end

class CourseCategory < Category
  has_many :courses
end

我使用 FactoryBot 創建這樣的數據：

# category

FactoryBot.define do
  factory :category do
    name {"Ruby on Rails"}
  end
end

# course_category

FactoryBot.define do
  factory :course_category, parent: :category, class: 'Category' do
    trait :with_course do
      after(:create) do |course_category| # my problem is here
      create :course, :with_steps, course_category: course_category
    end
  end
end

# course

FactoryBot.define do
  factory :course do
     course_category
  end
end

當我運行FactoryBot.create(:course_category, :with_course) ，進入trait :with_course ，我應該有CourseCategory類，但我收到Category 。

我可以訪問父類而不是子類到 FactoryBot 的回調中嗎？

Answer 1

您的工廠正在將:course_category為錯誤的類。

這條線需要改變...

factory :course_category, parent: :category, class: 'Category' do

進入...

factory :course_category, parent: :category, class: 'CourseCategory' do