mongo_mapper-unstable 2009.12.30 → 2010.1.4

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_embedded_proxy.rb

@@ -33,12 +33,9 @@ class ManyEmbeddedProxyTest < Test::Unit::TestCase
   end
   
   should "allow embedding arbitrarily deep" do
-    @document = Class.new do
-      include MongoMapper::Document
-      set_collection_name 'test'
+    @document = Doc do
       key :person, Person
     end
-    @document.collection.remove
     
     meg = Person.new(:name => "Meg")
     meg.child = Person.new(:name => "Steve")
@@ -80,12 +77,9 @@ class ManyEmbeddedProxyTest < Test::Unit::TestCase
 
   context "embedding many embedded documents" do
     setup do
-      @document = Class.new do
-        include MongoMapper::Document
-        set_collection_name 'test'
+      @document = Doc do
         many :people
       end
-      @document.collection.remove
     end
 
     should "persist all embedded documents" do
@@ -142,20 +136,16 @@ class ManyEmbeddedProxyTest < Test::Unit::TestCase
   
   context "extending the association" do
     setup do
-      @address_class = Class.new do
-        include MongoMapper::EmbeddedDocument
+      @address_class = EDoc do
         key :address, String
         key :city, String
         key :state, String
         key :zip, Integer
       end
       
-      @project_class = Class.new do
-        include MongoMapper::Document
+      @project_class = Doc do
         key :name, String
       end
-      
-      @project_class.collection.remove
     end
     
     should "work using a block passed to many" do

data/test/functional/associations/test_many_polymorphic_proxy.rb

@@ -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,149 @@
+require 'test_helper'
+
+class OneProxyTest < Test::Unit::TestCase
+  def setup
+    @post_class = Doc('Post')
+    @author_class = Doc do
+      key :post_id, ObjectId
+    end
+  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(:name => 'Frank')
+    post.author = author
+    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 "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
+  
+  should "work with criteria" do
+    @post_class.one :primary_author, :class => @author_class, :primary => true
+    @post_class.one :author, :class => @author_class
+    
+    post = @post_class.create
+    author = @author_class.create(:name => 'Frank', :primary => false, :post_id => post.id)
+    primary = @author_class.create(:name => 'Bill', :primary => true, :post_id => post.id)
+    post.author.should == author
+    post.primary_author.should == primary
+  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

@@ -2,12 +2,9 @@ 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'
+    klass = Doc do
       key :contents, Binary
     end
-    klass.collection.remove
     
     doc = klass.new(:contents => '010101')
     doc.save
@@ -15,4 +12,16 @@ 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 = Doc do
+        key :file, Binary
+      end
+    end
+
+    should "not fail" do
+      assert_nothing_raised { @document.new(:file => nil).save }
+    end
+  end
 end

data/test/functional/test_callbacks.rb

@@ -3,10 +3,7 @@ require 'test_helper'
 class CallbacksTest < Test::Unit::TestCase
   context "Defining and running callbacks" do
     setup do
-      @document = Class.new do
-        include MongoMapper::Document
-        set_collection_name 'test'
-        
+      @document = Doc do
         key :name, String
         
         [ :before_validation_on_create, :before_validation_on_update,
@@ -30,7 +27,6 @@ class CallbacksTest < Test::Unit::TestCase
           @history = nil
         end
       end
-      @document.collection.remove
     end
     
     should "get the order right for creating documents" do

data/test/functional/test_dirty.rb

@@ -3,12 +3,9 @@ require 'models'
 
 class DirtyTest < Test::Unit::TestCase
   def setup
-    @document = Class.new do
-      include MongoMapper::Document
-      set_collection_name 'test'
+    @document = Doc do
       key :phrase, String
     end
-    @document.collection.remove
     
     Status.collection.remove
     Project.collection.remove