1
0
Fork 0
mirror of https://github.com/thoughtbot/factory_bot.git synced 2022-11-09 11:43:51 -05:00
thoughtbot--factory_bot/lib/factory_bot/strategy_calculator.rb

27 lines
473 B
Ruby
Raw Normal View History

module FactoryBot
2012-05-05 02:31:31 -04:00
# @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
FactoryBot::Internal.strategy_by_name(@name_or_object)
end
end
end