Fixed that saving a model with multiple habtm associations, would only save...

Fixed that saving a model with multiple habtm associations, would only save the first habtm association.


git-svn-id: http://svn-commit.rubyonrails.org/rails/trunk@3331 5ecf4fe2-1ee6-0310-87b1-e25e094e27de

activerecord/CHANGELOG

 *SVN*
 
+* Fixed that saving a model with multiple habtm associations, would only save the first habtm association.  #3244 [yanowitz-rubyonrails@quantumfoam.org, Florian Weber]
+
 * Fix change_column to work with PostgreSQL 7.x and 8.x.  #3141 [wejn@box.cz, Rick Olson, Scott Barron] 
 
 * removed :piggyback in favor of just allowing :select on :through associations. [Tobias Luetke] 

activerecord/lib/active_record/associations.rb

@@ -727,7 +727,6 @@ def require_association_class(class_name)
         def add_multiple_associated_save_callbacks(association_name)
           method_name = "validate_associated_records_for_#{association_name}".to_sym
           define_method(method_name) do
-            @new_record_before_save = new_record?
             association = instance_variable_get("@#{association_name}")
             if association.respond_to?(:loaded?)
               if new_record?
@@ -741,6 +740,7 @@ def add_multiple_associated_save_callbacks(association_name)
           end
 
           validate method_name
+          before_save("@new_record_before_save = new_record?; true")
 
           after_callback = <<-end_eval
             association = instance_variable_get("@#{association_name}")
@@ -754,9 +754,6 @@ def add_multiple_associated_save_callbacks(association_name)
               records_to_save.each { |record| association.send(:insert_record, record) }
               association.send(:construct_sql)   # reconstruct the SQL queries now that we know the owner's id
             end
-            
-            @new_record_before_save = false
-            true
           end_eval
                 
           # Doesn't use after_save as that would save associations added in after_create/after_update twice

activerecord/test/associations_test.rb

@@ -1194,6 +1194,22 @@ def test_habtm_adding_before_save_with_join_attributes
     end
   end
 
+  def test_habtm_saving_multiple_relationships
+    new_project = Project.new("name" => "Grimetime")
+    amount_of_developers = 4
+    developers = (0..amount_of_developers).collect {|i| Developer.create(:name => "JME #{i}") }
+  
+    new_project.developer_ids = [developers[0].id, developers[1].id]
+    new_project.developers_with_callback_ids = [developers[2].id, developers[3].id]
+    assert new_project.save
+    
+    new_project.reload
+    assert_equal amount_of_developers, new_project.developers.size
+    amount_of_developers.times do |i|
+      assert_equal developers[i].name, new_project.developers[i].name
+    end
+  end
+
   def test_build
     devel = Developer.find(1)
     proj = devel.projects.build("name" => "Projekt")