42 lines
937 B
Ruby
42 lines
937 B
Ruby
module FactoryBot
|
|
class Callback
|
|
attr_reader :name
|
|
|
|
def initialize(name, block)
|
|
@name = name.to_sym
|
|
@block = block
|
|
ensure_valid_callback_name!
|
|
end
|
|
|
|
def run(instance, evaluator)
|
|
case block.arity
|
|
when 1, -1 then syntax_runner.instance_exec(instance, &block)
|
|
when 2 then syntax_runner.instance_exec(instance, evaluator, &block)
|
|
else syntax_runner.instance_exec(&block)
|
|
end
|
|
end
|
|
|
|
def ==(other)
|
|
name == other.name &&
|
|
block == other.block
|
|
end
|
|
|
|
protected
|
|
|
|
attr_reader :block
|
|
|
|
private
|
|
|
|
def ensure_valid_callback_name!
|
|
unless FactoryBot.callback_names.include?(name)
|
|
raise InvalidCallbackNameError, "#{name} is not a valid callback name. " +
|
|
"Valid callback names are #{FactoryBot.callback_names.inspect}"
|
|
end
|
|
end
|
|
|
|
def syntax_runner
|
|
@syntax_runner ||= SyntaxRunner.new
|
|
end
|
|
end
|
|
end
|