Adjust find_factories_by_class to work with Strings instead of Classes

This commit is contained in:
Sebastian Serth
2021-02-12 02:00:35 +01:00
parent 635b9aa16f
commit 1ed9a8f0b6

View File

@ -1,6 +1,6 @@
def find_factories_by_class(klass) def find_factories_by_class(klass)
FactoryBot.factories.select do |factory| FactoryBot.factories.select do |factory|
factory.instance_variable_get(:@class_name) == klass || factory.instance_variable_get(:@name) == klass.model_name.singular.to_sym factory.instance_variable_get(:@class_name).to_s == klass.to_s || factory.instance_variable_get(:@name) == klass.model_name.singular.to_sym
end end
end end