diff --git a/spec/support/missing_factory_helper.rb b/spec/support/missing_factory_helper.rb index 97f196272b7..5c6385d03d3 100644 --- a/spec/support/missing_factory_helper.rb +++ b/spec/support/missing_factory_helper.rb @@ -3,11 +3,11 @@ module Spec module Support module MissingFactoryHelper def build(factory, *args, &block) - registered_factory_symbols.include?(factory) ? super : class_from_symbol(factory).new(*args) + factory_exists?(factory) ? super : class_from_symbol(factory).new(*args) end def create(factory, *args, &block) - registered_factory_symbols.include?(factory) ? super : class_from_symbol(factory).create!(*args) + factory_exists?(factory) ? super : class_from_symbol(factory).create!(*args) end private @@ -16,6 +16,10 @@ def class_from_symbol(symbol) symbol.to_s.classify.constantize end + def factory_exists?(factory) + registered_factory_symbols.include?(factory.to_sym) + end + def registered_factory_symbols @registered_factory_symbols ||= begin require 'set'