mongo_mapper-unstable 2009.10.11

data/test/functional/test_callbacks.rb

@@ -0,0 +1,85 @@
+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'
+        
+        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
+      @document.collection.clear
+    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,138 @@
+require 'test_helper'
+require 'models'
+
+class DirtyTest < Test::Unit::TestCase
+  def setup
+    @document = Class.new do
+      include MongoMapper::Document
+      set_collection_name 'test'
+      key :phrase, String
+    end
+    @document.collection.clear
+    
+    Status.collection.clear
+    Project.collection.clear
+  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 "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
+  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.should == {'phrase' => [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