Added a case for build_class to handle class names with underscores passed as a string (#1642)

Co-authored-by: Michelle Yuen <[email protected]>
Co-authored-by: Stephanie Minn <[email protected]>

Author: 3 people

lib/factory_bot/factory.rb

@@ -22,6 +22,8 @@ def initialize(name, options = {})
     def build_class
       @build_class ||= if class_name.is_a? Class
         class_name
+      elsif class_name.to_s.safe_constantize
+        class_name.to_s.safe_constantize
       else
         class_name.to_s.camelize.constantize
       end

spec/factory_bot/factory_spec.rb

@@ -215,6 +215,15 @@ def self.to_s
 
     expect(factory.name).to eq name
   end
+
+  it "sets build_class correctly with a class with an underscore" do
+    name = :settings
+    define_class("Admin_Settings_1")
+    settings_class = Admin_Settings_1
+    factory = FactoryBot::Factory.new(name, class: "Admin_Settings_1")
+
+    expect(factory.build_class).to eq settings_class
+  end
 end
 
 describe FactoryBot::Factory, "for namespaced class" do