mongoid-locomotive 2.0.0.beta9

data/spec/unit/mongoid/persistence/insert_embedded_spec.rb

@@ -0,0 +1,154 @@
+require "spec_helper"
+
+describe Mongoid::Persistence::Insert do
+
+  let(:document) do
+    Patient.new(:title => "Mr")
+  end
+
+  let(:address) do
+    Address.new(:street => "Oxford St")
+  end
+
+  let(:collection) do
+    stub.quacks_like(Mongoid::Collection.allocate)
+  end
+
+  let(:email) do
+    Email.new(:address => "test@example.com")
+  end
+
+  before do
+    document.stubs(:collection).returns(collection)
+  end
+
+  describe "#persist" do
+
+    def root_insert_expectation
+      lambda {
+        collection.expects(:insert).with(
+          document.raw_attributes,
+          :safe => true
+        ).returns("Object")
+      }
+    end
+
+    def root_push_expectation
+      lambda {
+        collection.expects(:update).with(
+          { "_id" => document.id },
+          { "$push" => { "addresses" => address.raw_attributes } },
+          :multi => false,
+          :safe => true
+        ).returns("Object")
+      }
+    end
+
+    def root_set_expectation
+      lambda {
+        collection.expects(:update).with(
+          { "_id" => document.id },
+          { "$set" => { "email" => email.raw_attributes } },
+          :multi => false,
+          :safe => true
+        ).returns("Object")
+      }
+    end
+
+    context "when the embedded document is an embeds_one" do
+
+      before do
+        document.email = email
+      end
+
+      context "when the parent is new" do
+
+        let(:insert) do
+          Mongoid::Persistence::InsertEmbedded.new(email)
+        end
+
+        it "notifies its changes to parent and inserts the parent" do
+          root_insert_expectation.call
+          insert.persist.should == email
+        end
+      end
+
+      context "when the parent is not new" do
+
+        let(:insert) do
+          Mongoid::Persistence::InsertEmbedded.new(email)
+        end
+
+        before do
+          document.instance_variable_set(:@new_record, false)
+        end
+
+        it "performs an in place $set on the embedded document" do
+          root_set_expectation.call
+          insert.persist.should == email
+        end
+
+        it "sets new_record to false" do
+          root_set_expectation.call
+          insert.persist.new_record?.should be_false
+        end
+      end
+    end
+
+    context "when the embedded document is an embeds_many" do
+
+      before do
+        document.addresses << address
+      end
+
+      context "when the parent is new" do
+
+        let(:insert) do
+          Mongoid::Persistence::InsertEmbedded.new(address)
+        end
+
+        it "notifies its changes to the parent and inserts the parent" do
+          root_insert_expectation.call
+          insert.persist.should == address
+        end
+
+        it "sets new_record to false" do
+          root_insert_expectation.call
+          insert.persist.new_record?.should be_false
+        end
+      end
+
+      context "when the parent is not new" do
+
+        let(:insert) do
+          Mongoid::Persistence::InsertEmbedded.new(address)
+        end
+
+        before do
+          document.instance_variable_set(:@new_record, false)
+        end
+
+        it "performs a $push on the embedded array" do
+          root_push_expectation.call
+          insert.persist.should == address
+        end
+
+        context "when we add the parent to the child" do
+          let(:other_address) do
+            Address.new(:street => "Oxford St", :addressable => document)
+          end
+
+          it "performs a $push on the embedded array" do
+            collection.expects(:update).with(
+              { "_id" => document.id },
+              { "$push" => { "addresses" => other_address.raw_attributes } },
+              :multi => false,
+              :safe => true
+            ).returns("Object")
+            Mongoid::Persistence::InsertEmbedded.new(other_address).persist.should == other_address
+          end
+        end
+      end
+    end
+  end
+end

data/spec/unit/mongoid/persistence/insert_spec.rb

@@ -0,0 +1,144 @@
+require "spec_helper"
+
+describe Mongoid::Persistence::Insert do
+
+  let(:document) do
+    Patient.new(:title => "Mr")
+  end
+
+  let(:address) do
+    Address.new(:street => "Oxford St")
+  end
+
+  let(:collection) do
+    stub.quacks_like(Mongoid::Collection.allocate)
+  end
+
+  before do
+    document.stubs(:collection).returns(collection)
+  end
+
+  describe "#initialize" do
+
+    let(:insert) do
+      Mongoid::Persistence::Insert.new(document)
+    end
+
+    it "sets the document" do
+      insert.document.should == document
+    end
+
+    it "sets the collection" do
+      insert.collection.should == document.collection
+    end
+
+    it "defaults validation to true" do
+      insert.validate.should == true
+    end
+
+    it "sets the options" do
+      insert.options.should ==
+        { :safe => Mongoid.persist_in_safe_mode }
+    end
+  end
+
+  describe "#persist" do
+
+    def root_set_expectation
+      lambda {
+        collection.expects(:insert).with(
+          document.raw_attributes,
+          :safe => true
+        ).returns("Object")
+      }
+    end
+
+    def root_push_expectation
+      lambda {
+        collection.expects(:update).with(
+          { "_id" => document.id },
+          { "addresses" => { "$push" => address.raw_attributes } },
+          :safe => true
+        ).returns("Object")
+      }
+    end
+
+    let(:insert) do
+      Mongoid::Persistence::Insert.new(document)
+    end
+
+    context "when the document is valid" do
+
+      it "inserts the new document" do
+        root_set_expectation.call
+        insert.persist
+      end
+
+      it "returns the document" do
+        root_set_expectation.call
+        insert.persist.should == document
+      end
+
+      it "sets new_record to false" do
+        root_set_expectation.call
+        insert.persist
+        document.new_record?.should == false
+      end
+    end
+
+    context "when the document is not valid" do
+
+      before do
+        document.stubs(:valid?).returns(false)
+      end
+
+      it "returns the document" do
+        insert.persist.should == document
+      end
+
+      it "leaves the document as a new record" do
+        insert.persist
+        document.new_record?.should == true
+      end
+    end
+
+    context "when not validating" do
+
+      before do
+        insert.instance_variable_set(:@validate, false)
+        document.stubs(:valid?).returns(false)
+      end
+
+      after do
+        insert.instance_variable_set(:@validate, true)
+      end
+
+      it "inserts the document in the database" do
+        root_set_expectation.call
+        insert.persist
+        document.new_record?.should == false
+      end
+    end
+
+    context "when the document is embedded" do
+
+      before do
+        document.addresses << address
+      end
+
+      let(:insert) do
+        Mongoid::Persistence::Insert.new(address)
+      end
+
+      let(:persister) do
+        stub.quacks_like(Mongoid::Persistence::InsertEmbedded.allocate)
+      end
+
+      it "delegates to the embedded persister" do
+        Mongoid::Persistence::InsertEmbedded.expects(:new).with(address, true).returns(persister)
+        persister.expects(:persist).returns(address)
+        insert.persist.should == address
+      end
+    end
+  end
+end

data/spec/unit/mongoid/persistence/remove_all_spec.rb

@@ -0,0 +1,82 @@
+require "spec_helper"
+
+describe Mongoid::Persistence::RemoveAll do
+
+  let(:document) do
+    Patient.new(:title => "Mr")
+  end
+
+  let(:collection) do
+    stub_everything.quacks_like(Mongoid::Collection.allocate)
+  end
+
+  let(:cursor) do
+    stub.quacks_like(Mongoid::Cursor.allocate)
+  end
+
+  let(:selector) do
+    { :field1 => { "$exists" => true } }
+  end
+
+  before do
+    Patient.stubs(:_collection).returns(collection)
+  end
+
+  describe "#initialize" do
+
+    let(:remove_all) do
+      Mongoid::Persistence::RemoveAll.new(Patient, false, selector)
+    end
+
+    it "sets the collection" do
+      remove_all.collection.should == document.collection
+    end
+
+    it "sets the options" do
+      remove_all.options.should ==
+        { :safe => Mongoid.persist_in_safe_mode }
+    end
+
+    it "sets the selector" do
+      remove_all.selector.should == selector
+    end
+  end
+
+  describe "#persist" do
+
+    def root_delete_expectation
+      lambda {
+        collection.expects(:remove).with(selector, :safe => true).returns(true)
+      }
+    end
+
+    def root_find_expectation
+      lambda {
+        collection.expects(:find).with(
+          selector
+        ).returns(cursor)
+      }
+    end
+
+    let(:remove_all) do
+      Mongoid::Persistence::RemoveAll.new(Patient, false, selector)
+    end
+
+    context "when the document is a root document" do
+
+      it "remove_alls the document from the collection" do
+        root_delete_expectation.call
+        root_find_expectation.call
+        cursor.expects(:count).returns(30)
+        remove_all.persist
+      end
+
+      it "returns the count of documents removed" do
+        root_delete_expectation.call
+        root_find_expectation.call
+        cursor.expects(:count).returns(30)
+        remove_all.persist.should == 30
+      end
+    end
+  end
+end

data/spec/unit/mongoid/persistence/remove_embedded_spec.rb

@@ -0,0 +1,152 @@
+require "spec_helper"
+
+describe Mongoid::Persistence::Remove do
+
+  let(:document) do
+    Patient.new(:title => "Mr")
+  end
+
+  let(:address) do
+    Address.new(:street => "Oxford St")
+  end
+
+  let(:collection) do
+    stub.quacks_like(Mongoid::Collection.allocate)
+  end
+
+  let(:email) do
+    Email.new(:address => "test@example.com")
+  end
+
+  let(:location) do
+    Location.new(:name => "Home")
+  end
+
+  before do
+    document.stubs(:collection).returns(collection)
+  end
+
+  describe "#persist" do
+
+    def deep_pull_expectation
+      lambda {
+        collection.expects(:update).with(
+          { "_id" => document.id, "addresses._id" => address.id },
+          { "$pull" => { "addresses.0.locations" => { "_id" => location.id } } },
+          :multi => false,
+          :safe => true
+        ).returns("Object")
+      }
+    end
+
+    def root_pull_expectation
+      lambda {
+        collection.expects(:update).with(
+          { "_id" => document.id },
+          { "$pull" => { "addresses" => { "_id" => address.id } } },
+          :multi => false,
+          :safe => true
+        ).returns("Object")
+      }
+    end
+
+    def root_unset_expectation
+      lambda {
+        collection.expects(:update).with(
+          { "_id" => document.id },
+          { "$unset" => { "email" => true } },
+          :multi => false,
+          :safe => true
+        ).returns("Object")
+      }
+    end
+
+    context "when the embedded document is an embeds_one" do
+
+      before do
+        document.email = email
+      end
+
+      context "when the parent is new" do
+
+        let(:remove) do
+          Mongoid::Persistence::RemoveEmbedded.new(email)
+        end
+
+        it "notifies its changes to parent and removes the parent" do
+          remove.persist.should == true
+          document.email.should be_nil
+        end
+      end
+
+      context "when the parent is not new" do
+
+        let(:remove) do
+          Mongoid::Persistence::RemoveEmbedded.new(email)
+        end
+
+        before do
+          document.instance_variable_set(:@new_record, false)
+        end
+
+        it "performs an in place $set on the embedded document" do
+          root_unset_expectation.call
+          remove.persist.should == true
+        end
+      end
+    end
+
+    context "when the embedded document is an embeds_many" do
+
+      before do
+        address.locations << location
+        document.addresses << address
+      end
+
+      context "when the parent is new" do
+
+        let(:remove) do
+          Mongoid::Persistence::RemoveEmbedded.new(address)
+        end
+
+        it "notifies its changes to the parent and removes the document" do
+          remove.persist.should == true
+          document.addresses.should == []
+        end
+      end
+
+      context "when the parent is not new" do
+
+        let(:remove) do
+          Mongoid::Persistence::RemoveEmbedded.new(address)
+        end
+
+        before do
+          document.instance_variable_set(:@new_record, false)
+        end
+
+        it "performs a $pull on the embedded array" do
+          root_pull_expectation.call
+          remove.persist.should == true
+        end
+      end
+
+      context "when embedded multiple levels" do
+
+        let(:remove) do
+          Mongoid::Persistence::RemoveEmbedded.new(location)
+        end
+
+        before do
+          document.instance_variable_set(:@new_record, false)
+          address.instance_variable_set(:@new_record, false)
+        end
+
+        it "performs a $pull on the embedded array" do
+          deep_pull_expectation.call
+          remove.persist.should == true
+        end
+      end
+    end
+  end
+end