mongo_mapper-unstable 2010.3.8 → 2010.06.23

data/test/functional/associations/test_many_embedded_proxy.rb

@@ -11,7 +11,7 @@ class ManyEmbeddedProxyTest < Test::Unit::TestCase
       key :title, String
     end
     @post_class.many :comments, :class => @comment_class
-    
+
     @pet_class = EDoc do
       key :name, String
     end
@@ -21,17 +21,17 @@ class ManyEmbeddedProxyTest < Test::Unit::TestCase
     end
     @person_class.key :child, @person_class
     @person_class.many :pets, :class => @pet_class
-    
+
     @owner_class = Doc do
       key :name, String
     end
     @owner_class.many :pets, :class => @pet_class
   end
-    
+
   should "default reader to empty array" do
     @post_class.new.comments.should == []
   end
-  
+
   should "allow adding to association like it was an array" do
     post = @post_class.new
     post.comments << @comment_class.new
@@ -54,33 +54,33 @@ class ManyEmbeddedProxyTest < Test::Unit::TestCase
     post.comments[1].should == bill
     post.comments[1].new?.should == false
   end
-  
+
   should "allow embedding arbitrarily deep" do
     @klass = Doc()
     @klass.key :person, @person_class
-    
+
     meg             = @person_class.new(:name => 'Meg')
     meg.child       = @person_class.new(:name => 'Steve')
     meg.child.child = @person_class.new(:name => 'Linda')
-    
+
     doc = @klass.new(:person => meg)
     doc.save
     doc.reload
-    
+
     doc.person.name.should == 'Meg'
     doc.person.child.name.should == 'Steve'
     doc.person.child.child.name.should == 'Linda'
   end
-  
+
   should "allow assignment of many embedded documents using a hash" do
-    person_attributes = { 
-      'name' => 'Mr. Pet Lover', 
+    person_attributes = {
+      'name' => 'Mr. Pet Lover',
       'pets' => [
         {'name' => 'Jimmy', 'species' => 'Cocker Spainel'},
-        {'name' => 'Sasha', 'species' => 'Siberian Husky'}, 
-      ] 
+        {'name' => 'Sasha', 'species' => 'Siberian Husky'},
+      ]
     }
-    
+
     owner = @owner_class.new(person_attributes)
     owner.name.should == 'Mr. Pet Lover'
     owner.pets[0].name.should == 'Jimmy'
@@ -109,12 +109,12 @@ class ManyEmbeddedProxyTest < Test::Unit::TestCase
         @pet_class.new(:name => 'Sparky', :species => 'Dog'),
         @pet_class.new(:name => 'Koda', :species => 'Dog')
       ])
-      
+
       doc = @klass.new
       doc.people << meg
       doc.save
       doc.reload
-      
+
       doc.people.first.name.should == 'Meg'
       doc.people.first.pets.should_not == []
       doc.people.first.pets.first.name.should == 'Sparky'
@@ -127,7 +127,7 @@ class ManyEmbeddedProxyTest < Test::Unit::TestCase
       doc = @klass.new
       meg = @person_class.new(:name => 'Meg')
       pet = @pet_class.new(:name => 'Sparky', :species => 'Dog')
-      
+
       doc.people << meg
       meg.pets << pet
 
@@ -138,7 +138,7 @@ class ManyEmbeddedProxyTest < Test::Unit::TestCase
       doc = @klass.new
       meg = @person_class.new(:name => 'Meg')
       pet = @pet_class.new(:name => 'Sparky', :species => 'Dog')
-      
+
       doc.people << meg
       meg.pets << pet
 
@@ -161,7 +161,7 @@ class ManyEmbeddedProxyTest < Test::Unit::TestCase
       doc = @klass.new
       meg = @person_class.new(:name => 'Meg')
       pet = @pet_class.new(:name => 'Sparky', :species => 'Dog')
-      
+
       doc.people << meg
       meg.pets << pet
       doc.save
@@ -175,7 +175,7 @@ class ManyEmbeddedProxyTest < Test::Unit::TestCase
       doc = @klass.new
       meg = @person_class.new(:name => 'Meg')
       pet = @pet_class.new(:name => 'Sparky', :species => 'Dog')
-      
+
       doc.people << meg
       meg.pets << pet
       doc.save
@@ -184,20 +184,20 @@ class ManyEmbeddedProxyTest < Test::Unit::TestCase
       doc.people.first.pets.first.person.should == doc.people.first
     end
   end
-  
+
   should "allow finding by id" do
     sparky = @pet_class.new(:name => 'Sparky', :species => 'Dog')
     meg    = @owner_class.create(:name => 'Meg', :pets => [sparky])
-    
+
     meg.pets.find(sparky._id).should     == sparky  # oid
     meg.pets.find(sparky.id.to_s).should == sparky  # string
   end
-  
+
   context "count" do
     should "default to 0" do
       @owner_class.new.pets.count.should == 0
     end
-    
+
     should "return correct count if any are embedded" do
       owner = @owner_class.new(:name => 'Meg')
       owner.pets = [@pet_class.new, @pet_class.new]
@@ -207,7 +207,7 @@ class ManyEmbeddedProxyTest < Test::Unit::TestCase
       owner.pets.count.should == 2
     end
   end
-  
+
   context "extending the association" do
     setup do
       @address_class = EDoc do
@@ -216,27 +216,27 @@ class ManyEmbeddedProxyTest < Test::Unit::TestCase
         key :state, String
         key :zip, Integer
       end
-      
+
       @project_class = Doc do
         key :name, String
       end
     end
-    
+
     should "work using a block passed to many" do
       @project_class.many :addresses, :class => @address_class do
         def find_all_by_state(state)
           find_all { |a| a.state == state }
         end
       end
-      
+
       addr1 = @address_class.new(:address => "Gate-3 Lankershim Blvd.", :city => "Universal City", :state => "CA", :zip => "91608")
       addr2 = @address_class.new(:address => "3000 W. Alameda Ave.", :city => "Burbank", :state => "CA", :zip => "91523")
       addr3 = @address_class.new(:address => "111 Some Ln", :city => "Nashville", :state => "TN", :zip => "37211")
       project = @project_class.create(:name => "Some Project", :addresses => [addr1, addr2, addr3])
-      
+
       project.addresses.find_all_by_state("CA").should == [addr1, addr2]
     end
-  
+
     should "work using many's :extend option" do
       module FindByCity
         def find_by_city(city)
@@ -244,12 +244,12 @@ class ManyEmbeddedProxyTest < Test::Unit::TestCase
         end
       end
       @project_class.many :addresses, :class => @address_class, :extend => FindByCity
-      
+
       addr1 = @address_class.new(:address => "Gate-3 Lankershim Blvd.", :city => "Universal City", :state => "CA", :zip => "91608")
       addr2 = @address_class.new(:address => "3000 W. Alameda Ave.", :city => "Burbank", :state => "CA", :zip => "91523")
       addr3 = @address_class.new(:address => "111 Some Ln", :city => "Nashville", :state => "TN", :zip => "37211")
       project = @project_class.create(:name => "Some Project", :addresses => [addr1, addr2, addr3])
-      
+
       project.addresses.find_by_city('Burbank').should == [addr2]
     end
   end

data/test/functional/associations/test_many_polymorphic_proxy.rb

@@ -6,15 +6,15 @@ class ManyPolymorphicProxyTest < Test::Unit::TestCase
     Room.collection.remove
     Message.collection.remove
   end
-  
+
   should "default reader to empty array" do
     Room.new.messages.should == []
   end
-  
+
   should "add type key to polymorphic class base" do
     Message.keys.keys.should include('_type')
   end
-  
+
   should "allow adding to assiciation like it was an array" do
     room = Room.new
     room.messages << Enter.new
@@ -22,10 +22,10 @@ class ManyPolymorphicProxyTest < Test::Unit::TestCase
     room.messages.concat Exit.new
     room.messages.size.should == 3
   end
-  
+
   should "be able to replace the association" do
     room = Room.create(:name => 'Lounge')
-    
+
     lambda {
       room.messages = [
         Enter.new(:body => 'John entered room', :position => 1),
@@ -33,7 +33,7 @@ class ManyPolymorphicProxyTest < Test::Unit::TestCase
         Exit.new(:body => 'John exited room',   :position => 3)
       ]
     }.should change { Message.count }.by(3)
-    
+
     room = room.reload
     messages = room.messages.all :order => "position"
     messages.size.should == 3
@@ -41,89 +41,89 @@ class ManyPolymorphicProxyTest < Test::Unit::TestCase
     messages[1].body.should == 'Heyyyoooo!'
     messages[2].body.should == 'John exited room'
   end
-  
+
   should "correctly store type when using <<, push and concat" do
     room = Room.new
     room.messages <<      Enter.new(:body => 'John entered the room', :position => 1)
     room.messages.push    Exit.new(:body => 'John entered the room', :position => 2)
     room.messages.concat  Chat.new(:body => 'Holla!'             , :position => 3)
-    
+
     room = room.reload
     messages = room.messages.all :order => "position"
     messages[0]._type.should == 'Enter'
     messages[1]._type.should == 'Exit'
     messages[2]._type.should == 'Chat'
   end
-  
+
   context "build" do
     should "assign foreign key" do
       room = Room.create
       message = room.messages.build
       message.room_id.should == room._id
     end
-    
+
     should "assign _type" do
       room = Room.create
       message = room.messages.build
       message._type.should == 'Message'
     end
-    
+
     should "allow assigning attributes" do
       room = Room.create
       message = room.messages.build(:body => 'Foo!')
       message.body.should == 'Foo!'
     end
   end
-  
+
   context "create" do
     should "assign foreign key" do
       room = Room.create
       message = room.messages.create
       message.room_id.should == room._id
     end
-    
+
     should "assign _type" do
       room = Room.create
       message = room.messages.create
       message._type.should == 'Message'
     end
-    
+
     should "save record" do
       room = Room.create
       lambda {
         room.messages.create
       }.should change { Message.count }
     end
-    
+
     should "allow passing attributes" do
       room = Room.create
       message = room.messages.create(:body => 'Foo!')
       message.body.should == 'Foo!'
     end
   end
-  
+
   context "count" do
     should "work scoped to association" do
       room = Room.create
       3.times { room.messages.create }
-      
+
       other_room = Room.create
       2.times { other_room.messages.create }
-      
+
       room.messages.count.should == 3
       other_room.messages.count.should == 2
     end
-    
+
     should "work with conditions" do
       room = Room.create
       room.messages.create(:body => 'Foo')
       room.messages.create(:body => 'Other 1')
       room.messages.create(:body => 'Other 2')
-      
+
       room.messages.count(:body => 'Foo').should == 1
     end
   end
-  
+
   context "Finding scoped to association" do
     setup do
       @lounge = Room.create(:name => 'Lounge')
@@ -131,7 +131,7 @@ class ManyPolymorphicProxyTest < Test::Unit::TestCase
       @lm2 = Message.create(:body => 'I love loungin!', :position => 2)
       @lounge.messages = [@lm1, @lm2]
       @lounge.save
-      
+
       @hall = Room.create(:name => 'Hall')
       @hm1 = Message.create(:body => 'Do not fall in the hall', :position => 1)
       @hm3 = Message.create(:body => 'Loungin!', :position => 3)
@@ -139,24 +139,24 @@ class ManyPolymorphicProxyTest < Test::Unit::TestCase
       @hall.messages = [@hm1, @hm2, @hm3]
       @hall.save
     end
-    
+
     context "dynamic finders" do
       should "work with single key" do
         @lounge.messages.find_by_position(1).should == @lm1
         @hall.messages.find_by_position(2).should == @hm2
       end
-      
+
       should "work with multiple keys" do
         @lounge.messages.find_by_body_and_position('Loungin!', 1).should == @lm1
         @lounge.messages.find_by_body_and_position('Loungin!', 2).should be_nil
       end
-      
+
       should "raise error when using !" do
         lambda {
           @lounge.messages.find_by_position!(222)
         }.should raise_error(MongoMapper::DocumentNotFound)
       end
-      
+
       context "find_or_create_by" do
         should "not create document if found" do
           lambda {
@@ -175,103 +175,103 @@ class ManyPolymorphicProxyTest < Test::Unit::TestCase
         end
       end
     end
-    
+
     context "with #all" do
       should "work" do
         @lounge.messages.all(:order => "position").should == [@lm1, @lm2]
       end
-      
+
       should "work with conditions" do
         messages = @lounge.messages.all(:body => 'Loungin!', :order => "position")
         messages.should == [@lm1]
       end
-      
+
       should "work with order" do
         messages = @lounge.messages.all(:order => 'position desc')
         messages.should == [@lm2, @lm1]
       end
     end
-    
+
     context "with #first" do
       should "work" do
         @lounge.messages.first(:order => "position asc").should == @lm1
       end
-      
+
       should "work with conditions" do
         message = @lounge.messages.first(:body => 'I love loungin!', :order => "position asc")
         message.should == @lm2
       end
     end
-    
+
     context "with #last" do
       should "work" do
         @lounge.messages.last(:order => "position asc").should == @lm2
       end
-      
+
       should "work with conditions" do
         message = @lounge.messages.last(:body => 'Loungin!', :order => "position asc")
         message.should == @lm1
       end
     end
-    
+
     context "with one id" do
       should "work for id in association" do
         @lounge.messages.find(@lm2._id).should == @lm2
       end
-      
+
       should "not work for id not in association" do
         lambda {
           @lounge.messages.find!(@hm2._id)
         }.should raise_error(MongoMapper::DocumentNotFound)
       end
     end
-    
+
     context "with query options/criteria" do
       should "work with order on association" do
         @lounge.messages.should == [@lm1, @lm2]
       end
-      
+
       should "allow overriding the order provided to the association" do
         @lounge.messages.all(:order => 'position').should == [@lm1, @lm2]
       end
-      
+
       should "allow using conditions on association" do
         @hall.latest_messages.should == [@hm3, @hm2]
       end
     end
-    
+
     context "with multiple ids" do
       should "work for ids in association" do
         messages = @lounge.messages.find(@lm1._id, @lm2._id)
         messages.should == [@lm1, @lm2]
       end
-      
+
       should "not work for ids not in association" do
         assert_raises(MongoMapper::DocumentNotFound) do
           @lounge.messages.find!(@lm1._id, @lm2._id, @hm2._id)
         end
       end
     end
-    
+
     context "with #paginate" do
       setup do
         @messages = @hall.messages.paginate(:per_page => 2, :page => 1, :order => 'position asc')
       end
-      
+
       should "return total pages" do
         @messages.total_pages.should == 2
       end
-      
+
       should "return total entries" do
         @messages.total_entries.should == 3
       end
-      
+
       should "return the subject" do
         @messages.should == [@hm1, @hm2]
       end
     end
   end
-  
+
   context "extending the association" do
     should "work using a block passed to many" do
       room = Room.new(:name => "Amazing Room")
@@ -286,9 +286,9 @@ class ManyPolymorphicProxyTest < Test::Unit::TestCase
       room.save
       room.messages.older.should == messages[3..5]
     end
-  
+
     should "work using many's :extend option" do
-      
+
       room = Room.new(:name => "Amazing Room")
       accounts = room.accounts = [
         Bot.new(:last_logged_in => 3.weeks.ago),

data/test/functional/associations/test_one_embedded_proxy.rb

@@ -0,0 +1,67 @@
+require 'test_helper'
+
+class OneEmbeddedProxyTest < Test::Unit::TestCase
+  def setup
+    @post_class = Doc('Post') do
+      key :title, String
+    end
+    @author_class = EDoc('Author') do
+      key :name, String
+      embedded_in :post
+    end
+  end
+
+  should "default to nil" do
+    @post_class.one :author, :class => @author_class
+    @post_class.new.author.should be_nil
+  end
+
+  should "be able to build" do
+    @post_class.one :author, :class => @author_class
+
+    post = @post_class.create
+    author = post.author.build(:name => "John")
+    post.author.should be_instance_of(@author_class)
+    post.author.should be_new
+    post.author.name.should == 'John'
+    post.author.should == author
+    post.author.post.should == post
+  end
+
+  should "be able to replace the association" do
+    @post_class.one :author, :class => @author_class
+
+    post = @post_class.new
+    author = @author_class.new(:name => 'Frank')
+    post.author = author
+    post.save
+    post.reload
+
+    post.author.should == author
+    post.author.nil?.should be_false
+
+    new_author = @author_class.new(:name => 'Emily')
+    post.author = new_author
+    post.author.should == new_author
+  end
+
+  should "not have problem loading root document if embedded one is nil" do
+    @post_class.one :author, :class => @author_class
+    post = @post_class.create
+    
+    lambda {
+      @post_class.find(post.id)
+    }.should_not raise_error
+  end
+
+  should "have boolean method for testing presence" do
+    @post_class.one :author, :class => @author_class
+
+    post = @post_class.new
+    post.author?.should be_false
+
+    post.author = @author_class.new(:name => 'Frank')
+    post.author?.should be_true
+  end
+
+end