mongoid 7.2.1 → 9.0.0

data/spec/mongoid/clients/transactions_spec_models.rb

@@ -0,0 +1,159 @@
+# rubocop:todo all
+class TransactionsSpecCounter
+  def initialize
+    @called = 0
+  end
+
+  def inc
+    @called += 1
+  end
+
+  def value
+    @called
+  end
+
+  def reset
+    @called = 0
+  end
+end
+
+module TransactionsSpecCountable
+  def after_commit_counter
+    @after_commit_counter ||= TransactionsSpecCounter.new
+  end
+
+  def after_commit_counter=(new_counter)
+    @after_commit_counter = new_counter
+  end
+
+  def after_rollback_counter
+    @after_rollback_counter ||= TransactionsSpecCounter.new
+  end
+
+  def after_rollback_counter=(new_counter)
+    @after_rollback_counter = new_counter
+  end
+end
+
+class TransactionsSpecPerson
+  include Mongoid::Document
+  include TransactionsSpecCountable
+
+  field :name, type: String
+
+  after_commit do
+    after_commit_counter.inc
+  end
+
+  after_rollback do
+    after_rollback_counter.inc
+  end
+end
+
+class TransactionsSpecPersonWithOnCreate
+  include Mongoid::Document
+  include TransactionsSpecCountable
+
+  field :name, type: String
+
+  after_commit on: :create do
+    after_commit_counter.inc
+  end
+
+  after_rollback on: :create do
+    after_rollback_counter.inc
+  end
+end
+
+class TransactionsSpecPersonWithOnUpdate
+  include Mongoid::Document
+  include TransactionsSpecCountable
+
+  field :name, type: String
+
+  after_commit on: :update do
+    after_commit_counter.inc
+  end
+
+  after_rollback on: :update do
+    after_rollback_counter.inc
+  end
+end
+
+class TransactionsSpecPersonWithOnDestroy
+  include Mongoid::Document
+  include TransactionsSpecCountable
+
+  field :name, type: String
+
+  after_commit on: :destroy do
+    after_commit_counter.inc
+  end
+
+  after_rollback on: :destroy do
+    after_rollback_counter.inc
+  end
+end
+class TransactionSpecRaisesBeforeSave
+  include Mongoid::Document
+  include TransactionsSpecCountable
+
+  field :attr, type: String
+
+  before_save do
+    raise "I cannot be saved"
+  end
+
+  after_commit do
+    after_commit_counter.inc
+  end
+
+  after_rollback do
+    after_rollback_counter.inc
+  end
+end
+
+class TransactionSpecRaisesAfterSave
+  include Mongoid::Document
+  include TransactionsSpecCountable
+
+  field :attr, type: String
+
+  after_save do
+    raise "I cannot be saved"
+  end
+
+  after_commit do
+    after_commit_counter.inc
+  end
+
+  after_rollback do
+    after_rollback_counter.inc
+  end
+end
+
+class TransactionSpecRaisesBeforeCreate
+  include Mongoid::Document
+
+  def self.after_commit_counter
+    @@after_commit_counter ||= TransactionsSpecCounter.new
+  end
+
+  def self.after_rollback_counter
+    @@after_rollback_counter ||= TransactionsSpecCounter.new
+  end
+
+  field :attr, type: String
+
+  before_create do
+    raise "I cannot be saved"
+  end
+
+  after_commit do
+    self.class.after_commit_counter.inc
+  end
+
+  after_rollback do
+    self.class.after_rollback_counter.inc
+  end
+end

data/spec/mongoid/clients_spec.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
-# encoding: utf-8
+# rubocop:todo all
 
 require "spec_helper"
 
@@ -739,19 +739,6 @@ describe Mongoid::Clients do
       end
     end
 
-    context "when provided a class that extends another document" do
-
-      let(:klass) do
-        Class.new(Band)
-      end
-
-      it "raises an error" do
-        expect {
-          klass.store_in(database: :artists)
-        }.to raise_error(Mongoid::Errors::InvalidStorageParent)
-      end
-    end
-
     context "when provided a hash" do
 
       context "when the hash is not valid" do
@@ -763,6 +750,128 @@ describe Mongoid::Clients do
         end
       end
     end
+
+    context "when it is called on a subclass" do
+
+      let(:client) { StoreParent.collection.client }
+      let(:parent) { StoreParent.create! }
+      let(:child1) { StoreChild1.create! }
+      let(:child2) { StoreChild2.create! }
+
+      before do
+        class StoreParent
+          include Mongoid::Document
+        end
+
+        class StoreChild1 < StoreParent
+        end
+
+        class StoreChild2 < StoreParent
+        end
+      end
+
+      after do
+        Object.send(:remove_const, :StoreParent)
+        Object.send(:remove_const, :StoreChild1)
+        Object.send(:remove_const, :StoreChild2)
+      end
+
+      context "when it is not called on the parent" do
+
+        context "when it is called on all subclasses" do
+
+          before do
+            StoreChild1.store_in collection: :store_ones
+            StoreChild2.store_in collection: :store_twos
+            [ parent, child1, child2 ]
+          end
+
+          let(:db_parent) { client['store_parents'].find.first }
+          let(:db_child1) { client['store_ones'].find.first }
+          let(:db_child2) { client['store_twos'].find.first }
+
+          it "stores the documents in the correct collections" do
+            expect(db_parent).to eq({ "_id" => parent.id, "_type" => "StoreParent" })
+            expect(db_child1).to eq({ "_id" => child1.id, "_type" => "StoreChild1" })
+            expect(db_child2).to eq({ "_id" => child2.id, "_type" => "StoreChild2" })
+          end
+
+          it "only queries from its own collections" do
+            expect(StoreParent.count).to eq(1)
+            expect(StoreChild1.count).to eq(1)
+            expect(StoreChild2.count).to eq(1)
+          end
+        end
+
+        context "when it is called on one of the subclasses" do
+
+          before do
+            StoreChild1.store_in collection: :store_ones
+            [ parent, child1, child2 ]
+          end
+
+          let(:db_parent) { client['store_parents'].find.first }
+          let(:db_child1) { client['store_ones'].find.first }
+          let(:db_child2) { client['store_parents'].find.to_a.last }
+
+          it "stores the documents in the correct collections" do
+            expect(db_parent).to eq({ "_id" => parent.id, "_type" => "StoreParent" })
+            expect(db_child1).to eq({ "_id" => child1.id, "_type" => "StoreChild1" })
+            expect(db_child2).to eq({ "_id" => child2.id, "_type" => "StoreChild2" })
+          end
+
+          it "queries from its own collections" do
+            expect(StoreParent.count).to eq(2)
+            expect(StoreChild1.count).to eq(1)
+            expect(StoreChild2.count).to eq(1)
+          end
+        end
+      end
+
+      context "when it is called on the parent" do
+
+        before do
+          StoreParent.store_in collection: :st_parents
+        end
+
+        context "when it is called on all subclasses" do
+
+          before do
+            StoreChild1.store_in collection: :store_ones
+            StoreChild2.store_in collection: :store_twos
+            [ parent, child1, child2 ]
+          end
+
+          let(:db_parent) { client['st_parents'].find.first }
+          let(:db_child1) { client['store_ones'].find.first }
+          let(:db_child2) { client['store_twos'].find.first }
+
+          it "stores the documents in the correct collections" do
+            expect(db_parent).to eq({ "_id" => parent.id, "_type" => "StoreParent" })
+            expect(db_child1).to eq({ "_id" => child1.id, "_type" => "StoreChild1" })
+            expect(db_child2).to eq({ "_id" => child2.id, "_type" => "StoreChild2" })
+          end
+        end
+
+        context "when it is called on one of the subclasses" do
+
+          before do
+            StoreChild1.store_in collection: :store_ones
+            [ parent, child1, child2 ]
+          end
+
+          let(:db_parent) { client['st_parents'].find.first }
+          let(:db_child1) { client['store_ones'].find.first }
+          let(:db_child2) { client['st_parents'].find.to_a.last }
+
+          it "stores the documents in the correct collections" do
+            expect(db_parent).to eq({ "_id" => parent.id, "_type" => "StoreParent" })
+            expect(db_child1).to eq({ "_id" => child1.id, "_type" => "StoreChild1" })
+            expect(db_child2).to eq({ "_id" => child2.id, "_type" => "StoreChild2" })
+          end
+        end
+      end
+    end
   end
 
   describe ".with" do
@@ -792,18 +901,18 @@ describe Mongoid::Clients do
         end
       end
 
-      describe ".create" do
+      describe ".create!" do
 
         let!(:band) do
           Band.with(database: database_id_alt) do |klass|
-            klass.create
+            klass.create!
           end
         end
 
         it "does not persist to the default database" do
           expect {
             Band.find(band.id)
-          }.to raise_error(Mongoid::Errors::DocumentNotFound)
+          }.to raise_error(Mongoid::Errors::DocumentNotFound, /Document\(s\) not found for class Band with id\(s\)/)
         end
 
         let(:from_db) do
@@ -830,18 +939,18 @@ describe Mongoid::Clients do
 
     context "when sending operations to a different collection" do
 
-      describe ".create" do
+      describe ".create!" do
 
         let!(:band) do
           Band.with(collection: "artists") do |klass|
-            klass.create
+            klass.create!
           end
         end
 
         it "does not persist to the default database" do
           expect {
             Band.find(band.id)
-          }.to raise_error(Mongoid::Errors::DocumentNotFound)
+          }.to raise_error(Mongoid::Errors::DocumentNotFound, /Document\(s\) not found for class Band with id\(s\)/)
         end
 
         let(:from_db) do
@@ -945,7 +1054,7 @@ describe Mongoid::Clients do
       describe ".save" do
 
         before do
-          Person.create(ssn: "432-97-1113")
+          Person.create!(ssn: "432-97-1113")
         end
 
         context "when a mongodb error occurs" do
@@ -1013,6 +1122,61 @@ describe Mongoid::Clients do
         end
       end
     end
+
+    context 'when requesting named client' do
+      let(:secondary_client) do
+        double(Mongo::Client).tap do |client|
+          allow(client).to receive(:cluster).and_return(double("cluster"))
+        end
+      end
+
+      before do
+        expect(Mongoid::Clients::Factory).to receive(:create)
+          .with(:secondary)
+          .and_return(secondary_client)
+      end
+
+      after do
+        Mongoid::Clients.clients.delete(:secondary)
+      end
+
+      it 'does not close the client' do
+        expect(secondary_client).not_to receive(:close)
+
+        Band.with(client: :default) do |klass|
+          klass.mongo_client
+        end
+
+        Band.with(client: :secondary) do |klass|
+          klass.mongo_client
+        end
+      end
+    end
+
+    context 'when using on different objects' do
+      require_mri
+
+      let(:first_band) do
+        Band.create!(name: "The Beatles")
+      end
+
+      let(:second_band) do
+        Band.create!(name: 'Led Zeppelin')
+      end
+
+      it 'does not create extra symbols' do
+        first_band.with(write: { w: 0, j: false }) do |band|
+          band.set(active: false)
+        end
+        initial_symbols_count = Symbol.all_symbols.size
+        second_band.with(write: { w: 0, j: false }) do |band|
+          band.set(active: false)
+        end
+        expect(Symbol.all_symbols.size).to be <= initial_symbols_count
+      end
+
+    end
+
   end
 
   context "when overriding the default database" do
@@ -1038,7 +1202,7 @@ describe Mongoid::Clients do
       end
 
       let!(:band) do
-        Band.create(name: "Tool")
+        Band.create!(name: "Tool")
       end
 
       it "persists to the overridden database" do
@@ -1052,4 +1216,48 @@ describe Mongoid::Clients do
       end
     end
   end
+
+  context "#disconnect" do
+
+    let(:clients) do
+      Mongoid::Clients.clients.values
+    end
+
+    before do
+      Band.all.entries
+    end
+
+    it "disconnects from all active clients" do
+      clients.each do |client|
+        expect(client).to receive(:close).and_call_original
+      end
+      Mongoid::Clients.disconnect
+    end
+
+    it "returns true" do
+      expect(Mongoid::Clients.disconnect).to eq(true)
+    end
+  end
+
+  context "#reconnect" do
+
+    let(:clients) do
+      Mongoid::Clients.clients.values
+    end
+
+    before do
+      Band.all.entries
+    end
+
+    it "reconnects all active clients" do
+      clients.each do |client|
+        expect(client).to receive(:reconnect).and_call_original
+      end
+      Mongoid::Clients.reconnect
+    end
+
+    it "returns true" do
+      expect(Mongoid::Clients.reconnect).to eq(true)
+    end
+  end
 end

data/spec/mongoid/collection_configurable_spec.rb

@@ -0,0 +1,159 @@
+# frozen_string_literal: true
+# rubocop:todo all
+
+require "spec_helper"
+
+describe Mongoid::CollectionConfigurable do
+
+  before(:all) do
+    class CollectionConfigurableValidOptions
+      include Mongoid::Document
+
+      store_in collection_options: {
+        capped: true,
+        size: 2560
+      }
+    end
+
+    class CollectionConfigurableUnknownOptions
+      include Mongoid::Document
+
+      store_in collection_options: {
+        i_am_not_an_option: true,
+        let_me: { fail: true }
+      }
+    end
+
+    class CollectionConfigurableInvalidOptions
+      include Mongoid::Document
+
+      store_in collection_options: {
+        capped: true
+        # Creating a capped collection requires `size` field; therefore,
+        # an attempt to create a collection with only `capped : true` option
+        # fails.
+      }
+    end
+  end
+
+  after(:all) do
+    Mongoid.deregister_model(CollectionConfigurableValidOptions)
+    Object.send(:remove_const, :CollectionConfigurableValidOptions)
+    Mongoid.deregister_model(CollectionConfigurableUnknownOptions)
+    Object.send(:remove_const, :CollectionConfigurableUnknownOptions)
+    Mongoid.deregister_model(CollectionConfigurableInvalidOptions)
+    Object.send(:remove_const, :CollectionConfigurableInvalidOptions)
+  end
+
+  after(:each) do
+    [
+      CollectionConfigurableValidOptions,
+      CollectionConfigurableUnknownOptions
+    ].each do |klazz|
+      klazz.collection.drop
+    end
+  end
+
+  context 'when collection does not exist' do
+    context 'with valid options' do
+      let(:subject) do
+        CollectionConfigurableValidOptions
+      end
+
+      before(:each) do
+        subject.create_collection
+      end
+
+      let(:coll_options) do
+        subject.collection.database.list_collections(filter: { name: subject.collection_name.to_s }).first
+      end
+
+      it 'creates the collection' do
+        expect(coll_options).not_to be_nil
+      end
+
+      it 'passes collection options' do
+        expect(coll_options.dig('options', 'capped')).to eq(true)
+        expect(coll_options.dig('options', 'size')).to eq(2560)
+      end
+    end
+
+    context 'with an unknown collection options' do
+      let(:subject) do
+        CollectionConfigurableUnknownOptions
+      end
+
+      it 'raises an error' do
+        expect do
+          subject.create_collection
+        end.to raise_error(Mongoid::Errors::CreateCollectionFailure)
+      end
+    end
+
+    context 'with invalid options' do
+      let(:subject) do
+        CollectionConfigurableInvalidOptions
+      end
+
+      it 'raises an error' do
+        expect do
+          subject.create_collection
+        end.to raise_error(Mongoid::Errors::CreateCollectionFailure)
+      end
+    end
+  end
+
+  context 'when collection exists' do
+    let(:subject) do
+      CollectionConfigurableValidOptions
+    end
+
+    context 'when force is false' do
+      let(:logger) do
+        double("logger").tap do |log|
+          expect(log).to receive(:debug).once.with(/Collection '#{subject.collection_name}' already exist/)
+        end
+      end
+
+      before do
+        allow(subject).to receive(:logger).and_return(logger)
+        subject.collection.create
+      end
+
+      it 'logs a message' do
+        subject.create_collection
+      end
+    end
+
+    context 'when force is true' do
+      let(:logger) do
+        double("logger")
+      end
+
+      let(:coll_options) do
+        subject.collection.database.list_collections(filter: { name: subject.collection_name.to_s }).first
+      end
+
+      before do
+        allow(subject).to receive(:logger).and_return(logger)
+        subject.collection.create
+      end
+
+      it 'does not log a message' do
+        expect(logger).to receive(:debug).never.with(/Collection '#{subject.collection_name}' already exist/)
+        subject.create_collection(force: true)
+      end
+
+      it 'creates the collection' do
+        subject.create_collection(force: true)
+        expect(coll_options).not_to be_nil
+      end
+
+      it 'passes collection options' do
+        subject.create_collection(force: true)
+        expect(coll_options.dig('options', 'capped')).to eq(true)
+        expect(coll_options.dig('options', 'size')).to eq(2560)
+      end
+    end
+  end
+end

data/spec/mongoid/composable_spec.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
-# encoding: utf-8
+# rubocop:todo all
 
 require "spec_helper"