Merge pull request #26 from itsameandrea/bugfix/associated-object-und…

…erscore

Fix generated associated_object_path for CamelCase class names

lib/generators/associated/associated_generator.rb

@@ -8,6 +8,7 @@ def generate_associated_object_files
 
   def connect_associated_object
     record_file = "#{destination_root}/app/models/#{record_path}.rb"
+
     raise "Record class '#{record_klass}' does not exist" unless File.exist?(record_file)
 
     inject_into_class record_file, record_klass do
@@ -20,8 +21,8 @@ def connect_associated_object
   # The `:name` argument can handle model names, but associated object class names aren't singularized.
   # So these record and associated_object methods prevent that.
   def record_path  = record_klass.downcase.underscore
-  def record_klass = name.deconstantize
+  def record_klass = name.camelize.deconstantize
 
-  def associated_object_path  = associated_object_class.downcase.underscore
-  def associated_object_class = name.demodulize
+  def associated_object_path  = associated_object_class.underscore
+  def associated_object_class = name.camelize.demodulize
 end

test/lib/generators/associated_generator_test.rb

@@ -42,6 +42,28 @@ class Organization
     RUBY
   end
 
+  test "connects record: Camelized name" do
+    run_generator ["Organization::SeatsManager"]
+
+    assert_file "app/models/organization.rb", <<~RUBY
+      class Organization
+        has_object :seats_manager
+      end
+    RUBY
+  end
+
+  test "connects record: lower_snake_case name" do
+    run_generator ["organization/seats_manager"]
+
+    assert_file "app/models/organization.rb", <<~RUBY
+      class Organization
+        has_object :seats_manager
+      end
+    RUBY
+  end
+
+
+
   test "raises error if associated record doesn't exist" do
     assert_raise RuntimeError do
       run_generator ["Business::Monkey"]