jmonteiro-mongo_mapper 0.1.0

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_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.remove
+    
+    class AwesomeTag
+      include MongoMapper::EmbeddedDocument
+      
+      key :name, String
+      key :post_id, ObjectId
+      
+      belongs_to :post, :class_name => 'AssociationsTest::AwesomeUser'
+    end
+    
+    class AwesomePost
+      include MongoMapper::Document
+      
+      key :creator_id, ObjectId
+      
+      belongs_to :creator, :class_name => 'AssociationsTest::AwesomeUser'
+      many :tags, :class_name => 'AssociationsTest::AwesomeTag', :foreign_key => :post_id
+    end
+    
+    AwesomeUser.collection.remove
+    AwesomePost.collection.remove
+    
+    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 = post1.reload
+    post1.tags.should == [tag1]
+  end
+end

data/test/functional/test_binary.rb

@@ -0,0 +1,27 @@
+require 'test_helper'
+
+class BinaryTest < Test::Unit::TestCase  
+  should "serialize and deserialize correctly" do
+    klass = Doc do
+      key :contents, Binary
+    end
+    
+    doc = klass.new(:contents => '010101')
+    doc.save
+    
+    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

@@ -0,0 +1,81 @@
+require 'test_helper'
+
+class CallbacksTest < Test::Unit::TestCase
+  context "Defining and running callbacks" do
+    setup do
+      @document = Doc do
+        key :name, String
+        
+        [ :before_validation_on_create, :before_validation_on_update,
+          :before_validation, :after_validation,
+          :before_create,     :after_create, 
+          :before_update,     :after_update,
+          :before_save,       :after_save,
+          :before_destroy,    :after_destroy].each do |callback|
+          callback_method = "#{callback}_callback"
+          send(callback, callback_method)
+          define_method(callback_method) do
+            history << callback.to_sym
+          end
+        end
+        
+        def history
+          @history ||= []
+        end
+        
+        def clear_history
+          @history = nil
+        end
+      end
+    end
+    
+    should "get the order right for creating documents" do
+      doc = @document.create(:name => 'John Nunemaker')
+      doc.history.should == [:before_validation, :before_validation_on_create, :after_validation, :before_save, :before_create, :after_create, :after_save]
+    end
+    
+    should "get the order right for updating documents" do
+      doc = @document.create(:name => 'John Nunemaker')
+      doc.clear_history
+      doc.name = 'John'
+      doc.save
+      doc.history.should == [:before_validation, :before_validation_on_update, :after_validation, :before_save, :before_update, :after_update, :after_save]
+    end
+    
+    should "work for before and after validation" do
+      doc = @document.new(:name => 'John Nunemaker')
+      doc.valid?
+      doc.history.should include(:before_validation)
+      doc.history.should include(:after_validation)
+    end
+    
+    should "work for before and after create" do
+      doc = @document.create(:name => 'John Nunemaker')
+      doc.history.should include(:before_create)
+      doc.history.should include(:after_create)
+    end
+    
+    should "work for before and after update" do
+      doc = @document.create(:name => 'John Nunemaker')
+      doc.name = 'John Doe'
+      doc.save
+      doc.history.should include(:before_update)
+      doc.history.should include(:after_update)
+    end
+    
+    should "work for before and after save" do
+      doc = @document.new
+      doc.name = 'John Doe'
+      doc.save
+      doc.history.should include(:before_save)
+      doc.history.should include(:after_save)
+    end
+    
+    should "work for before and after destroy" do
+      doc = @document.create(:name => 'John Nunemaker')
+      doc.destroy
+      doc.history.should include(:before_destroy)
+      doc.history.should include(:after_destroy)
+    end
+  end
+end

data/test/functional/test_dirty.rb

@@ -0,0 +1,156 @@
+require 'test_helper'
+require 'models'
+
+class DirtyTest < Test::Unit::TestCase
+  def setup
+    @document = Doc do
+      key :phrase, String
+    end
+    
+    Status.collection.remove
+    Project.collection.remove
+  end
+  
+  context "marking changes" do
+    should "not happen if there are none" do
+      doc = @document.new
+      doc.phrase_changed?.should be_false
+      doc.phrase_change.should be_nil
+    end
+    
+    should "happen when change happens" do
+      doc = @document.new
+      doc.phrase = 'Golly Gee Willikers Batman'
+      doc.phrase_changed?.should be_true
+      doc.phrase_was.should be_nil
+      doc.phrase_change.should == [nil, 'Golly Gee Willikers Batman']
+    end
+    
+    should "happen when initializing" do
+      doc = @document.new(:phrase => 'Foo')
+      doc.changed?.should be_true
+    end
+    
+    should "clear changes on save" do
+      doc = @document.new
+      doc.phrase = 'Golly Gee Willikers Batman'
+      doc.phrase_changed?.should be_true
+      doc.save
+      doc.phrase_changed?.should_not be_true
+      doc.phrase_change.should be_nil
+    end
+    
+    should "clear changes on save!" do
+      doc = @document.new
+      doc.phrase = 'Golly Gee Willikers Batman'
+      doc.phrase_changed?.should be_true
+      doc.save!
+      doc.phrase_changed?.should_not be_true
+      doc.phrase_change.should be_nil
+    end
+    
+    should "not happen when loading from database" do
+      doc = @document.create(:phrase => 'Foo')
+      doc.phrase = 'Fart'
+      doc.changed?.should be_true
+      doc.reload
+      doc.changed?.should be_false
+    end
+    
+    should "happen if changed after loading from database" do
+      doc = @document.create(:phrase => 'Foo')
+      doc.reload
+      doc.changed?.should be_false
+      doc.phrase = 'Bar'
+      doc.changed?.should be_true
+    end
+  end
+  
+  context "blank new value and type integer" do
+    should "not mark changes" do
+      @document.key :age, Integer
+
+      [nil, ''].each do |value|
+        doc = @document.new
+        doc.age = value
+        doc.age_changed?.should be_false
+        doc.age_change.should be_nil
+      end
+    end
+  end
+  
+  context "blank new value and type float" do
+    should "not mark changes" do
+      @document.key :amount, Float
+
+      [nil, ''].each do |value|
+        doc = @document.new
+        doc.amount = value
+        doc.amount_changed?.should be_false
+        doc.amount_change.should be_nil
+      end
+    end
+  end  
+  
+  context "changed?" do
+    should "be true if key changed" do
+      doc = @document.new
+      doc.phrase = 'A penny saved is a penny earned.'
+      doc.changed?.should be_true
+    end
+    
+    should "be false if no keys changed" do
+      @document.new.changed?.should be_false
+    end
+  end
+  
+  context "changes" do
+    should "be empty hash if no changes" do
+      @document.new.changes.should == {}
+    end
+    
+    should "be hash of keys with values of changes if there are changes" do
+      doc = @document.new
+      doc.phrase = 'A penny saved is a penny earned.'
+      doc.changes['phrase'].should == [nil, 'A penny saved is a penny earned.']
+    end
+  end
+  
+  context "changed" do
+    should "be empty array if no changes" do
+      @document.new.changed.should == []
+    end
+    
+    should "be array of keys that have changed if there are changes" do
+      doc = @document.new
+      doc.phrase = 'A penny saved is a penny earned.'
+      doc.changed.should == ['phrase']
+    end
+  end
+  
+  context "will_change!" do
+    should "mark changes" do
+      doc = @document.create(:phrase => 'Foo')
+      
+      doc.phrase << 'bar'
+      doc.phrase_changed?.should be_false
+      
+      doc.phrase_will_change!
+      doc.phrase_changed?.should be_true
+      doc.phrase_change.should == ['Foobar', 'Foobar']
+      
+      doc.phrase << '!'
+      doc.phrase_changed?.should be_true
+      doc.phrase_change.should == ['Foobar', 'Foobar!']
+    end
+  end
+  
+  context "changing a foreign key through association" do
+    should "mark changes" do
+      status = Status.create(:name => 'Foo')
+      status.project = Project.create(:name => 'Bar')
+      status.changed?.should be_true
+      status.changed.should == %w(project_id)
+    end
+  end
+end