drogus-mongo_mapper 0.6.10

data/test/functional/associations/test_many_embedded_proxy.rb

@@ -0,0 +1,256 @@
+require 'test_helper'
+require 'models'
+
+class ManyEmbeddedProxyTest < Test::Unit::TestCase
+  def setup
+    @comment_class = EDoc do
+      key :name, String
+      key :body, String
+    end
+    @post_class = Doc do
+      key :title, String
+    end
+    @post_class.many :comments, :class => @comment_class
+    
+    @pet_class = EDoc do
+      key :name, String
+    end
+    @pet_class.embedded_in :person
+    @person_class = EDoc do
+      key :name, String
+    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
+    post.comments.push @comment_class.new
+    post.comments.size.should == 2
+  end
+
+  should "be embedded in document on save" do
+    frank = @comment_class.new(:name => 'Frank', :body => 'Hi!')
+    bill = @comment_class.new(:name => 'Bill', :body => 'Hi!')
+    post = @post_class.new
+    post.comments << frank
+    post.comments << bill
+    post.save
+
+    post.reload
+    post.comments.size.should == 2
+    post.comments[0].should == frank
+    post.comments[0].new?.should == false
+    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', 
+      '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
+
+  context "embedding many embedded documents" do
+    setup do
+      @klass = Doc()
+      @klass.many :people, :class => @person_class
+    end
+
+    should "persist all embedded documents" do
+      meg = @person_class.new(:name => 'Meg', :pets => [
+        @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'
+      doc.people.first.pets.first.species.should == 'Dog'
+      doc.people.first.pets[1].name.should == 'Koda'
+      doc.people.first.pets[1].species.should == 'Dog'
+    end
+
+    should "create a reference to the root document for all embedded documents before save" do
+      doc = @klass.new
+      meg = @person_class.new(:name => 'Meg')
+      pet = @pet_class.new(:name => 'Sparky', :species => 'Dog')
+      
+      doc.people << meg
+      meg.pets << pet
+
+      doc.people.first._root_document.should == doc
+      doc.people.first.pets.first._root_document.should == doc
+    end
+    should "create a reference to the owning document for all embedded documents before save" do
+      doc = @klass.new
+      meg = @person_class.new(:name => 'Meg')
+      pet = @pet_class.new(:name => 'Sparky', :species => 'Dog')
+      
+      doc.people << meg
+      meg.pets << pet
+
+      doc.people.first._parent_document.should == doc
+      doc.people.first.pets.first._parent_document.should == doc.people.first
+    end
+
+    should "create a reference to the root document for all embedded documents" do
+      sparky = @pet_class.new(:name => 'Sparky', :species => 'Dog')
+      meg    = @person_class.new(:name => 'Meg', :pets => [sparky])
+      doc    = @klass.new
+      doc.people << meg
+      doc.save
+
+      doc.reload
+      doc.people.first._root_document.should == doc
+      doc.people.first.pets.first._root_document.should == doc
+    end
+    should "create a reference to the owning document for all embedded documents" do
+      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
+
+      doc.reload
+      doc.people.first._parent_document.should == doc
+      doc.people.first.pets.first._parent_document.should == doc.people.first
+    end
+
+    should "create embedded_in relationship for embedded docs" do
+      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
+
+      doc.reload
+      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]
+      owner.pets.count.should == 2
+      owner.save
+      owner.reload
+      owner.pets.count.should == 2
+    end
+  end
+  
+  context "extending the association" do
+    setup do
+      @address_class = EDoc do
+        key :address, String
+        key :city, String
+        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)
+          find_all { |a| a.city == city }
+        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
+end

data/test/functional/associations/test_many_polymorphic_proxy.rb

@@ -0,0 +1,302 @@
+require 'test_helper'
+require 'models'
+
+class ManyPolymorphicProxyTest < Test::Unit::TestCase
+  def setup
+    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
+    room.messages.push Exit.new
+    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),
+        Chat.new(:body => 'Heyyyoooo!',         :position => 2),
+        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
+    messages[0].body.should == 'John entered room'
+    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')
+      @lm1 = Message.create(:body => 'Loungin!', :position => 1)
+      @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)
+      @hm2 = Message.create(:body => 'Hall the king!', :position => 2)
+      @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 {
+            message = @lounge.messages.find_or_create_by_body('Loungin!')
+            message.room.should == @lounge
+            message.should == @lm1
+          }.should_not change { Message.count }
+        end
+
+        should "create document if not found" do
+          lambda {
+            message = @lounge.messages.find_or_create_by_body('Yo dawg!')
+            message.room.should == @lounge
+            message._type.should == 'Message'
+          }.should change { Message.count }.by(1)
+        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")
+      messages = room.messages = [
+        Enter.new(:body => 'John entered room',  :position => 3),
+        Chat.new(:body => 'Heyyyoooo!',          :position => 4),
+        Exit.new(:body => 'John exited room',    :position => 5),
+        Enter.new(:body => 'Steve entered room', :position => 6),
+        Chat.new(:body => 'Anyone there?',       :position => 7),
+        Exit.new(:body => 'Steve exited room',   :position => 8)
+      ]
+      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),
+        AccountUser.new(:last_logged_in => nil),
+        Bot.new(:last_logged_in => 1.week.ago)
+      ]
+      room.save
+      room.accounts.inactive.should == [accounts[1]]
+    end
+  end
+end