mirror of
https://github.com/thoughtbot/factory_bot.git
synced 2022-11-09 11:43:51 -05:00
5ccc564923
If a new trait is defined with attributes, check if attributes matches the name of the trait. Raise an error if so, to avoid a recursive call to the trait. Co-authored-by: Daniel Colsen <daniel.j.colson@thoughtbot.com> Co-authored-by: Paras Sanghavi <sanghaviparas@gmail.com>
33 lines
657 B
Ruby
33 lines
657 B
Ruby
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
|