Contents

module FactoryBot
  # @api private
  class Trait
    attr_reader :name, :definition

    def initialize(name, &block)
      @name = name.to_s
      @block = block
      @definition = Definition.new(@name)
      proxy = FactoryBot::DefinitionProxy.new(@definition)

      if block_given?
        proxy.instance_eval(&@block)
      end
    end

    delegate :add_callback, :declare_attribute, :to_create, :define_trait, :constructor,
      :callbacks, :attributes, to: :@definition

    def names
      [@name]
    end

    def ==(other)
      name == other.name &&
        block == other.block
    end

    protected

    attr_reader :block
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
factory_bot-6.1.0	lib/factory_bot/trait.rb
factory_bot-6.0.2	lib/factory_bot/trait.rb
factory_bot-6.0.1	lib/factory_bot/trait.rb
factory_bot-6.0.0	lib/factory_bot/trait.rb