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_girl/strategy_calculator.rb

27 lines
507 B
Ruby
Raw Normal View History

module FactoryGirl
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) ||
raise("unrecognized method #{@name_or_object}")
end
end
end