jamieorc-mongo_mapper 0.11.1.1

data/test/functional/associations/test_one_embedded_polymorphic_proxy.rb

@@ -0,0 +1,208 @@
+require 'test_helper'
+require 'models'
+
+class OneEmbeddedPolymorhpicProxyTest < Test::Unit::TestCase
+  def setup
+    @post_class = Doc('Post') do
+      key :title, String
+    end
+  end
+
+  should "default to nil" do
+    @post_class.one :author, :polymorphic => true, :class => Robot
+    @post_class.new.author.should be_nil
+  end
+
+  should "return nil instead of a proxy" do
+    @post_class.one :author, :polymorphic => true, :class => Robot
+    nil.should === @post_class.new.author
+  end
+
+  should "be able to build" do
+    @post_class.one :author, :polymorphic => true, :class => Robot
+    post = @post_class.create
+    author = post.build_author(:serial_number => "1B")
+    post.author.should be_instance_of(Robot)
+    post.author.should be_new
+    post.author.serial_number.should == '1B'
+    post.author.should == author
+    post.author.post.should == post
+  end
+
+  should "allow assignment of associated document using a hash" do
+    @post_class.one :author, :polymorphic => :true, :class => Robot
+  
+    post = @post_class.new('author' => { 'name' => 'Frank', '_type' => 'Human' })
+    post.author.name.should == 'Frank'
+    post.author.class.should == Human
+  
+    post.save.should be_true
+    post.reload
+  
+    post.author.name.should == 'Frank'
+    post.author.class.should == Human
+  end
+
+  context "replacing the association" do
+    context "with an object" do
+      setup do
+        @post_class.one :author, :polymorphic => true, :class => Robot
+        @post = @post_class.create
+        @human = Human.new(:name => 'Frank')
+      end
+
+      should "work" do
+        @post.author = @human
+        @post.save
+        @post.reload
+
+        @post.author.should == @human
+        @post.author.nil?.should be_false
+        @post.author.class.should == Human
+
+        new_human = Human.new(:name => 'Emily')
+        @post.author = new_human
+        @post.author.should == new_human
+      end
+
+      should "generate a new proxy instead of modifying the existing one" do
+        @post.author = @human
+        @post.save
+        @post.reload
+
+        @post.author.should == @human
+        @post.author.nil?.should be_false
+
+        original_author = @post.author
+        original_author.name.should == 'Frank'
+        new_human = Human.new(:name => 'Emily')
+        @post.author = new_human
+        @post.author.should == new_human
+
+        original_author.name.should == 'Frank'
+      end
+
+      should "assign _type" do
+        @post.author = @human
+        @post.author._type.should == "Human"
+      end
+    end
+
+    context "with a Hash" do
+      setup do
+        @post_class.one :author, :polymorphic => true, :class => Robot
+        @post = @post_class.create
+      end
+
+      should "convert to an object of the class and work" do
+        @post.author = {'serial_number' => '1B'}
+        @post.save
+        @post.reload
+
+        @post.author.serial_number.should == '1B'
+        @post.author.nil?.should be_false
+
+        @post.author = {'serial_number' => '2C'}
+        @post.author.serial_number.should == '2C'
+      end
+
+      should "convert to an object of _type if given" do
+        @post.author = {'name' => 'Frank', '_type' => 'Human'}
+        @post.author.name.should == 'Frank'
+        @post.author.class.should == Human
+        @post.save
+        @post.reload
+
+        @post.author.name.should == 'Frank'
+        @post.author.class.should == Human
+      end
+
+      should "assign _type" do
+        @post.author = {'name' => 'Frank', '_type' => 'Human'}
+        @post.save
+        @post.reload
+        @post.author._type.should == "Human"
+      end
+    end
+  end
+
+  should "unset the association" do
+    @post_class.one :author, :polymorphic => true, :class => Robot
+    post = @post_class.create
+    human = Human.new
+    post.update_attributes!(:author => human)
+    post.reload
+    post.author = nil
+    post.author.should == nil
+  end
+
+  should "set modularized associated models correctly" do
+    @post_class.one :author, :polymorphic => true, :class => Robot
+
+    post = @post_class.new('author' => {'_type' => 'TrModels::Ambulance', 'license_plate' => 'GGG123', 'icu' => true})
+
+    post.author.class.should == TrModels::Ambulance
+    post.author.license_plate.should == 'GGG123'
+    post.author.icu.should be_true
+    post.save.should be_true
+
+    post = post.reload
+    post.author.class.should == TrModels::Ambulance
+    post.author.license_plate.should == 'GGG123'
+    post.author.icu.should be_true
+  end
+
+  should "not have problem loading root document if embedded one is nil" do
+    @post_class.one :author, :polymorphic => true, :class => Robot
+    post = @post_class.create
+
+    lambda {
+      @post_class.find(post.id)
+    }.should_not raise_error
+  end
+
+  should "load the parent and root documents for nested embedded documents" do
+    @address_class = EDoc('Address') do
+      key :city, String
+      key :state, String
+    end
+    @author_class = EDoc('EmbeddedAuthor')
+    @author_class.one :address, :polymorphic => true, :class => @address_class
+    @post_class.one :author, :polymorphic => true, :class => @author_class
+  
+    post = @post_class.create(:title => 'Post Title', :author => { :name => 'Frank', :address => { :city => 'Boston', :state => 'MA' } })
+  
+    post.author.address._parent_document.should == post.author
+    post.author.address._root_document.should == post
+  end
+  
+  should "have boolean method for testing presence" do
+    @post_class.one :author, :polymorphic => true, :class => Robot
+  
+    post = @post_class.new
+    post.author?.should be_false
+  
+    post.author = Human.new(:name => 'Frank')
+    post.author?.should be_true
+  end
+
+  should "initialize id for nested embedded document created from hash" do
+    @address_class = EDoc('Address') do
+      key :city, String
+      key :state, String
+    end
+    @author_class = EDoc('EmbeddedAuthor')
+    @author_class.one :address, :polymorphic => true, :class => @address_class
+    @post_class.one :author, :polymorphic => true, :class => @author_class  
+
+    post = @post_class.create(:title => 'Post Title', :author => {
+      :name => 'Frank',
+      :address => {
+        :city => 'Boston',
+        :state => 'MA'
+      }
+    })
+  
+    post.author.address.id.should_not be_nil
+  end
+end

data/test/functional/associations/test_one_embedded_proxy.rb

@@ -0,0 +1,100 @@
+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.build_author(: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 "load the parent and root documents for nested embedded documents" do
+    @address_class = EDoc('Address') do
+      key :city, String
+      key :state, String
+    end
+    @author_class.one :address, :class => @address_class
+    @post_class.one :author, :class => @author_class
+
+    post = @post_class.create(:title => 'Post Title', :author => { :name => 'Frank', :address => { :city => 'Boston', :state => 'MA' } })
+
+    post.author.address._parent_document.should == post.author
+    post.author.address._root_document.should == post
+  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 "initialize id for nested embedded document created from hash" do
+    @address_class = EDoc('Address') do
+      key :city, String
+      key :state, String
+    end
+    @author_class.one(:address, :class => @address_class)
+    @post_class.one(:author, :class => @author_class)
+
+    post = @post_class.create(:title => 'Post Title', :author => {
+      :name => 'Frank',
+      :address => {
+        :city => 'Boston',
+        :state => 'MA'
+      }
+    })
+
+    post.author.address.id.should_not be_nil
+  end
+
+end

data/test/functional/associations/test_one_proxy.rb

@@ -0,0 +1,383 @@
+require 'test_helper'
+require 'models'
+
+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 "return nil instead of a proxy" do
+    @post_class.one :author, :class => @author_class
+    nil.should === @post_class.new.author
+  end
+
+  should "allow assignment of associated document using a hash" do
+    @post_class.one :author, :class => @author_class
+
+    post = @post_class.new('author' => { 'name' => 'Frank' })
+    post.author.name.should == 'Frank'
+
+    post.save.should be_true
+    post.reload
+
+    post.author.name.should == 'Frank'
+  end
+
+  context "replacing the association" do
+    context "with an object of the class" do
+      should "work" 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 "generate a new proxy instead of modifying the existing one" 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
+
+        original_author = post.author
+        original_author.name.should == 'Frank'
+        new_author = @author_class.new(:name => 'Emily')
+        post.author = new_author
+        post.author.should == new_author
+
+        original_author.name.should == 'Frank'
+      end
+    end
+
+    context "with a Hash" do
+      should "convert to an object of the class and work" do
+        @post_class.one :author, :class => @author_class
+
+        post = @post_class.new
+        post.author = {'name' => 'Frank'}
+        post.reload
+
+        post.author.name.should == 'Frank'
+        post.author.nil?.should be_false
+
+        post.author = {'name' => 'Emily'}
+        post.author.name.should == 'Emily'
+      end
+    end
+    
+    context "with :dependent" do
+      context "=> delete" do
+        setup do
+          @post_class.one :author, :class => @author_class, :dependent => :delete
+
+          @post = @post_class.create
+          @author = @author_class.new
+          @post.author = @author
+        end
+        
+        should "call delete on the existing document" do
+          @author_class.any_instance.expects(:delete).once
+          @post.author = @author_class.new
+        end
+        
+        should "remove the existing document from the database" do
+          @post.author = @author_class.new
+          lambda { @author.reload }.should raise_error(MongoMapper::DocumentNotFound)
+        end
+        
+        should "do nothing if it's the same document" do
+          @author_class.any_instance.expects(:delete).never
+          @post.author = @author
+        end        
+      end
+    
+      context "=> destory" do
+        setup do
+          @post_class.one :author, :class => @author_class, :dependent => :destroy
+
+          @post = @post_class.create
+          @author = @author_class.new
+          @post.author = @author
+        end
+
+        should "call destroy the existing document" do
+          @author_class.any_instance.expects(:destroy).once
+          @post.author = @author_class.new
+        end
+        
+        should "remove the existing document from the database" do
+          @post.author = @author_class.new
+          lambda { @author.reload }.should raise_error(MongoMapper::DocumentNotFound)
+        end
+        
+        should "do nothing if it's the same document" do
+          @author_class.any_instance.expects(:destroy).never
+          @post.author = @author
+        end        
+      end
+
+      context "=> nullify" do
+        setup do
+          @post_class.one :author, :class => @author_class, :dependent => :nullify
+
+          @post = @post_class.create
+          @author = @author_class.new
+          @post.author = @author
+        end
+        
+        should "nullify the existing document" do
+          @author.reload
+          @author.post_id.should == @post.id
+
+          @post.author = @author_class.new
+
+          @author.reload
+          @author.post_id.should be_nil
+        end
+        
+        should "work when it's the same document" do
+          old_author = @post.author
+          @post.author = @author
+          old_author.should == @post.author
+        end        
+      end
+      
+      context "unspecified" do
+        should "nullify the existing document" do
+          @post_class.one :author, :class => @author_class
+
+          post = @post_class.create
+          author = @author_class.new
+          post.author = author
+          author.reload
+          author.post_id.should == post.id
+
+          post.author = @author_class.new
+
+          author.reload
+          author.post_id.should be_nil
+        end
+      end
+    end
+    
+    context "with nil" do
+      setup do
+        @post_class.one :author, :class => @author_class
+
+        @post = @post_class.new
+        @author = @author_class.new(:name => 'Frank')        
+        @post.author = @author
+      end
+      
+      should "nullify the existing document" do
+        @post.author = nil
+        @author.reload
+        @author.post_id.should be_nil
+      end
+
+      should "set the target to nil" do
+        @post.author = nil
+        @post.author.should == nil
+      end
+    end
+  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, :primary => false
+
+    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.reload
+    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.create
+    author = @author_class.create
+    post.update_attributes!(:author => author)
+    post.reload
+    post.author = nil
+    post.author.nil?.should be_true
+  end
+  
+  context "destroying parent with :dependent" do
+    context "=> destroy" do
+      setup do
+        @post_class.one :author, :class => @author_class, :dependent => :destroy
+
+        @post = @post_class.create
+        @author = @author_class.new
+        @post.author = @author
+      end
+
+      should "should call destroy on the associated documents" do
+        @author_class.any_instance.expects(:destroy).once
+        @post.destroy
+      end
+      
+      should "should remove the associated documents" do
+        @author_class.count.should == 1
+        @post.destroy
+        @post.author.should == nil
+        @author_class.count.should == 0
+      end
+    end
+
+    context "=> delete" do
+      setup do
+        @post_class.one :author, :class => @author_class, :dependent => :delete
+
+        @post = @post_class.create
+        @author = @author_class.new
+        @post.author = @author
+      end
+
+      should "should call delete the associated documents" do
+        @author_class.any_instance.expects(:delete).once
+        @post.destroy
+      end
+
+      should "remove the associated documents" do
+        @author_class.count.should == 1
+        @post.destroy
+        @post.author.should == nil
+        @author_class.count.should == 0
+      end
+    end
+
+    context "=> nullify" do
+      should "should nullify the relationship but not destroy the associated document" do
+        @post_class.one :author, :class => @author_class, :dependent => :nullify
+
+        post = @post_class.create
+        author = @author_class.new
+        post.author = author
+
+        @author_class.count.should == 1
+        post.destroy
+        post.author.should == nil
+        @author_class.count.should == 1
+
+        @author_class.first.should == author
+        author.post_id.should == nil
+      end
+    end
+    
+    context "unspecified" do
+      should "should nullify the relationship but not destroy the associated document" do
+        @post_class.one :author, :class => @author_class
+
+        post = @post_class.create
+        author = @author_class.new
+        post.author = author
+
+        @author_class.count.should == 1
+        post.destroy
+        post.author.should == nil
+        @author_class.count.should == 1
+
+        @author_class.first.should == author
+        author.post_id.should == nil
+      end
+    end
+  end
+  
+
+  should "be able to build" do
+    @post_class.one :author, :class => @author_class
+
+    post = @post_class.create
+    author = post.build_author(: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.create_author(: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.create_author!
+      end
+    end
+
+    should "work if valid" do
+      post = @post_class.create
+      author = post.create_author!(: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
+
+  context "namespaced foreign keys" do
+    setup do
+      News::Paper.one :article, :class_name => 'News::Article'
+      News::Article.belongs_to :paper, :class_name => 'News::Paper'
+
+      @paper = News::Paper.create
+    end
+
+    should "properly infer the foreign key" do
+      article = @paper.create_article
+      article.should respond_to(:paper_id)
+      article.paper_id.should == @paper.id
+    end
+  end
+end