mongo_mapper-unstable 2010.3.8 → 2010.06.23

data/test/functional/associations/test_many_documents_proxy.rb

@@ -5,13 +5,49 @@ class ManyDocumentsProxyTest < Test::Unit::TestCase
   def setup
     Project.collection.remove
     Status.collection.remove
+
+    @pet_class = Doc do
+      key :name, String
+      key :owner_id, ObjectId
+    end
+
+    @owner_class = Doc do
+      key :name, String
+    end
+    @owner_class.many :pets, :class => @pet_class, :foreign_key => :owner_id, :order => 'name'
   end
-  
+
   should "default reader to empty array" do
     project = Project.new
     project.statuses.should == []
   end
 
+  should "allow assignment of many associated documents using a hash" do
+    person_attributes = {
+      'name' => 'Mr. Pet Lover',
+      'pets' => [
+        {'name' => 'Jimmy', 'species' => 'Cocker Spainel'},
+        {'name' => 'Sasha', 'species' => 'Siberian Husky'},
+      ]
+    }
+
+    owner = @owner_class.new(person_attributes)
+    owner.name.should == 'Mr. Pet Lover'
+    owner.pets[0].name.should == 'Jimmy'
+    owner.pets[0].species.should == 'Cocker Spainel'
+    owner.pets[1].name.should == 'Sasha'
+    owner.pets[1].species.should == 'Siberian Husky'
+
+    owner.save.should be_true
+    owner.reload
+
+    owner.name.should == 'Mr. Pet Lover'
+    owner.pets[0].name.should == 'Jimmy'
+    owner.pets[0].species.should == 'Cocker Spainel'
+    owner.pets[1].name.should == 'Sasha'
+    owner.pets[1].species.should == 'Siberian Husky'
+  end
+
   should "allow adding to association like it was an array" do
     project = Project.new
     project.statuses <<     Status.new(:name => 'Foo1!')
@@ -20,28 +56,62 @@ class ManyDocumentsProxyTest < Test::Unit::TestCase
     project.statuses.size.should == 3
   end
 
-  should "be able to replace the association" do
-    project = Project.new
-    project.statuses = [Status.new(:name => "ready")]
-    project.save.should be_true
+  context "replacing the association" do
+    context "with objects of the class" do
+      should "work" do
+        project = Project.new
+        project.statuses = [Status.new(:name => "ready")]
+        project.save.should be_true
 
-    project.reload
-    project.statuses.size.should == 1
-    project.statuses[0].name.should == "ready"
+        project.reload
+        project.statuses.size.should == 1
+        project.statuses[0].name.should == "ready"
+      end
+    end
+
+    context "with Hashes" do
+      should "convert to objects of the class and work" do
+        project = Project.new
+        project.statuses = [{ 'name' => 'ready' }]
+        project.save.should be_true
+
+        project.reload
+        project.statuses.size.should == 1
+        project.statuses[0].name.should == "ready"
+      end
+    end
   end
-  
-  should "correctly assign foreign key when using <<, push and concat" do
-    project = Project.new
-    project.statuses <<     Status.new(:name => '<<')
-    project.statuses.push   Status.new(:name => 'push')
-    project.statuses.concat Status.new(:name => 'concat')
-    
-    project.reload
-    project.statuses[0].project_id.should == project.id
-    project.statuses[1].project_id.should == project.id
-    project.statuses[2].project_id.should == project.id
+
+  context "using <<, push and concat" do
+    context "with objects of the class" do
+      should "correctly assign foreign key" do
+        project = Project.new
+        project.statuses <<     Status.new(:name => '<<')
+        project.statuses.push   Status.new(:name => 'push')
+        project.statuses.concat Status.new(:name => 'concat')
+
+        project.reload
+        project.statuses[0].project_id.should == project.id
+        project.statuses[1].project_id.should == project.id
+        project.statuses[2].project_id.should == project.id
+      end
+    end
+
+    context "with Hashes" do
+      should "correctly convert to objects and assign foreign key" do
+        project = Project.new
+        project.statuses <<     { 'name' => '<<' }
+        project.statuses.push(  { 'name' => 'push' })
+        project.statuses.concat({ 'name' => 'concat' })
+
+        project.reload
+        project.statuses[0].project_id.should == project.id
+        project.statuses[1].project_id.should == project.id
+        project.statuses[2].project_id.should == project.id
+      end
+    end
   end
-  
+
   context "build" do
     should "assign foreign key" do
       project = Project.create
@@ -62,22 +132,35 @@ class ManyDocumentsProxyTest < Test::Unit::TestCase
       status.save!
       project.statuses.size.should == 1
     end
+
+    should "update collection without save" do
+      project = Project.create
+      project.statuses.build(:name => 'Foo')
+      project.statuses.size.should == 1
+    end
+
+    should "save built document when saving parent" do
+      project = Project.create
+      status = project.statuses.build(:name => 'Foo')
+      project.save!
+      status.should_not be_new
+    end
   end
-  
+
   context "create" do
     should "assign foreign key" do
       project = Project.create
       status = project.statuses.create(:name => 'Foo!')
       status.project_id.should == project.id
     end
-    
+
     should "save record" do
       project = Project.create
       lambda {
         project.statuses.create(:name => 'Foo!')
       }.should change { Status.count }
     end
-    
+
     should "allow passing attributes" do
       project = Project.create
       status = project.statuses.create(:name => 'Foo!')
@@ -91,27 +174,27 @@ class ManyDocumentsProxyTest < Test::Unit::TestCase
       project.statuses.size.should == 1
     end
   end
-  
+
   context "create!" do
     should "assign foreign key" do
       project = Project.create
       status = project.statuses.create!(:name => 'Foo!')
       status.project_id.should == project.id
     end
-    
+
     should "save record" do
       project = Project.create
       lambda {
         project.statuses.create!(:name => 'Foo!')
       }.should change { Status.count }
     end
-    
+
     should "allow passing attributes" do
       project = Project.create
       status = project.statuses.create!(:name => 'Foo!')
       status.name.should == 'Foo!'
     end
-    
+
     should "raise exception if not valid" do
       project = Project.create
       lambda {
@@ -126,29 +209,29 @@ class ManyDocumentsProxyTest < Test::Unit::TestCase
       project.statuses.size.should == 1
     end
   end
-  
+
   context "count" do
     should "work scoped to association" do
       project = Project.create
       3.times { project.statuses.create(:name => 'Foo!') }
-      
+
       other_project = Project.create
       2.times { other_project.statuses.create(:name => 'Foo!') }
-      
+
       project.statuses.count.should == 3
       other_project.statuses.count.should == 2
     end
-    
+
     should "work with conditions" do
       project = Project.create
       project.statuses.create(:name => 'Foo')
       project.statuses.create(:name => 'Other 1')
       project.statuses.create(:name => 'Other 2')
-      
+
       project.statuses.count(:name => 'Foo').should == 1
     end
   end
-  
+
   context "Unassociating documents" do
     setup do
       @project = Project.create
@@ -168,7 +251,7 @@ class ManyDocumentsProxyTest < Test::Unit::TestCase
       @project2.statuses.count.should == 2
       Status.count.should == 2
     end
-    
+
     should "work with destroy all and conditions" do
       @project.statuses.count.should == 2
       @project.statuses.destroy_all(:name => '1')
@@ -177,37 +260,37 @@ class ManyDocumentsProxyTest < Test::Unit::TestCase
       @project2.statuses.count.should == 2
       Status.count.should == 3
     end
-    
+
     should "work with delete all" do
       @project.statuses.count.should == 2
       @project.statuses.delete_all
       @project.statuses.count.should == 0
-      
+
       @project2.statuses.count.should == 2
       Status.count.should == 2
     end
-    
+
     should "work with delete all and conditions" do
       @project.statuses.count.should == 2
       @project.statuses.delete_all(:name => '1')
       @project.statuses.count.should == 1
-      
+
       @project2.statuses.count.should == 2
       Status.count.should == 3
     end
-    
+
     should "work with nullify" do
       @project.statuses.count.should == 2
       @project.statuses.nullify
       @project.statuses.count.should == 0
-      
+
       @project2.statuses.count.should == 2
       Status.count.should == 4
       Status.count(:name => '1').should == 2
       Status.count(:name => '2').should == 2
     end
-  end  
-  
+  end
+
   context "Finding scoped to association" do
     setup do
       @project1          = Project.new(:name => 'Project 1')
@@ -215,7 +298,7 @@ class ManyDocumentsProxyTest < Test::Unit::TestCase
       @complete          = Status.create(:name => 'Complete', :position => 2)
       @project1.statuses = [@brand_new, @complete]
       @project1.save
-      
+
       @project2          = Project.create(:name => 'Project 2')
       @in_progress       = Status.create(:name => 'In Progress')
       @archived          = Status.create(:name => 'Archived')
@@ -223,7 +306,7 @@ class ManyDocumentsProxyTest < Test::Unit::TestCase
       @project2.statuses = [@in_progress, @archived, @another_complete]
       @project2.save
     end
-    
+
     context "dynamic finders" do
       should "work with single key" do
         @project1.statuses.find_by_name('New').should == @brand_new
@@ -231,19 +314,19 @@ class ManyDocumentsProxyTest < Test::Unit::TestCase
         @project2.statuses.find_by_name('In Progress').should == @in_progress
         @project2.statuses.find_by_name!('In Progress').should == @in_progress
       end
-      
+
       should "work with multiple keys" do
         @project1.statuses.find_by_name_and_position('New', 1).should == @brand_new
         @project1.statuses.find_by_name_and_position!('New', 1).should == @brand_new
         @project1.statuses.find_by_name_and_position('New', 2).should be_nil
       end
-      
+
       should "raise error when using !" do
         lambda {
           @project1.statuses.find_by_name!('Fake')
         }.should raise_error(MongoMapper::DocumentNotFound)
       end
-      
+
       context "find_or_create_by" do
         should "not create document if found" do
           lambda {
@@ -261,75 +344,114 @@ class ManyDocumentsProxyTest < Test::Unit::TestCase
         end
       end
     end
-    
+
+    context "sexy querying" do
+      should "work with where" do
+        @project1.statuses.where(:name => 'New').all.should == [@brand_new]
+      end
+
+      should "work with sort" do
+        @project1.statuses.sort(:name).all.should == [@complete, @brand_new]
+      end
+
+      should "work with limit" do
+        @project1.statuses.sort(:name).limit(1).all.should == [@complete]
+      end
+
+      should "work with skip" do
+        @project1.statuses.sort(:name).skip(1).all.should == [@brand_new]
+      end
+
+      should "work with fields" do
+        @project1.statuses.fields(:position).all.each do |status|
+          status.position.should_not be_nil
+          status.name.should be_nil
+        end
+      end
+
+      should "work with scopes" do
+        @project1.statuses.complete.all.should == [@complete]
+      end
+
+      should "work with methods on class that return query" do
+        @project1.statuses.by_position(1).first.should == @brand_new
+      end
+
+      should "not work with methods on class that do not return query" do
+        Status.class_eval { def self.foo; 'foo' end }
+        lambda { @project1.statuses.foo }.
+          should raise_error(NoMethodError)
+      end
+    end
+
     context "all" do
       should "work" do
         @project1.statuses.all(:order => "position asc").should == [@brand_new, @complete]
       end
-      
+
       should "work with conditions" do
         @project1.statuses.all(:name => 'Complete').should == [@complete]
       end
     end
-    
+
     context "first" do
       should "work" do
         @project1.statuses.first(:order => 'name').should == @complete
       end
-      
+
       should "work with conditions" do
         @project1.statuses.first(:name => 'Complete').should == @complete
       end
     end
-    
+
     context "last" do
       should "work" do
         @project1.statuses.last(:order => "position asc").should == @complete
       end
-      
+
       should "work with conditions" do
         @project1.statuses.last(:order => 'position', :name => 'New').should == @brand_new
       end
     end
-    
+
     context "with one id" do
       should "work for id in association" do
         @project1.statuses.find(@complete.id).should == @complete
       end
-      
+
       should "not work for id not in association" do
         lambda {
           @project1.statuses.find!(@archived.id)
         }.should raise_error(MongoMapper::DocumentNotFound)
       end
     end
-    
+
     context "with multiple ids" do
       should "work for ids in association" do
         statuses = @project1.statuses.find(@brand_new.id, @complete.id)
         statuses.should == [@brand_new, @complete]
       end
-      
+
       should "not work for ids not in association" do
         assert_raises(MongoMapper::DocumentNotFound) do
           @project1.statuses.find!(@brand_new.id, @complete.id, @archived.id)
         end
       end
     end
-    
+
     context "with #paginate" do
       setup do
         @statuses = @project2.statuses.paginate(:per_page => 2, :page => 1, :order => 'name asc')
       end
-      
+
       should "return total pages" do
         @statuses.total_pages.should == 2
       end
-      
+
       should "return total entries" do
         @statuses.total_entries.should == 3
       end
-      
+
       should "return the subject" do
         @statuses.collect(&:name).should == %w(Archived Complete)
       end
@@ -344,13 +466,13 @@ class ManyDocumentsProxyTest < Test::Unit::TestCase
       status3 = Status.new(:name => "Closed")
       project.statuses = [status1, status2, status3]
       project.save
-      
+
       open_statuses = project.statuses.open
       open_statuses.should include(status1)
       open_statuses.should include(status2)
       open_statuses.should_not include(status3)
     end
-  
+
     should "work using many's :extend option" do
       project = Project.new(:name => "Some Project")
       collaborator1 = Collaborator.new(:name => "zing")
@@ -360,7 +482,7 @@ class ManyDocumentsProxyTest < Test::Unit::TestCase
       project.collaborators.top.should == collaborator1
     end
   end
-  
+
   context ":dependent" do
     setup do
       # FIXME: make use of already defined models

data/test/functional/associations/test_many_embedded_polymorphic_proxy.rb

@@ -6,7 +6,7 @@ class ManyEmbeddedPolymorphicProxyTest < Test::Unit::TestCase
     Catalog.collection.remove
     TrModels::Fleet.collection.remove
   end
-  
+
   should "default reader to empty array" do
     catalog = Catalog.new
     catalog.medias.should == []
@@ -29,12 +29,12 @@ class ManyEmbeddedPolymorphicProxyTest < Test::Unit::TestCase
     catalog.medias[0].file.should == 'video.mpg'
     catalog.medias[0].new?.should == false
   end
-  
+
   context "count" do
     should "default to 0" do
       Catalog.new.medias.count.should == 0
     end
-    
+
     should 'return correct count if any are embedded' do
       catalog = Catalog.new
       catalog.medias = [
@@ -48,8 +48,8 @@ class ManyEmbeddedPolymorphicProxyTest < Test::Unit::TestCase
       catalog.medias.count.should == 3
     end
   end
-  
-  should "store different associations" do      
+
+  should "store different associations" do
     catalog = Catalog.new
     catalog.medias = [
       Video.new('file' => 'video.mpg', 'length' => 3600),
@@ -57,7 +57,7 @@ class ManyEmbeddedPolymorphicProxyTest < Test::Unit::TestCase
       Image.new('file' => 'image.png', 'width' => 800, 'height' => 600)
     ]
     catalog.save.should be_true
-    
+
     catalog = catalog.reload
     catalog.medias.size.should == 3
     catalog.medias[0].file.should == 'video.mpg'
@@ -68,18 +68,18 @@ class ManyEmbeddedPolymorphicProxyTest < Test::Unit::TestCase
     catalog.medias[2].width.should == 800
     catalog.medias[2].height.should == 600
   end
-  
+
   context "With modularized models" do
     should "set associations correctly" do
-      fleet_attributes = { 
-        'name' => 'My Fleet', 
+      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' => '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
@@ -92,9 +92,9 @@ class ManyEmbeddedPolymorphicProxyTest < Test::Unit::TestCase
       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.transports[2].year.should == 2008
       fleet.save.should be_true
-      
+
       fleet = fleet.reload
       fleet.transports.size.should == 3
       fleet.transports[0].license_plate.should == 'GGG123'
@@ -104,31 +104,31 @@ class ManyEmbeddedPolymorphicProxyTest < Test::Unit::TestCase
       fleet.transports[1].year.should == 2001
       fleet.transports[2].license_plate.should == 'DEF123'
       fleet.transports[2].model.should == 'Honda Accord'
-      fleet.transports[2].year.should == 2008      
+      fleet.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
-    
+
       fleet = fleet.reload
       fleet.transports.size.should == 1
       fleet.transports[0].license_plate.should == 'DCU2013'
     end
-    
+
     should "store different associations" do
       fleet = TrModels::Fleet.new
       fleet.transports = [
@@ -137,7 +137,7 @@ class ManyEmbeddedPolymorphicProxyTest < Test::Unit::TestCase
         TrModels::Ambulance.new('license_plate' => 'HDD3030', 'icu' => true)
       ]
       fleet.save.should be_true
-    
+
       fleet = fleet.reload
       fleet.transports.size.should == 3
       fleet.transports[0].license_plate.should == 'ABC1223'
@@ -161,7 +161,7 @@ class ManyEmbeddedPolymorphicProxyTest < Test::Unit::TestCase
       catalog.save
       catalog.medias.visible.should == [medias[0], medias[1]]
     end
-  
+
     should "work using many's :extend option" do
       fleet = TrModels::Fleet.new
       transports = fleet.transports = [