djsun-mongomapper 0.3.5.5 → 0.4.1.2

data/test/functional/associations/test_many_embedded_polymorphic_proxy.rb

@@ -1,131 +1,131 @@
-require 'test_helper'
-require 'models'
-
-class ManyEmbeddedPolymorphicProxyTest < Test::Unit::TestCase
-  def setup
-    clear_all_collections
-  end
-  
-  should "default reader to empty array" do
-    Catalog.new.medias.should == []
-    Catalog.new.medias.inspect.should == "[]"
-  end
-
-  should "allow adding to association like it was an array" do
-    catalog = Catalog.new
-    catalog.medias << Video.new
-    catalog.medias.push Video.new
-    catalog.medias.size.should == 2
-  end
-
-  should "be able to replace the association" do
-    catalog = Catalog.new
-    catalog.medias = [Video.new("file" => "video.mpg", "length" => 3600)]
-    catalog.save.should be_true
-
-    from_db = Catalog.find(catalog.id)
-    from_db.medias.size.should == 1
-    from_db.medias[0].file.should == "video.mpg"
-  end
-
-  should "store different associations" do      
-    catalog = Catalog.new
-    catalog.medias = [
-      Video.new("file" => "video.mpg", "length" => 3600),
-      Music.new("file" => "music.mp3", "bitrate" => "128kbps"),
-      Image.new("file" => "image.png", "width" => 800, "height" => 600)
-    ]
-    catalog.save.should be_true
-    
-    from_db = Catalog.find(catalog.id)
-    from_db.medias.size.should == 3
-    from_db.medias[0].file.should == "video.mpg"
-    from_db.medias[0].length.should == 3600
-    from_db.medias[1].file.should == "music.mp3"
-    from_db.medias[1].bitrate.should == "128kbps"
-    from_db.medias[2].file.should == "image.png"
-    from_db.medias[2].width.should == 800
-    from_db.medias[2].height.should == 600
-  end
-  
-  context "With modularized models" do
-    should "set associations correctly" do
-      fleet_attributes = { 
-        "name" => "My Fleet", 
-        "transports" => [
-          {"_type" => "TrModels::Ambulance", "license_plate" => "GGG123", "icu" => true},
-          {"_type" => "TrModels::Car", "license_plate" => "ABC123", "model" => "VW Golf", "year" => 2001}, 
-          {"_type" => "TrModels::Car", "license_plate" => "DEF123", "model" => "Honda Accord", "year" => 2008},
-        ] 
-      }
-      
-      fleet = TrModels::Fleet.new(fleet_attributes)
-      fleet.transports.size.should == 3
-      fleet.transports[0].class.should == TrModels::Ambulance
-      fleet.transports[0].license_plate.should == "GGG123"
-      fleet.transports[0].icu.should be_true
-      fleet.transports[1].class.should == TrModels::Car
-      fleet.transports[1].license_plate.should == "ABC123"
-      fleet.transports[1].model.should == "VW Golf"
-      fleet.transports[1].year.should == 2001
-      fleet.transports[2].class.should == TrModels::Car
-      fleet.transports[2].license_plate.should == "DEF123"
-      fleet.transports[2].model.should == "Honda Accord"
-      fleet.transports[2].year.should == 2008      
-      fleet.save.should be_true
-      
-      from_db = TrModels::Fleet.find(fleet.id)
-      from_db.transports.size.should == 3
-      from_db.transports[0].license_plate.should == "GGG123"
-      from_db.transports[0].icu.should be_true
-      from_db.transports[1].license_plate.should == "ABC123"
-      from_db.transports[1].model.should == "VW Golf"
-      from_db.transports[1].year.should == 2001
-      from_db.transports[2].license_plate.should == "DEF123"
-      from_db.transports[2].model.should == "Honda Accord"
-      from_db.transports[2].year.should == 2008      
-    end
-    
-    should "default reader to empty array" do
-      TrModels::Fleet.new.transports.should == []
-      TrModels::Fleet.new.transports.inspect.should == "[]"
-    end
-    
-    should "allow adding to association like it was an array" do
-      fleet = TrModels::Fleet.new
-      fleet.transports << TrModels::Car.new
-      fleet.transports.push TrModels::Bus.new
-      fleet.transports.size.should == 2
-    end
-    
-    should "be able to replace the association" do
-      fleet = TrModels::Fleet.new
-      fleet.transports = [TrModels::Car.new("license_plate" => "DCU2013", "model" => "Honda Civic")]
-      fleet.save.should be_true
-    
-      from_db = TrModels::Fleet.find(fleet.id)
-      from_db.transports.size.should == 1
-      from_db.transports[0].license_plate.should == "DCU2013"
-    end
-    
-    should "store different associations" do
-      fleet = TrModels::Fleet.new
-      fleet.transports = [
-        TrModels::Car.new("license_plate" => "ABC1223", "model" => "Honda Civic", "year" => 2003),
-        TrModels::Bus.new("license_plate" => "XYZ9090", "max_passengers" => 51),
-        TrModels::Ambulance.new("license_plate" => "HDD3030", "icu" => true)
-      ]
-      fleet.save.should be_true
-    
-      from_db = TrModels::Fleet.find(fleet.id)
-      from_db.transports.size.should == 3
-      from_db.transports[0].license_plate.should == "ABC1223"
-      from_db.transports[0].model.should == "Honda Civic"
-      from_db.transports[0].year.should == 2003
-      from_db.transports[1].license_plate.should == "XYZ9090"
-      from_db.transports[1].max_passengers.should == 51
-      from_db.transports[2].license_plate.should == "HDD3030"
-      from_db.transports[2].icu.should == true
-    end
-  end
+require 'test_helper'
+require 'models'
+
+class ManyEmbeddedPolymorphicProxyTest < Test::Unit::TestCase
+  def setup
+    clear_all_collections
+  end
+  
+  should "default reader to empty array" do
+    catalog = Catalog.new
+    catalog.medias.should == []
+  end
+
+  should "allow adding to association like it was an array" do
+    catalog = Catalog.new
+    catalog.medias << Video.new
+    catalog.medias.push Video.new
+    catalog.medias.size.should == 2
+  end
+
+  should "be able to replace the association" do
+    catalog = Catalog.new
+    catalog.medias = [Video.new("file" => "video.mpg", "length" => 3600)]
+    catalog.save.should be_true
+
+    from_db = Catalog.find(catalog.id)
+    from_db.medias.size.should == 1
+    from_db.medias[0].file.should == "video.mpg"
+  end
+
+  should "store different associations" do      
+    catalog = Catalog.new
+    catalog.medias = [
+      Video.new("file" => "video.mpg", "length" => 3600),
+      Music.new("file" => "music.mp3", "bitrate" => "128kbps"),
+      Image.new("file" => "image.png", "width" => 800, "height" => 600)
+    ]
+    catalog.save.should be_true
+    
+    from_db = Catalog.find(catalog.id)
+    from_db.medias.size.should == 3
+    from_db.medias[0].file.should == "video.mpg"
+    from_db.medias[0].length.should == 3600
+    from_db.medias[1].file.should == "music.mp3"
+    from_db.medias[1].bitrate.should == "128kbps"
+    from_db.medias[2].file.should == "image.png"
+    from_db.medias[2].width.should == 800
+    from_db.medias[2].height.should == 600
+  end
+  
+  context "With modularized models" do
+    should "set associations correctly" do
+      fleet_attributes = { 
+        "name" => "My Fleet", 
+        "transports" => [
+          {"_type" => "TrModels::Ambulance", "license_plate" => "GGG123", "icu" => true},
+          {"_type" => "TrModels::Car", "license_plate" => "ABC123", "model" => "VW Golf", "year" => 2001}, 
+          {"_type" => "TrModels::Car", "license_plate" => "DEF123", "model" => "Honda Accord", "year" => 2008},
+        ] 
+      }
+      
+      fleet = TrModels::Fleet.new(fleet_attributes)
+      fleet.transports.size.should == 3
+      fleet.transports[0].class.should == TrModels::Ambulance
+      fleet.transports[0].license_plate.should == "GGG123"
+      fleet.transports[0].icu.should be_true
+      fleet.transports[1].class.should == TrModels::Car
+      fleet.transports[1].license_plate.should == "ABC123"
+      fleet.transports[1].model.should == "VW Golf"
+      fleet.transports[1].year.should == 2001
+      fleet.transports[2].class.should == TrModels::Car
+      fleet.transports[2].license_plate.should == "DEF123"
+      fleet.transports[2].model.should == "Honda Accord"
+      fleet.transports[2].year.should == 2008      
+      fleet.save.should be_true
+      
+      from_db = TrModels::Fleet.find(fleet.id)
+      from_db.transports.size.should == 3
+      from_db.transports[0].license_plate.should == "GGG123"
+      from_db.transports[0].icu.should be_true
+      from_db.transports[1].license_plate.should == "ABC123"
+      from_db.transports[1].model.should == "VW Golf"
+      from_db.transports[1].year.should == 2001
+      from_db.transports[2].license_plate.should == "DEF123"
+      from_db.transports[2].model.should == "Honda Accord"
+      from_db.transports[2].year.should == 2008      
+    end
+    
+    should "default reader to empty array" do
+      fleet = TrModels::Fleet.new
+      fleet.transports.should == []
+    end
+    
+    should "allow adding to association like it was an array" do
+      fleet = TrModels::Fleet.new
+      fleet.transports << TrModels::Car.new
+      fleet.transports.push TrModels::Bus.new
+      fleet.transports.size.should == 2
+    end
+    
+    should "be able to replace the association" do
+      fleet = TrModels::Fleet.new
+      fleet.transports = [TrModels::Car.new("license_plate" => "DCU2013", "model" => "Honda Civic")]
+      fleet.save.should be_true
+    
+      from_db = TrModels::Fleet.find(fleet.id)
+      from_db.transports.size.should == 1
+      from_db.transports[0].license_plate.should == "DCU2013"
+    end
+    
+    should "store different associations" do
+      fleet = TrModels::Fleet.new
+      fleet.transports = [
+        TrModels::Car.new("license_plate" => "ABC1223", "model" => "Honda Civic", "year" => 2003),
+        TrModels::Bus.new("license_plate" => "XYZ9090", "max_passengers" => 51),
+        TrModels::Ambulance.new("license_plate" => "HDD3030", "icu" => true)
+      ]
+      fleet.save.should be_true
+    
+      from_db = TrModels::Fleet.find(fleet.id)
+      from_db.transports.size.should == 3
+      from_db.transports[0].license_plate.should == "ABC1223"
+      from_db.transports[0].model.should == "Honda Civic"
+      from_db.transports[0].year.should == 2003
+      from_db.transports[1].license_plate.should == "XYZ9090"
+      from_db.transports[1].max_passengers.should == 51
+      from_db.transports[2].license_plate.should == "HDD3030"
+      from_db.transports[2].icu.should == true
+    end
+  end
 end

data/test/functional/associations/test_many_embedded_proxy.rb

@@ -1,107 +1,169 @@
-require 'test_helper'
-require 'models'
-
-class ManyEmbeddedProxyTest < Test::Unit::TestCase
-  def setup
-    clear_all_collections
-  end
-  
-  should "default reader to empty array" do
-    Project.new.addresses.should == []
-    Project.new.addresses.inspect.should == "[]"
-  end
-  
-  should "allow adding to association like it was an array" do
-    project = Project.new
-    project.addresses << Address.new
-    project.addresses.push Address.new
-    project.addresses.size.should == 2
-  end
-
-  should "be embedded in document on save" do
-    sb = Address.new(:city => 'South Bend', :state => 'IN')
-    chi = Address.new(:city => 'Chicago', :state => 'IL')
-    project = Project.new
-    project.addresses << sb
-    project.addresses << chi
-    project.save
-
-    from_db = Project.find(project.id)
-    from_db.addresses.size.should == 2
-    from_db.addresses[0].should == sb
-    from_db.addresses[1].should == chi
-  end
-  
-  should "allow embedding arbitrarily deep" do
-    @document = Class.new do
-      include MongoMapper::Document
-      key :person, Person
-    end
-    @document.collection.clear
-    
-    meg = Person.new(:name => "Meg")
-    meg.child = Person.new(:name => "Steve")
-    meg.child.child = Person.new(:name => "Linda")
-    
-    doc = @document.new(:person => meg)
-    doc.save
-    
-    from_db = @document.find(doc.id)
-    from_db.person.name.should == 'Meg'
-    from_db.person.child.name.should == 'Steve'
-    from_db.person.child.child.name.should == 'Linda'
-  end
-  
-  should "allow assignment of 'many' embedded documents using a hash" do
-    person_attributes = { 
-      "name" => "Mr. Pet Lover", 
-      "pets" => [
-        {"name" => "Jimmy", "species" => "Cocker Spainel"},
-        {"name" => "Sasha", "species" => "Siberian Husky"}, 
-      ] 
-    }
-    
-    pet_lover = RealPerson.new(person_attributes)
-    pet_lover.name.should == "Mr. Pet Lover"
-    pet_lover.pets[0].name.should == "Jimmy"
-    pet_lover.pets[0].species.should == "Cocker Spainel"
-    pet_lover.pets[1].name.should == "Sasha"
-    pet_lover.pets[1].species.should == "Siberian Husky"
-    pet_lover.save.should be_true
-    
-    from_db = RealPerson.find(pet_lover.id)
-    from_db.name.should == "Mr. Pet Lover"
-    from_db.pets[0].name.should == "Jimmy"
-    from_db.pets[0].species.should == "Cocker Spainel"
-    from_db.pets[1].name.should == "Sasha"
-    from_db.pets[1].species.should == "Siberian Husky"
-  end
-  
-  should "allow saving embedded documents in 'many' embedded documents" do
-    @document = Class.new do
-      include MongoMapper::Document
-      many :people
-    end
-    @document.collection.clear
-    
-    meg = Person.new(:name => "Meg")
-    sparky = Pet.new(:name => "Sparky", :species => "Dog")
-    koda = Pet.new(:name => "Koda", :species => "Dog")
-    
-    doc = @document.new
-    
-    meg.pets << sparky
-    meg.pets << koda
-    
-    doc.people << meg
-    doc.save
-    
-    from_db = @document.find(doc.id)
-    from_db.people.first.name.should == "Meg"
-    from_db.people.first.pets.should_not == []
-    from_db.people.first.pets.first.name.should == "Sparky"
-    from_db.people.first.pets.first.species.should == "Dog"
-    from_db.people.first.pets[1].name.should == "Koda"
-    from_db.people.first.pets[1].species.should == "Dog"
-  end
+require 'test_helper'
+require 'models'
+
+class ManyEmbeddedProxyTest < Test::Unit::TestCase
+  def setup
+    clear_all_collections
+  end
+  
+  should "default reader to empty array" do
+    Project.new.addresses.should == []
+  end
+  
+  should "allow adding to association like it was an array" do
+    project = Project.new
+    project.addresses << Address.new
+    project.addresses.push Address.new
+    project.addresses.size.should == 2
+  end
+  
+  should "allow finding :all embedded documents" do
+    project = Project.new
+    project.addresses << Address.new
+    project.addresses << Address.new
+    project.save
+  end
+
+  should "be embedded in document on save" do
+    sb = Address.new(:city => 'South Bend', :state => 'IN')
+    chi = Address.new(:city => 'Chicago', :state => 'IL')
+    project = Project.new
+    project.addresses << sb
+    project.addresses << chi
+    project.save
+
+    from_db = Project.find(project.id)
+    from_db.addresses.size.should == 2
+    from_db.addresses[0].should == sb
+    from_db.addresses[1].should == chi
+  end
+  
+  should "allow embedding arbitrarily deep" do
+    @document = Class.new do
+      include MongoMapper::Document
+      key :person, Person
+    end
+    
+    meg = Person.new(:name => "Meg")
+    meg.child = Person.new(:name => "Steve")
+    meg.child.child = Person.new(:name => "Linda")
+    
+    doc = @document.new(:person => meg)
+    doc.save
+    
+    from_db = @document.find(doc.id)
+    from_db.person.name.should == 'Meg'
+    from_db.person.child.name.should == 'Steve'
+    from_db.person.child.child.name.should == 'Linda'
+  end
+  
+  should "allow assignment of 'many' embedded documents using a hash" do
+    person_attributes = { 
+      "name" => "Mr. Pet Lover", 
+      "pets" => [
+        {"name" => "Jimmy", "species" => "Cocker Spainel"},
+        {"name" => "Sasha", "species" => "Siberian Husky"}, 
+      ] 
+    }
+    
+    pet_lover = RealPerson.new(person_attributes)
+    pet_lover.name.should == "Mr. Pet Lover"
+    pet_lover.pets[0].name.should == "Jimmy"
+    pet_lover.pets[0].species.should == "Cocker Spainel"
+    pet_lover.pets[1].name.should == "Sasha"
+    pet_lover.pets[1].species.should == "Siberian Husky"
+    pet_lover.save.should be_true
+    
+    from_db = RealPerson.find(pet_lover.id)
+    from_db.name.should == "Mr. Pet Lover"
+    from_db.pets[0].name.should == "Jimmy"
+    from_db.pets[0].species.should == "Cocker Spainel"
+    from_db.pets[1].name.should == "Sasha"
+    from_db.pets[1].species.should == "Siberian Husky"
+  end
+
+  context "embedding many embedded documents" do
+    setup do
+      @document = Class.new do
+        include MongoMapper::Document
+        many :people
+      end
+    end
+
+    should "persist all embedded documents" do
+      meg = Person.new(:name => "Meg")
+      sparky = Pet.new(:name => "Sparky", :species => "Dog")
+      koda = Pet.new(:name => "Koda", :species => "Dog")
+
+      doc = @document.new
+
+      meg.pets << sparky
+      meg.pets << koda
+
+      doc.people << meg
+      doc.save
+
+      from_db = @document.find(doc.id)
+      from_db.people.first.name.should == "Meg"
+      from_db.people.first.pets.should_not == []
+      from_db.people.first.pets.first.name.should == "Sparky"
+      from_db.people.first.pets.first.species.should == "Dog"
+      from_db.people.first.pets[1].name.should == "Koda"
+      from_db.people.first.pets[1].species.should == "Dog"
+    end
+
+    should "create a reference to the root 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._root_document.should == doc
+      doc.people.first.pets.first._root_document.should == doc
+    end
+
+    should "create properly-named reference to parent document when building off association proxy" do
+      person = RealPerson.new
+      pet = person.pets.build
+      person.should == pet.real_person
+    end
+
+
+    should "create a reference to the root document for all embedded documents" do
+      meg = Person.new(:name => "Meg")
+      sparky = Pet.new(:name => "Sparky", :species => "Dog")
+
+      doc = @document.new
+
+      meg.pets << sparky
+
+      doc.people << meg
+      doc.save
+
+      from_db = @document.find(doc.id)
+      from_db.people.first._root_document.should == doc
+      from_db.people.first.pets.first._root_document.should == doc
+    end
+  end
+  
+  should "allow retrieval via find(:all)" do
+    meg = Person.new(:name => "Meg")
+    sparky = Pet.new(:name => "Sparky", :species => "Dog")
+
+    meg.pets << sparky
+    
+    meg.pets.find(:all).should include(sparky)
+  end
+  
+  should "allow retrieval via find(id)" do
+    meg = Person.new(:name => "Meg")
+    sparky = Pet.new(:name => "Sparky", :species => "Dog")
+
+    meg.pets << sparky
+    
+    meg.pets.find(sparky.id).should == sparky
+  end
 end