mongo_mapper-unstable 2009.10.11

data/test/functional/associations/test_many_polymorphic_proxy.rb

@@ -0,0 +1,297 @@
+require 'test_helper'
+require 'models'
+
+class ManyPolymorphicProxyTest < Test::Unit::TestCase
+  def setup
+    Room.collection.clear
+  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)
+    
+    from_db = Room.find(room.id)
+    messages = from_db.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)
+    
+    from_db = Room.find(room.id)
+    messages = from_db.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)
+      @hm2 = Message.create(:body => 'Hall the king!', :position => 2)
+      @hm3 = Message.create(:body => 'Loungin!', :position => 3)
+      @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.find(:all, :order => "position").should == [@lm1, @lm2]
+      end
+      
+      should "work with conditions" do
+        messages = @lounge.messages.find(:all, :conditions => {:body => 'Loungin!'}, :order => "position")
+        messages.should == [@lm1]
+      end
+      
+      should "work with order" do
+        messages = @lounge.messages.find(:all, :order => 'position desc')
+        messages.should == [@lm2, @lm1]
+      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(:conditions => {: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.find(:first, :order => "position asc").should == @lm1
+      end
+      
+      should "work with conditions" do
+        message = @lounge.messages.find(:first, :conditions => {:body => 'I love loungin!'}, :order => "position asc")
+        message.should == @lm2
+      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(:conditions => {:body => 'I love loungin!'}, :order => "position asc")
+        message.should == @lm2
+      end
+    end
+    
+    context "with :last" do
+      should "work" do
+        @lounge.messages.find(:last, :order => "position asc").should == @lm2
+      end
+      
+      should "work with conditions" do
+        message = @lounge.messages.find(:last, :conditions => {:body => 'Loungin!'}, :order => "position asc")
+        message.should == @lm1
+      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(:conditions => {: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 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
+        lambda {
+          @lounge.messages.find(@lm1.id, @lm2.id, @hm2.id)
+        }.should raise_error(MongoMapper::DocumentNotFound)
+      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
+end

data/test/functional/associations/test_many_proxy.rb

@@ -0,0 +1,331 @@
+require 'test_helper'
+require 'models'
+
+class ManyProxyTest < Test::Unit::TestCase
+  def setup
+    Project.collection.clear
+    Status.collection.clear
+  end
+  
+  should "default reader to empty array" do
+    project = Project.new
+    project.statuses.should == []
+  end
+
+  should "allow adding to association like it was an array" do
+    project = Project.new
+    project.statuses <<     Status.new
+    project.statuses.push   Status.new
+    project.statuses.concat Status.new
+    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
+
+    from_db = Project.find(project.id)
+    from_db.statuses.size.should == 1
+    from_db.statuses[0].name.should == "ready"
+  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')
+    
+    from_db = Project.find(project.id)
+    from_db.statuses[0].project_id.should == project.id
+    from_db.statuses[1].project_id.should == project.id
+    from_db.statuses[2].project_id.should == project.id
+  end
+  
+  context "build" do
+    should "assign foreign key" do
+      project = Project.create
+      status = project.statuses.build
+      status.project_id.should == project.id
+    end
+
+    should "allow assigning attributes" do
+      project = Project.create
+      status = project.statuses.build(:name => 'Foo')
+      status.name.should == 'Foo'
+    end
+  end
+  
+  context "create" do
+    should "assign foreign key" do
+      project = Project.create
+      status = project.statuses.create
+      status.project_id.should == project.id
+    end
+    
+    should "save record" do
+      project = Project.create
+      lambda {
+        project.statuses.create
+      }.should change { Status.count }
+    end
+    
+    should "allow passing attributes" do
+      project = Project.create
+      status = project.statuses.create(:name => 'Foo!')
+      status.name.should == 'Foo!'
+    end
+  end
+  
+  context "count" do
+    should "work scoped to association" do
+      project = Project.create
+      3.times { project.statuses.create }
+      
+      other_project = Project.create
+      2.times { other_project.statuses.create }
+      
+      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
+      @project.statuses << Status.create(:name => '1')
+      @project.statuses << Status.create(:name => '2')
+
+      @project2 = Project.create
+      @project2.statuses << Status.create(:name => '1')
+      @project2.statuses << Status.create(:name => '2')
+    end
+
+    should "work with destroy all" do
+      @project.statuses.count.should == 2
+      @project.statuses.destroy_all
+      @project.statuses.count.should == 0
+
+      @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')
+      @project.statuses.count.should == 1
+
+      @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  
+  
+  context "Finding scoped to association" do
+    setup do
+      @project1          = Project.new(:name => 'Project 1')
+      @brand_new         = Status.create(:name => 'New', :position => 1 )
+      @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')
+      @another_complete  = Status.create(:name => 'Complete')
+      @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
+        @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', 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 {
+            status = @project1.statuses.find_or_create_by_name('New')
+            status.project.should == @project1
+            status.should == @brand_new
+          }.should_not change { Status.count }
+        end
+
+        should "create document if not found" do
+          lambda {
+            status = @project1.statuses.find_or_create_by_name('Delivered')
+            status.project.should == @project1
+          }.should change { Status.count }.by(1)
+        end
+      end
+    end
+    
+    context "with :all" do
+      should "work" do
+        @project1.statuses.find(:all, :order => "position asc").should == [@brand_new, @complete]
+      end
+      
+      should "work with conditions" do
+        statuses = @project1.statuses.find(:all, :conditions => {'name' => 'Complete'})
+        statuses.should == [@complete]
+      end
+      
+      should "work with order" do
+        statuses = @project1.statuses.find(:all, :order => 'name asc')
+        statuses.should == [@complete, @brand_new]
+      end
+    end
+
+    context "with #all" do
+      should "work" do
+        @project1.statuses.all(:order => "position asc").should == [@brand_new, @complete]
+      end
+      
+      should "work with conditions" do
+        statuses = @project1.statuses.all(:conditions => {'name' => 'Complete'})
+        statuses.should == [@complete]
+      end
+      
+      should "work with order" do
+        statuses = @project1.statuses.all(:order => 'name asc')
+        statuses.should == [@complete, @brand_new]
+      end
+    end
+    
+    context "with :first" do
+      should "work" do
+        @project1.statuses.find(:first, :order => 'name').should == @complete
+      end
+      
+      should "work with conditions" do
+        status = @project1.statuses.find(:first, :conditions => {:name => 'Complete'})
+        status.should == @complete
+      end
+    end
+    
+    context "with #first" do
+      should "work" do
+        @project1.statuses.first(:order => 'name').should == @complete
+      end
+      
+      should "work with conditions" do
+        status = @project1.statuses.first(:conditions => {:name => 'Complete'})
+        status.should == @complete
+      end
+    end
+    
+    context "with :last" do
+      should "work" do
+        @project1.statuses.find(:last, :order => "position asc").should == @complete
+      end
+      
+      should "work with conditions" do
+        status = @project1.statuses.find(:last, :order => 'position', :conditions => {:name => 'New'})
+        status.should == @brand_new
+      end
+    end
+    
+    context "with #last" do
+      should "work" do
+        @project1.statuses.last(:order => "position asc").should == @complete
+      end
+      
+      should "work with conditions" do
+        status = @project1.statuses.last(:order => 'position', :conditions => {:name => 'New'})
+        status.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
+        lambda {
+          @project1.statuses.find(@brand_new.id, @complete.id, @archived.id)
+        }.should raise_error(MongoMapper::DocumentNotFound)
+      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
+    end
+  end
+end

data/test/functional/test_associations.rb

@@ -0,0 +1,44 @@
+require 'test_helper'
+require 'models'
+
+class AssociationsTest < Test::Unit::TestCase    
+  should "allow changing class names" do
+    class AwesomeUser
+      include MongoMapper::Document
+      
+      many :posts, :class_name => 'AssociationsTest::AwesomePost', :foreign_key => :creator_id
+    end
+    AwesomeUser.collection.clear
+    
+    class AwesomeTag
+      include MongoMapper::EmbeddedDocument
+      
+      key :name, String
+      key :post_id, String
+      
+      belongs_to :post, :class_name => 'AssociationsTest::AwesomeUser'
+    end
+    
+    class AwesomePost
+      include MongoMapper::Document
+      
+      key :creator_id, String
+      
+      belongs_to :creator, :class_name => 'AssociationsTest::AwesomeUser'
+      many :tags, :class_name => 'AssociationsTest::AwesomeTag', :foreign_key => :post_id
+    end
+    
+    AwesomeUser.collection.clear
+    AwesomePost.collection.clear
+    
+    user = AwesomeUser.create
+    tag1 = AwesomeTag.new(:name => 'awesome')
+    tag2 = AwesomeTag.new(:name => 'grand')
+    post1 = AwesomePost.create(:creator => user, :tags => [tag1])
+    post2 = AwesomePost.create(:creator => user, :tags => [tag2])
+    user.posts.should == [post1, post2]
+    
+    post1_from_db = AwesomePost.find(post1.id)
+    post1_from_db.tags.should == [tag1]
+  end  
+end

data/test/functional/test_binary.rb

@@ -0,0 +1,18 @@
+require 'test_helper'
+
+class BinaryTest < Test::Unit::TestCase  
+  should "serialize and deserialize correctly" do
+    klass = Class.new do
+      include MongoMapper::Document
+      set_collection_name 'test'
+      key :contents, Binary
+    end
+    klass.collection.clear
+    
+    doc = klass.new(:contents => '010101')
+    doc.save
+    
+    doc = klass.find(doc.id)
+    doc.contents.to_s.should == ByteBuffer.new('010101').to_s
+  end
+end