mirror of
https://github.com/thoughtbot/factory_bot.git
synced 2022-11-09 11:43:51 -05:00
26 lines
465 B
Ruby
26 lines
465 B
Ruby
module FactoryGirl
|
|
# @api private
|
|
class StrategyCalculator
|
|
def initialize(name_or_object)
|
|
@name_or_object = name_or_object
|
|
end
|
|
|
|
def strategy
|
|
if strategy_is_object?
|
|
@name_or_object
|
|
else
|
|
strategy_name_to_object
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def strategy_is_object?
|
|
@name_or_object.is_a?(Class)
|
|
end
|
|
|
|
def strategy_name_to_object
|
|
FactoryGirl.strategy_by_name(@name_or_object)
|
|
end
|
|
end
|
|
end
|