Contents

# frozen_string_literal: true

require "factory_bot"
require "factory_bot_rails/generator"
require "factory_bot_rails/reloader"
require "factory_bot_rails/factory_validator"
require "rails"

module FactoryBotRails
  class Railtie < Rails::Railtie
    config.factory_bot = ActiveSupport::OrderedOptions.new
    config.factory_bot.definition_file_paths = FactoryBot.definition_file_paths
    config.factory_bot.reject_primary_key_attributes = true
    config.factory_bot.validator = FactoryBotRails::FactoryValidator.new

    initializer "factory_bot.set_fixture_replacement" do
      Generator.new(config).run
    end

    initializer "factory_bot.set_factory_paths" do
      FactoryBot.definition_file_paths = definition_file_paths
    end

    ActiveSupport.on_load :active_record do
      config = Rails.configuration.factory_bot

      if config.reject_primary_key_attributes
        require "factory_bot_rails/factory_validator/active_record_validator"

        config.validator.add_validator FactoryValidator::ActiveRecordValidator.new
      end
    end

    config.after_initialize do |app|
      FactoryBot.find_definitions
      Reloader.new(app).run
      app.config.factory_bot.validator.run
    end

    private

    def definition_file_paths
      config.factory_bot.definition_file_paths.map do |path|
        Rails.root.join(path)
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
factory_bot_rails-6.4.0	lib/factory_bot_rails/railtie.rb
factory_bot_rails-6.3.0	lib/factory_bot_rails/railtie.rb