mongoid 8.1.7 → 9.0.0

data/spec/mongoid/tasks/database_spec.rb

@@ -1,8 +1,9 @@
 # frozen_string_literal: true
+# rubocop:todo all
 
 require "spec_helper"
 
-describe "Mongoid::Tasks::Database" do
+describe Mongoid::Tasks::Database do
 
   before(:all) do
     module DatabaseSpec
@@ -212,6 +213,52 @@ describe "Mongoid::Tasks::Database" do
     end
   end
 
+  describe '.create_search_indexes' do
+    let(:searchable_model) do
+      Class.new do
+        include Mongoid::Document
+        store_in collection: BSON::ObjectId.new.to_s
+
+        search_index mappings: { dynamic: true }
+      end
+    end
+
+    let(:index_names) { %w[ name1 name2 ] }
+    let(:searchable_model_spy) do
+      class_spy(searchable_model,
+                create_search_indexes: index_names,
+                search_index_specs: [ { mappings: { dynamic: true } } ])
+    end
+
+    context 'when wait is true' do
+      it 'invokes both create_search_indexes and wait_for_search_indexes' do
+        expect(searchable_model_spy).to receive(:create_search_indexes)
+        expect(described_class).to receive(:wait_for_search_indexes).with({ searchable_model_spy => index_names })
+
+        described_class.create_search_indexes([searchable_model_spy], wait: true)
+      end
+    end
+
+    context 'when wait is false' do
+      it 'invokes only create_search_indexes' do
+        expect(searchable_model_spy).to receive(:create_search_indexes)
+        expect(described_class).to_not receive(:wait_for_search_indexes)
+
+        described_class.create_search_indexes([searchable_model_spy], wait: false)
+      end
+    end
+  end
+
+  describe '.remove_search_indexes' do
+    it 'calls remove_search_indexes on all non-embedded models' do
+      models.each do |model|
+        expect(model).to receive(:remove_search_indexes) unless model.embedded?
+      end
+
+      described_class.remove_search_indexes(models)
+    end
+  end
+
   describe ".undefined_indexes" do
 
     before(:each) do

data/spec/mongoid/tasks/encryption_spec.rb

@@ -0,0 +1,187 @@
+# frozen_string_literal: true
+# rubocop:todo all
+
+require "spec_helper"
+
+describe "Mongoid::Tasks::Encryption" do
+  require_enterprise
+  require_libmongocrypt
+  include_context 'with encryption'
+  restore_config_clients
+
+  describe ".create_data_key" do
+    let(:config) do
+      {
+        default: { hosts: SpecConfig.instance.addresses, database: database_id },
+        encrypted: {
+          hosts: SpecConfig.instance.addresses,
+          database: database_id,
+          options: {
+            auto_encryption_options: {
+              kms_providers: kms_providers,
+              key_vault_namespace: key_vault_namespace,
+              extra_options: extra_options
+            }
+          }
+        }
+      }
+    end
+
+    let(:data_key_id) do
+      BSON::Binary.new('data_key_id', :uuid)
+    end
+
+    let(:key_alt_name) do
+      'mongoid_test_alt_name'
+    end
+
+    before do
+      key_vault_client[key_vault_collection].drop
+      Mongoid::Config.send(:clients=, config)
+    end
+
+    context 'when all parameters are correct' do
+      context 'when all parameters are provided' do
+        before do
+          expect_any_instance_of(Mongo::ClientEncryption)
+            .to receive(:create_data_key)
+                  .with('local', { key_alt_names: [key_alt_name] })
+                  .and_return(data_key_id)
+        end
+        it 'creates a data key' do
+          result = Mongoid::Tasks::Encryption.create_data_key(
+            kms_provider_name: 'local',
+            client_name: :encrypted,
+            key_alt_name: key_alt_name
+          )
+          expect(result).to eq(
+            {
+              key_id: Base64.strict_encode64(data_key_id.data),
+              key_vault_namespace: key_vault_namespace,
+              kms_provider: 'local',
+              key_alt_name: key_alt_name
+            }
+          )
+        end
+      end
+
+      context 'when kms_provider_name is not provided' do
+        context 'and there is only one kms provider' do
+          context 'without key_alt_name' do
+            before do
+              expect_any_instance_of(Mongo::ClientEncryption)
+                .to receive(:create_data_key)
+                      .with('local', {})
+                      .and_return(data_key_id)
+            end
+            it 'creates a data key' do
+              result = Mongoid::Tasks::Encryption.create_data_key(client_name: :encrypted)
+              expect(result).to eq(
+                {
+                  key_id: Base64.strict_encode64(data_key_id.data),
+                  key_vault_namespace: key_vault_namespace,
+                  kms_provider: 'local'
+                }
+              )
+            end
+          end
+
+          context 'with key_alt_name' do
+            before do
+              expect_any_instance_of(Mongo::ClientEncryption)
+                .to receive(:create_data_key)
+                      .with('local', {key_alt_names: [key_alt_name]})
+                      .and_return(data_key_id)
+            end
+            it 'creates a data key' do
+              result = Mongoid::Tasks::Encryption.create_data_key(
+                client_name: :encrypted,
+                key_alt_name: key_alt_name
+              )
+              expect(result).to eq(
+                {
+                  key_id: Base64.strict_encode64(data_key_id.data),
+                  key_vault_namespace: key_vault_namespace,
+                  kms_provider: 'local',
+                  key_alt_name: key_alt_name
+                }
+              )
+            end
+          end
+        end
+      end
+    end
+
+    context 'when the client name is incorrect' do
+      it 'raises an error' do
+        expect {
+          Mongoid::Tasks::Encryption.create_data_key(kms_provider_name: 'local', client_name: :wrong_client)
+        }.to raise_error(Mongoid::Errors::NoClientConfig)
+      end
+    end
+
+    context 'when the client does not have auto_encryption_options' do
+      it 'raises an error' do
+        expect {
+          Mongoid::Tasks::Encryption.create_data_key(kms_provider_name: 'local', client_name: :default)
+        }.to raise_error(Mongoid::Errors::InvalidAutoEncryptionConfiguration)
+      end
+    end
+
+    context 'when key_value_namespace is not set' do
+      let(:config) do
+        {
+          default: { hosts: SpecConfig.instance.addresses, database: database_id },
+          encrypted: {
+            hosts: SpecConfig.instance.addresses,
+            database: database_id,
+            options: {
+              auto_encryption_options: {
+                kms_providers: kms_providers,
+                extra_options: extra_options
+              }
+            }
+          }
+        }
+      end
+
+      it 'raises an error' do
+        expect {
+          Mongoid::Tasks::Encryption.create_data_key(kms_provider_name: 'local', client_name: :encrypted)
+        }.to raise_error(Mongoid::Errors::InvalidAutoEncryptionConfiguration)
+      end
+    end
+
+    context 'when kms_providers is not set' do
+      let(:config) do
+        {
+          default: { hosts: SpecConfig.instance.addresses, database: database_id },
+          encrypted: {
+            hosts: SpecConfig.instance.addresses,
+            database: database_id,
+            options: {
+              auto_encryption_options: {
+                key_vault_namespace: key_vault_namespace,
+                extra_options: extra_options
+              }
+            }
+          }
+        }
+      end
+
+      it 'raises an error' do
+        expect {
+          Mongoid::Tasks::Encryption.create_data_key(kms_provider_name: 'local', client_name: :encrypted)
+        }.to raise_error(Mongoid::Errors::InvalidAutoEncryptionConfiguration)
+      end
+    end
+
+    context 'when kms_providers does not include used provider' do
+      it 'raises an error' do
+        expect {
+          Mongoid::Tasks::Encryption.create_data_key(kms_provider_name: 'aws', client_name: :encrypted)
+        }.to raise_error(Mongoid::Errors::InvalidAutoEncryptionConfiguration)
+      end
+    end
+  end
+end

data/spec/mongoid/threaded_spec.rb

@@ -1,4 +1,5 @@
 # frozen_string_literal: true
+# rubocop:todo all
 
 require "spec_helper"
 
@@ -303,4 +304,41 @@ describe Mongoid::Threaded do
       end
     end
   end
+
+  describe "#get_session" do
+    let(:session) do
+      double('session')
+    end
+
+    context "without client specified" do
+      before do
+        described_class.set_session(session)
+      end
+
+      it 'returns session' do
+        expect(described_class.get_session).to be(session)
+      end
+    end
+
+    context 'with client' do
+      let(:client) do
+        double('client')
+      end
+
+      let(:client_session) do
+        double('client_session')
+      end
+
+      before do
+        # Set a 'global' thread session (deprecated)
+        described_class.set_session(session)
+        # Set a session for a client
+        described_class.set_session(client_session, client: client)
+      end
+
+      it 'returns session' do
+        expect(described_class.get_session(client: client)).to be(client_session)
+      end
+    end
+  end
 end

data/spec/mongoid/timestamps/created/short_spec.rb

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

data/spec/mongoid/timestamps/created_spec.rb

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

data/spec/mongoid/timestamps/timeless_spec.rb

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

data/spec/mongoid/timestamps/updated/short_spec.rb

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

data/spec/mongoid/timestamps/updated_spec.rb

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

data/spec/mongoid/timestamps_spec.rb

@@ -1,4 +1,5 @@
 # frozen_string_literal: true
+# rubocop:todo all
 
 require "spec_helper"
 require_relative './timestamps_spec_models'

data/spec/mongoid/timestamps_spec_models.rb

@@ -1,4 +1,5 @@
 # frozen_string_literal: true
+# rubocop:todo all
 
 module TimestampsSpec
   module Touch