mongo_mapper 0.7.6 → 0.8.0

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 == BSON::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_caching.rb

@@ -0,0 +1,76 @@
+require 'test_helper'
+
+class CachingTest < Test::Unit::TestCase
+  context "Caching" do
+    setup do
+      @klass = Class.new do
+        extend MongoMapper::Plugins
+        plugin MongoMapper::Plugins::Caching
+      end
+      @klass.stubs(:name).returns('Post')
+      @klass.any_instance.stubs(:[]).returns(nil)
+      @klass.any_instance.stubs(:[]=).returns(nil)
+    end
+
+    context "new" do
+      setup do
+        @doc = @klass.new
+        @doc.stubs(:new?).returns(true)
+      end
+
+      should "be class/new" do
+        @doc.cache_key.should == 'Post/new'
+      end
+
+      should "work with suffix" do
+        @doc.cache_key(:foo).
+          should == 'Post/new/foo'
+
+        @doc.cache_key(:foo, :bar).
+          should == 'Post/new/foo/bar'
+      end
+    end
+
+    context "not new" do
+      setup do
+        @object_id = BSON::ObjectID.new
+        @doc = @klass.new
+        @doc.stubs(:new?).returns(false)
+        @doc.stubs(:id).returns(@object_id)
+      end
+
+      context "with updated_at" do
+        setup do
+          time = Time.utc(2010, 6, 20, 8, 10, 7)
+          @doc.stubs(:[]).with(:updated_at).returns(time)
+        end
+
+        should "be class/id-timestamp" do
+          @doc.cache_key.should == "Post/#{@object_id}-20100620081007"
+        end
+
+        should "work with suffix" do
+          @doc.cache_key(:foo).
+            should == "Post/#{@object_id}-20100620081007/foo"
+
+          @doc.cache_key(:foo, :bar).
+            should == "Post/#{@object_id}-20100620081007/foo/bar"
+        end
+      end
+
+      context "without updated_at" do
+        should "be class/id" do
+          @doc.cache_key.should == "Post/#{@object_id}"
+        end
+
+        should "work with suffix" do
+          @doc.cache_key(:foo).
+            should == "Post/#{@object_id}/foo"
+
+          @doc.cache_key(:foo, :bar, :baz).
+            should == "Post/#{@object_id}/foo/bar/baz"
+        end
+      end
+    end
+  end
+end

data/test/functional/test_callbacks.rb

@@ -0,0 +1,151 @@
+require 'test_helper'
+
+module CallbacksSupport
+  def self.included base
+    base.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"
+      base.send(callback, callback_method)
+      define_method(callback_method) do
+        history << callback.to_sym
+      end
+    end
+  end
+
+  def history
+    @history ||= []
+  end
+
+  def clear_history
+    embedded_associations.each { |a| self.send(a.name).each(&:clear_history) }
+    @history = nil
+  end
+end
+
+class CallbacksTest < Test::Unit::TestCase
+  CreateCallbackOrder = [:before_validation, :before_validation_on_create, :after_validation, :before_save, :before_create, :after_create, :after_save]
+  UpdateCallbackOrder = [:before_validation, :before_validation_on_update, :after_validation, :before_save, :before_update, :after_update, :after_save]
+
+  context "Defining and running callbacks on documents" do
+    setup do
+      @document = Doc { include CallbacksSupport }
+    end
+
+    should "get the order right for creating documents" do
+      doc = @document.create(:name => 'John Nunemaker')
+      doc.history.should == CreateCallbackOrder
+    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 == UpdateCallbackOrder
+    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
+
+  context "Defining and running callbacks on many embedded documents" do
+    setup do
+      @root_class        = Doc  { include CallbacksSupport }
+      @child_class       = EDoc { include CallbacksSupport }
+      @grand_child_class = EDoc { include CallbacksSupport }
+
+      @root_class.many :children, :class => @child_class
+      @child_class.many :children, :class => @grand_child_class
+    end
+
+    should "get the order right based on root document creation" do
+      grand = @grand_child_class.new(:name => 'Grand Child')
+      child = @child_class.new(:name => 'Child', :children => [grand])
+      root  = @root_class.create(:name => 'Parent', :children => [child])
+
+      child = root.children.first
+      child.history.should == CreateCallbackOrder
+
+      grand = root.children.first.children.first
+      grand.history.should == CreateCallbackOrder
+    end
+
+    should "get the order right based on root document updating" do
+      grand = @grand_child_class.new(:name => 'Grand Child')
+      child = @child_class.new(:name => 'Child', :children => [grand])
+      root  = @root_class.create(:name => 'Parent', :children => [child])
+      root.clear_history
+      root.update_attributes(:name => 'Updated Parent')
+
+      child = root.children.first
+      child.history.should == UpdateCallbackOrder
+
+      grand = root.children.first.children.first
+      grand.history.should == UpdateCallbackOrder
+    end
+
+    should "work for before and after destroy" do
+      grand = @grand_child_class.new(:name => 'Grand Child')
+      child = @child_class.new(:name => 'Child', :children => [grand])
+      root  = @root_class.create(:name => 'Parent', :children => [child])
+      root.destroy
+      child = root.children.first
+      child.history.should include(:before_destroy)
+      child.history.should include(:after_destroy)
+
+      grand = root.children.first.children.first
+      grand.history.should include(:before_destroy)
+      grand.history.should include(:after_destroy)
+    end
+
+    should "not attempt to run callback defined on root that is not defined on embedded association" do
+      @root_class.define_callbacks :after_publish
+      @root_class.after_save { |d| d.run_callbacks(:after_publish) }
+
+      assert_nothing_raised do
+        child = @child_class.new(:name => 'Child')
+        root  = @root_class.create(:name => 'Parent', :children => [child])
+        child.history.should_not include(:after_publish)
+      end
+    end
+  end
+end

data/test/functional/test_dirty.rb

@@ -0,0 +1,163 @@
+require 'test_helper'
+
+class DirtyTest < Test::Unit::TestCase
+  def setup
+    @document = Doc { key :phrase, String }
+  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 = @document.find(doc.id)
+
+      doc.changed?.should be_false
+      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
+      project_class = Doc do
+        key :name, String
+      end
+
+      milestone_class = Doc do
+        key :project_id, ObjectId
+        key :name, String
+      end
+      milestone_class.belongs_to :project, :class => project_class
+
+      milestone = milestone_class.create(:name => 'Launch')
+      milestone.project = project_class.create(:name => 'Harmony')
+      milestone.changed?.should be_true
+      milestone.changed.should == %w(project_id)
+    end
+  end
+end

data/test/functional/test_document.rb

@@ -0,0 +1,253 @@
+require 'test_helper'
+require 'models'
+
+class DocumentTest < Test::Unit::TestCase
+  def setup
+    @document = Doc do
+      key :first_name, String
+      key :last_name, String
+      key :age, Integer
+      key :date, Date
+    end
+  end
+
+  context "array key" do
+    setup do
+      @document.key :tags, Array
+    end
+
+    should "give correct default" do
+      doc = @document.new
+      doc.tags.should == []
+    end
+
+    should "work with assignment" do
+      doc = @document.new
+      doc.tags = %w(foo bar)
+      doc.tags.should == %w(foo bar)
+    end
+
+    should "work with assignment after saving" do
+      doc = @document.new
+      doc.tags = %w(foo bar)
+      doc.save
+      doc.tags.should == %w(foo bar)
+      doc.reload.tags.should == %w(foo bar)
+    end
+
+    should "work with assignment then <<" do
+      doc = @document.new
+      doc.tags = []
+      doc.tags << "foo"
+      doc.tags.should == ["foo"]
+    end
+
+    should "work with <<" do
+      doc = @document.new
+      doc.tags << "foo"
+      doc.tags.should == ["foo"]
+    end
+
+    should "work with << then save" do
+      doc = @document.new
+      doc.tags << "foo"
+      doc.tags << "bar"
+      doc.save
+      doc.tags.should == %w(foo bar)
+      doc.reload.tags.should == %w(foo bar)
+    end
+  end
+
+  context "hash key" do
+    setup do
+      @document.key :foo, Hash
+    end
+
+    should "give correct default" do
+      doc = @document.new
+      doc.foo.should == {}
+    end
+
+    should "work with []=" do
+      doc = @document.new
+      doc.foo["quux"] = "bar"
+      doc.foo["quux"].should == "bar"
+      doc.foo.should == { "quux" => "bar" }
+    end
+
+    should "work with indifferent access" do
+      doc = @document.new
+      doc.foo = {:baz => 'bar'}
+      doc.foo[:baz].should == 'bar'
+      doc.foo['baz'].should == 'bar'
+    end
+
+    should "work with indifferent access after save" do
+      doc = @document.new
+      doc.foo = {:baz => 'bar'}
+      doc.save
+
+      doc = doc.reload
+      doc.foo[:baz].should == 'bar'
+      doc.foo['baz'].should == 'bar'
+    end
+  end
+
+  context "custom type key with default" do
+    setup do
+      @document.key :window, WindowSize, :default => WindowSize.new(600, 480)
+    end
+
+    should "default to default" do
+      doc = @document.new
+      doc.window.should == WindowSize.new(600, 480)
+
+    end
+
+    should "save and load from mongo" do
+      doc = @document.new
+      doc.save
+
+      doc = doc.reload
+      doc.window.should == WindowSize.new(600, 480)
+    end
+  end
+
+  context "key with proc default value" do
+    setup do
+      @document.key :proc_default, String, :default => lambda { return 'string' }
+    end
+
+    should "detect and run proc default" do
+      doc = @document.new
+      doc.proc_default.should == 'string'
+    end
+
+    should "save and load from mongo" do
+      doc = @document.create
+      doc = doc.reload
+      doc.proc_default.should == 'string'
+    end
+  end
+
+  should "have instance method for collection" do
+    @document.new.collection.name.should == @document.collection.name
+  end
+
+  should "have instance method for database" do
+    @document.new.database.should == @document.database
+  end
+
+  context "#destroyed?" do
+    setup do
+      @doc1 = @document.create(:first_name => 'John', :last_name => 'Nunemaker', :age => '27')
+    end
+
+    should "be true if deleted" do
+      @doc1.delete
+      assert @doc1.destroyed?
+    end
+
+    should "be true if destroyed" do
+      @doc1.destroy
+      assert @doc1.destroyed?
+    end
+
+    should "be false if not deleted or destroyed" do
+      assert ! @doc1.destroyed?
+    end
+  end
+
+  context "#persisted?" do
+    setup do
+      @doc = @document.new(:first_name => 'John', :last_name => 'Nunemaker', :age => '27')
+    end
+
+    should "be false if new" do
+      @doc.should_not be_persisted
+    end
+
+    should "be false if destroyed" do
+      @doc.save
+      @doc.destroy
+      @doc.should be_destroyed
+      @doc.should_not be_persisted
+    end
+
+    should "be true if not new or destroyed" do
+      @doc.save
+      @doc.should be_persisted
+    end
+  end
+
+  context "#reload" do
+    setup do
+      @foo_class = Doc do
+        key :name
+      end
+
+      @bar_class = EDoc do
+        key :name
+      end
+
+      @document.many :foos, :class => @foo_class
+      @document.many :bars, :class => @bar_class
+
+      @instance = @document.create({
+        :age => 39,
+        :foos => [@foo_class.new(:name => '1')],
+        :bars => [@bar_class.new(:name => '1')],
+      })
+    end
+
+    should "reload keys from the database" do
+      @instance.age = 37
+      @instance.age.should == 37
+      @instance.reload
+      @instance.age.should == 39
+    end
+
+    should "reset all associations" do
+      @instance.foos.expects(:reset).at_least_once
+      @instance.bars.expects(:reset).at_least_once
+      @instance.reload
+    end
+
+    should "reinstantiate embedded associations" do
+      @instance.reload
+      @instance.bars.first.name.should == '1'
+    end
+
+    should "return self" do
+      @instance.reload.object_id.should == @instance.object_id
+    end
+
+    should "raise DocumentNotFound if not found" do
+      @instance.destroy
+      assert_raises(MongoMapper::DocumentNotFound) { @instance.reload }
+    end
+  end
+
+  context "database has keys not defined in model" do
+    setup do
+      @id = BSON::ObjectID.new
+      @document.collection.insert({
+        :_id            => @id,
+        :first_name     => 'John',
+        :last_name      => 'Nunemaker',
+        :age            => 27,
+        :favorite_color => 'red',
+        :skills         => ['ruby', 'rails', 'javascript', 'xhtml', 'css']
+      })
+    end
+
+    should "assign all keys from database" do
+      doc = @document.find(@id)
+      doc.first_name.should == 'John'
+      doc.last_name.should == 'Nunemaker'
+      doc.age.should == 27
+      doc.favorite_color.should == 'red'
+      doc.skills.should == ['ruby', 'rails', 'javascript', 'xhtml', 'css']
+    end
+  end
+end