Support for _parent_document tracking on EmbeddedDocuments added to a…

… collection

lib/mongomapper/associations/many_embedded_proxy.rb

@@ -6,6 +6,18 @@ def replace(v)
         reset
       end
 
+      def <<(*docs)
+        if load_target
+          parent = @owner._parent_document || @owner
+          docs.each do |doc|
+            doc._parent_document = parent
+            @target << doc
+          end
+        end
+      end
+      alias_method :push, :<<
+      alias_method :concat, :<<
+
       protected
         def find_target
           (@_values || []).map do |e|

test/functional/associations/test_many_embedded_proxy.rb

@@ -107,6 +107,19 @@ def setup
       from_db.people.first.pets[1].species.should == "Dog"
     end
 
+    should "create a reference to the parent document for all embedded documents before save" do
+      meg = Person.new(:name => "Meg")
+      sparky = Pet.new(:name => "Sparky", :species => "Dog")
+
+      doc = @document.new
+
+      doc.people << meg
+      meg.pets << sparky
+
+      doc.people.first._parent_document.should == doc
+      doc.people.first.pets.first._parent_document.should == doc
+    end
+
     should "create a reference to the parent document for all embedded documents" do
       meg = Person.new(:name => "Meg")
       sparky = Pet.new(:name => "Sparky", :species => "Dog")