mongo_mapper 0.6.8 → 0.6.9

data/test/functional/associations/test_in_array_proxy.rb

@@ -0,0 +1,309 @@
+require 'test_helper'
+
+class InArrayProxyTest < Test::Unit::TestCase
+  context "description" do
+    setup do
+      class ::List
+        include MongoMapper::Document
+        key :name, String, :required => true
+      end
+
+      class ::User
+        include MongoMapper::Document
+        key :name, String, :required => true
+        key :list_ids, Array
+        many :lists, :in => :list_ids
+      end
+      User.collection.remove
+      List.collection.remove
+    end
+
+    teardown do
+      Object.send :remove_const, 'List' if defined?(::List)
+      Object.send :remove_const, 'User' if defined?(::User)
+    end
+    
+    should "default reader to empty array" do
+      User.new.lists.should == []
+    end
+
+    should "allow adding to association like it was an array" do
+      user = User.new(:name => 'John')
+      user.lists <<     List.new(:name => 'Foo1!')
+      user.lists.push   List.new(:name => 'Foo2!')
+      user.lists.concat List.new(:name => 'Foo3!')
+      user.lists.size.should == 3
+    end
+    
+    should "ignore adding duplicate ids" do
+      user = User.create(:name => 'John')
+      list = List.create(:name => 'Foo')
+      user.lists << list
+      user.lists << list
+      user.lists << list
+      
+      user.list_ids.should == [list.id]
+      user.lists.count.should == 1
+    end
+
+    should "be able to replace the association" do
+      user = User.new(:name => 'John')
+      list = List.new(:name => 'Foo')
+      user.lists = [list]
+      user.save.should be_true
+
+      user.reload
+      user.list_ids.should == [list.id]
+      user.lists.size.should == 1
+      user.lists[0].name.should == 'Foo'
+    end
+
+    context "create" do
+      setup do
+        @user = User.create(:name => 'John')
+        @list = @user.lists.create(:name => 'Foo!')
+      end
+
+      should "add id to key" do
+        @user.list_ids.should include(@list.id)
+      end
+
+      should "persist id addition to key in database" do
+        @user.reload
+        @user.list_ids.should include(@list.id)
+      end
+
+      should "add doc to association" do
+        @user.lists.should include(@list)
+      end
+
+      should "save doc" do
+        @list.should_not be_new
+      end
+    end
+
+    context "create!" do
+      setup do
+        @user = User.create(:name => 'John')
+        @list = @user.lists.create!(:name => 'Foo!')
+      end
+
+      should "add id to key" do
+        @user.list_ids.should include(@list.id)
+      end
+
+      should "persist id addition to key in database" do
+        @user.reload
+        @user.list_ids.should include(@list.id)
+      end
+
+      should "add doc to association" do
+        @user.lists.should include(@list)
+      end
+
+      should "save doc" do
+        @list.should_not be_new
+      end
+
+      should "raise exception if invalid" do
+        assert_raises(MongoMapper::DocumentNotValid) do
+          @user.lists.create!
+        end
+      end
+    end
+
+    context "Finding scoped to association" do
+      setup do
+        @user = User.create(:name => 'John')
+        @user2 = User.create(:name => 'Brandon')
+        @list1 = @user.lists.create!(:name => 'Foo 1', :position => 1)
+        @list2 = @user.lists.create!(:name => 'Foo 2', :position => 2)
+        @list3 = @user2.lists.create!(:name => 'Foo 3', :position => 1)
+      end
+
+      context "all" do
+        should "work" do
+          @user.lists.find(:all, :order => :position.asc).should == [@list1, @list2]
+          @user.lists.all(:order => :position.asc).should == [@list1, @list2]
+        end
+
+        should "work with conditions" do
+          @user.lists.find(:all, :name => 'Foo 1').should == [@list1]
+          @user.lists.all(:name => 'Foo 1').should == [@list1]
+        end
+      end
+
+      context "first" do
+        should "work" do
+          @user.lists.find(:first, :order => 'position').should == @list1
+          @user.lists.first(:order => 'position').should == @list1
+        end
+
+        should "work with conditions" do
+          @user.lists.find(:first, :position => 2).should == @list2
+          @user.lists.first(:position => 2).should == @list2
+        end
+      end
+
+      context "last" do
+        should "work" do
+          @user.lists.find(:last, :order => 'position').should == @list2
+          @user.lists.last(:order => 'position').should == @list2
+        end
+
+        should "work with conditions" do
+          @user.lists.find(:last, :position => 2, :order => 'position').should == @list2
+          @user.lists.last(:position => 2, :order => 'position').should == @list2
+        end
+      end
+
+      context "with one id" do
+        should "work for id in association" do
+          @user.lists.find(@list1.id).should == @list1
+        end
+
+        should "not work for id not in association" do
+          @user.lists.find(@list3.id).should be_nil
+        end
+
+        should "raise error when using ! and not found" do
+          assert_raises MongoMapper::DocumentNotFound do
+            @user.lists.find!(@list3.id)
+          end
+        end
+      end
+
+      context "with multiple ids" do
+        should "work for ids in association" do
+          @user.lists.find(@list1.id, @list2.id).should == [@list1, @list2]
+        end
+
+        should "not work for ids not in association" do
+          @user.lists.find(@list1.id, @list2.id, @list3.id).should == [@list1, @list2]
+        end
+      end
+
+      context "with #paginate" do
+        setup do
+          @lists = @user.lists.paginate(:per_page => 1, :page => 1, :order => 'position')
+        end
+
+        should "return total pages" do
+          @lists.total_pages.should == 2
+        end
+
+        should "return total entries" do
+          @lists.total_entries.should == 2
+        end
+
+        should "return the subject" do
+          @lists.collect(&:name).should == ['Foo 1']
+        end
+      end
+
+      context "dynamic finders" do
+        should "work with single key" do
+          @user.lists.find_by_name('Foo 1').should == @list1
+          @user.lists.find_by_name!('Foo 1').should == @list1
+          @user.lists.find_by_name('Foo 3').should be_nil
+        end
+
+        should "work with multiple keys" do
+          @user.lists.find_by_name_and_position('Foo 1', 1).should == @list1
+          @user.lists.find_by_name_and_position!('Foo 1', 1).should == @list1
+          @user.lists.find_by_name_and_position('Foo 3', 1).should be_nil
+        end
+
+        should "raise error when using ! and not found" do
+          assert_raises(MongoMapper::DocumentNotFound) do
+            @user.lists.find_by_name!('Foo 3')
+          end
+        end
+
+        context "find_or_create_by" do
+          should "not create document if found" do
+            lambda {
+              list = @user.lists.find_or_create_by_name('Foo 1')
+              list.should == @list1
+            }.should_not change { List.count }
+          end
+
+          should "create document if not found" do
+            lambda {
+              list = @user.lists.find_or_create_by_name('Home')
+              @user.lists.should include(list)
+            }.should change { List.count }
+          end
+        end
+      end
+    end
+
+    context "count" do
+      setup do
+        @user = User.create(:name => 'John')
+        @user2 = User.create(:name => 'Brandon')
+        @list1 = @user.lists.create!(:name => 'Foo 1')
+        @list2 = @user.lists.create!(:name => 'Foo 2')
+        @list3 = @user2.lists.create!(:name => 'Foo 3')
+      end
+
+      should "return number of ids" do
+        @user.lists.count.should == 2
+        @user2.lists.count.should == 1
+      end
+
+      should "return correct count when given criteria" do
+        @user.lists.count(:name => 'Foo 1').should == 1
+        @user2.lists.count(:name => 'Foo 1').should == 0
+      end
+    end
+
+    context "Removing documents" do
+      setup do
+        @user = User.create(:name => 'John')
+        @user2 = User.create(:name => 'Brandon')
+        @list1 = @user.lists.create!(:name => 'Foo 1', :position => 1)
+        @list2 = @user.lists.create!(:name => 'Foo 2', :position => 2)
+        @list3 = @user2.lists.create!(:name => 'Foo 3', :position => 1)
+      end
+
+      context "destroy_all" do
+        should "work" do
+          @user.lists.count.should == 2
+          @user.lists.destroy_all
+          @user.lists.count.should == 0
+        end
+
+        should "work with conditions" do
+          @user.lists.count.should == 2
+          @user.lists.destroy_all(:name => 'Foo 1')
+          @user.lists.count.should == 1
+        end
+      end
+
+      context "delete_all" do
+        should "work" do
+          @user.lists.count.should == 2
+          @user.lists.delete_all
+          @user.lists.count.should == 0
+        end
+
+        should "work with conditions" do
+          @user.lists.count.should == 2
+          @user.lists.delete_all(:name => 'Foo 1')
+          @user.lists.count.should == 1
+        end
+      end
+
+      should "work with nullify" do
+        @user.lists.count.should == 2
+
+        lambda {
+          @user.lists.nullify
+        }.should_not change { List.count }
+
+        @user.lists.count.should == 0
+      end
+    end
+  end
+end

data/test/functional/associations/test_many_documents_proxy.rb

@@ -22,10 +22,10 @@ class ManyDocumentsProxyTest < Test::Unit::TestCase
 
   should "be able to replace the association" do
     project = Project.new
-    project.statuses = [Status.new("name" => "ready")]
+    project.statuses = [Status.new(:name => "ready")]
     project.save.should be_true
 
-    project = project.reload
+    project.reload
     project.statuses.size.should == 1
     project.statuses[0].name.should == "ready"
   end
@@ -36,7 +36,7 @@ class ManyDocumentsProxyTest < Test::Unit::TestCase
     project.statuses.push   Status.new(:name => 'push')
     project.statuses.concat Status.new(:name => 'concat')
     
-    project = project.reload
+    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
@@ -235,84 +235,44 @@ class ManyDocumentsProxyTest < Test::Unit::TestCase
           lambda {
             status = @project1.statuses.find_or_create_by_name('Delivered')
             status.project.should == @project1
-          }.should change { Status.count }.by(1)
+          }.should change { Status.count }
         end
       end
     end
     
-    context "with :all" do
+    context "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, :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(:name => 'Complete')
-        statuses.should == [@complete]
-      end
-      
-      should "work with order" do
-        statuses = @project1.statuses.all(:order => 'name asc')
-        statuses.should == [@complete, @brand_new]
+        @project1.statuses.find(:all, :name => 'Complete').should == [@complete]
+        @project1.statuses.all(:name => 'Complete').should == [@complete]
       end
     end
     
-    context "with :first" do
+    context "first" do
       should "work" do
         @project1.statuses.find(:first, :order => 'name').should == @complete
-      end
-      
-      should "work with conditions" do
-        status = @project1.statuses.find(:first, :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(:name => 'Complete')
-        status.should == @complete
+        @project1.statuses.find(:first, :name => 'Complete').should == @complete
+        @project1.statuses.first(:name => 'Complete').should == @complete
       end
     end
     
-    context "with :last" do
+    context "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', :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', :name => 'New')
-        status.should == @brand_new
+        @project1.statuses.find(:last, :order => 'position', :name => 'New').should == @brand_new
+        @project1.statuses.last(:order => 'position', :name => 'New').should == @brand_new
       end
     end
     
@@ -384,4 +344,94 @@ class ManyDocumentsProxyTest < Test::Unit::TestCase
       project.collaborators.top.should == collaborator1
     end
   end
+  
+  context ":dependent" do
+    setup do
+      # FIXME: make use of already defined models
+      class ::Property
+        include MongoMapper::Document
+      end
+      Property.collection.remove
+
+      class ::Thing
+        include MongoMapper::Document
+        key :name, String
+      end
+      Thing.collection.remove
+    end
+
+    teardown do
+      Object.send :remove_const, 'Property' if defined?(::Property)
+      Object.send :remove_const, 'Thing' if defined?(::Thing)
+    end
+
+    context "=> destroy" do
+      setup do
+        Property.key :thing_id, ObjectId
+        Property.belongs_to :thing, :dependent => :destroy
+        Thing.many :properties, :dependent => :destroy
+
+        @thing = Thing.create(:name => "Tree")
+        @property1 = Property.create
+        @property2 = Property.create
+        @property3 = Property.create
+        @thing.properties << @property1
+        @thing.properties << @property2
+        @thing.properties << @property3
+      end
+
+      should "should destroy the associated documents" do
+        @thing.properties.count.should == 3
+        @thing.destroy
+        @thing.properties.count.should == 0
+        Property.count.should == 0
+      end
+    end
+
+    context "=> delete_all" do
+      setup do
+        Property.key :thing_id, ObjectId
+        Property.belongs_to :thing
+        Thing.has_many :properties, :dependent => :delete_all
+
+        @thing = Thing.create(:name => "Tree")
+        @property1 = Property.create
+        @property2 = Property.create
+        @property3 = Property.create
+        @thing.properties << @property1
+        @thing.properties << @property2
+        @thing.properties << @property3
+      end
+
+      should "should delete associated documents" do
+        @thing.properties.count.should == 3
+        @thing.destroy
+        @thing.properties.count.should == 0
+        Property.count.should == 0
+      end
+    end
+
+    context "=> nullify" do
+      setup do
+        Property.key :thing_id, ObjectId
+        Property.belongs_to :thing
+        Thing.has_many :properties, :dependent => :nullify
+
+        @thing = Thing.create(:name => "Tree")
+        @property1 = Property.create
+        @property2 = Property.create
+        @property3 = Property.create
+        @thing.properties << @property1
+        @thing.properties << @property2
+        @thing.properties << @property3
+      end
+
+      should "should nullify relationship but not destroy associated documents" do
+        @thing.properties.count.should == 3
+        @thing.destroy
+        @thing.properties.count.should == 0
+        Property.count.should == 3
+      end
+    end
+  end
 end

data/test/functional/associations/test_many_polymorphic_proxy.rb

@@ -134,8 +134,8 @@ class ManyPolymorphicProxyTest < Test::Unit::TestCase
       
       @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)
+      @hm2 = Message.create(:body => 'Hall the king!', :position => 2)
       @hall.messages = [@hm1, @hm2, @hm3]
       @hall.save
     end
@@ -270,7 +270,7 @@ class ManyPolymorphicProxyTest < Test::Unit::TestCase
       end
       
       should "allow overriding the order provided to the association" do
-        @lounge.messages.all(:order => 'position desc').should == [@lm2, @lm1]
+        @lounge.messages.all(:order => 'position').should == [@lm1, @lm2]
       end
       
       should "allow using conditions on association" do
@@ -326,10 +326,11 @@ class ManyPolymorphicProxyTest < Test::Unit::TestCase
     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),
-        User.new(:last_logged_in => nil),
+        AccountUser.new(:last_logged_in => nil),
         Bot.new(:last_logged_in => 1.week.ago)
       ]
       room.save

data/test/functional/associations/test_one_proxy.rb

@@ -0,0 +1,131 @@
+require 'test_helper'
+
+class OneProxyTest < Test::Unit::TestCase
+  def setup
+    @post_class = Class.new do
+      include MongoMapper::Document
+      def self.name; 'Post' end
+    end
+    
+    @author_class = Class.new do
+      include MongoMapper::Document
+      key :post_id, ObjectId
+    end
+    
+    @post_class.collection.remove
+    @author_class.collection.remove
+  end
+  
+  should "default to nil" do
+    @post_class.one :author, :class => @author_class
+    @post_class.new.author.nil?.should be_true
+  end
+  
+  should "be able to replace the association" do
+    @post_class.one :author, :class => @author_class
+    post = @post_class.new
+    author = @author_class.new
+    post.author = author
+    post.reload
+    
+    post.author.should == author
+    post.author.nil?.should be_false
+  end
+  
+  should "unset the association" do
+    @post_class.one :author, :class => @author_class
+    post = @post_class.new
+    author = @author_class.new
+    post.author = author
+    post.reload
+    
+    post.author = nil
+    post.author.nil?.should be_false
+  end
+  
+  should "work with :dependent delete" do
+    @post_class.one :author, :class => @author_class, :dependent => :delete
+    
+    post = @post_class.create
+    author = @author_class.new
+    post.author = author
+    post.reload
+    
+    @author_class.any_instance.expects(:delete).once
+    post.author = @author_class.new
+  end
+  
+  should "work with :dependent destroy" do
+    @post_class.one :author, :class => @author_class, :dependent => :destroy
+    
+    post = @post_class.create
+    author = @author_class.new
+    post.author = author
+    post.reload
+    
+    @author_class.any_instance.expects(:destroy).once
+    post.author = @author_class.new
+  end
+  
+  should "work with :dependent nullify" do
+    @post_class.one :author, :class => @author_class, :dependent => :nullify
+    
+    post = @post_class.create
+    author = @author_class.new
+    post.author = author
+    post.reload
+    
+    post.author = @author_class.new
+    
+    author.reload
+    author.post_id.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_id.should == post.id
+  end
+  
+  should "be able to create" do
+    @post_class.one :author, :class => @author_class
+    
+    post = @post_class.create
+    author = post.author.create(:name => 'John')
+    post.author.should be_instance_of(@author_class)
+    post.author.should_not be_new
+    post.author.name.should == 'John'
+    post.author.should == author
+    post.author.post_id.should == post.id
+  end
+  
+  context "#create!" do
+    setup do
+      @author_class.key :name, String, :required => true
+      @post_class.one :author, :class => @author_class
+    end
+    
+    should "raise exception if invalid" do
+      post = @post_class.create
+      assert_raises(MongoMapper::DocumentNotValid) do
+        post.author.create!
+      end
+    end
+    
+    should "work if valid" do
+      post = @post_class.create
+      author = post.author.create!(:name => 'John')
+      post.author.should be_instance_of(@author_class)
+      post.author.should_not be_new
+      post.author.name.should == 'John'
+      post.author.should == author
+      post.author.post_id.should == post.id
+    end
+  end
+end

data/test/functional/test_binary.rb

@@ -15,4 +15,19 @@ class BinaryTest < Test::Unit::TestCase
     doc = doc.reload
     doc.contents.to_s.should == ByteBuffer.new('010101').to_s
   end
+  
+  context "Saving a document with a blank binary value" do
+    setup do
+      @document = Class.new do
+        include MongoMapper::Document
+        set_collection_name 'test'
+        key :file, Binary
+      end
+      @document.collection.remove
+    end
+
+    should "not fail" do
+      assert_nothing_raised { @document.new(:file => nil).save }
+    end
+  end
 end