mongo 2.18.1 → 2.18.2

data/lib/mongo/protocol/insert.rb

@@ -1,181 +0,0 @@
-# frozen_string_literal: true
-# encoding: utf-8
-
-# Copyright (C) 2014-2020 MongoDB Inc.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#   http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-module Mongo
-  module Protocol
-
-    # MongoDB Wire protocol Insert message.
-    #
-    # This is a client request message that is sent to the server in order
-    # to insert documents within a namespace.
-    #
-    # The operation only has one flag +:continue_on_error+ which the user
-    # can use to instruct the database server to continue processing a bulk
-    # insertion if one happens to fail (e.g. due to duplicate IDs). This makes
-    # builk insert behave similarly to a seires of single inserts, except
-    # lastError will be set if any insert fails, not just the last one.
-    #
-    # If multiple errors occur, only the most recent will be reported by the
-    # getLastError mechanism.
-    #
-    # @api semipublic
-    class Insert < Message
-
-      # Creates a new Insert message
-      #
-      # @example Insert a user document
-      #   Insert.new('xgen', 'users', [{:name => 'Tyler'}])
-      #
-      # @example Insert serveral user documents and continue on errors
-      #   Insert.new('xgen', 'users', users, :flags => [:continue_on_error])
-      #
-      # @param database [String, Symbol]  The database to insert into.
-      # @param collection [String, Symbol] The collection to insert into.
-      # @param documents [Array<Hash>] The documents to insert.
-      # @param options [Hash] Additional options for the insertion.
-      #
-      # @option options :flags [Array] The flags for the insertion message.
-      #   Supported flags: +:continue_on_error+
-      # @option options [ true, false ] validating_keys Whether keys should be
-      #   validated for being valid document keys (i.e. not begin with $ and
-      #   not contain dots).
-      def initialize(database, collection, documents, options = {})
-        @database = database
-        @namespace = "#{database}.#{collection}"
-        @documents = documents
-        @flags = options[:flags] || []
-        @upconverter = Upconverter.new(collection, documents, options)
-        @options = options
-        super
-      end
-
-      # Return the event payload for monitoring.
-      #
-      # @example Return the event payload.
-      #   message.payload
-      #
-      # @return [ BSON::Document ] The event payload.
-      #
-      # @since 2.1.0
-      def payload
-        BSON::Document.new(
-          command_name: 'insert',
-          database_name: @database,
-          command: upconverter.command,
-          request_id: request_id
-        )
-      end
-
-      protected
-
-      attr_reader :upconverter
-
-      private
-
-      def validating_keys?
-        @options.fetch(:validating_keys, true)
-      end
-
-      # The operation code required to specify an Insert message.
-      # @return [Fixnum] the operation code.
-      #
-      # @since 2.5.0
-      OP_CODE = 2002
-
-      # Available flags for an Insert message.
-      FLAGS = [:continue_on_error]
-
-      # @!attribute
-      # @return [Array<Symbol>] The flags for this Insert message.
-      field :flags, BitVector.new(FLAGS)
-
-      # @!attribute
-      # @return [String] The namespace for this Insert message.
-      field :namespace, CString
-
-      # @!attribute
-      # @return [Array<Hash>] The documents to insert.
-      field :documents, Document, true
-
-      # Converts legacy insert messages to the appropriare OP_COMMAND style
-      # message.
-      #
-      # @since 2.1.0
-      class Upconverter
-
-        # Insert field constant.
-        #
-        # @since 2.1.0
-        INSERT = 'insert'.freeze
-
-        # Documents field constant.
-        #
-        # @since 2.1.0
-        DOCUMENTS = 'documents'.freeze
-
-        # Write concern field constant.
-        #
-        # @since 2.1.0
-        WRITE_CONCERN = 'writeConcern'.freeze
-
-        # @return [ String ] collection The name of the collection.
-        attr_reader :collection
-
-        # @return [ Array<BSON::Document> ] documents The documents to insert.
-        attr_reader :documents
-
-        # @return [ Hash ] options The options.
-        attr_reader :options
-
-        # Instantiate the upconverter.
-        #
-        # @example Instantiate the upconverter.
-        #   Upconverter.new('users', documents)
-        #
-        # @param [ String ] collection The name of the collection.
-        # @param [ Array<BSON::Document> ] documents The documents.
-        # @param [ Hash ] options The options.
-        #
-        # @since 2.1.0
-        def initialize(collection, documents, options)
-          @collection = collection
-          @documents = documents
-          @options = options
-        end
-
-        # Get the upconverted command.
-        #
-        # @example Get the command.
-        #   upconverter.command
-        #
-        # @return [ BSON::Document ] The upconverted command.
-        #
-        # @since 2.1.0
-        def command
-          document = BSON::Document.new
-          document.store(INSERT, collection)
-          document.store(DOCUMENTS, documents)
-          document.store(Message::ORDERED, options.fetch(:ordered, true))
-          document.merge!(WRITE_CONCERN => options[:write_concern].options) if options[:write_concern]
-          document
-        end
-      end
-
-      Registry.register(OP_CODE, self)
-    end
-  end
-end

data/lib/mongo/protocol/update.rb

@@ -1,214 +0,0 @@
-# frozen_string_literal: true
-# encoding: utf-8
-
-# Copyright (C) 2014-2020 MongoDB Inc.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#   http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-module Mongo
-  module Protocol
-
-    # MongoDB Wire protocol Update message.
-    #
-    # This is a client request message that is sent to the server in order
-    # to update documents matching the provided query.
-    #
-    # The default is to update a single document. In order to update many at
-    # a time users should set the +:multi_update+ flag for the update.
-    #
-    # If an upsert (update or insert) is desired, users can set the +:upsert+
-    # flag in order to indicate they would like to insert the merged selector
-    # and update if no document matching the update query currently exists.
-    #
-    # @api semipublic
-    class Update < Message
-
-      # Creates a new Update message
-      #
-      # @example Update single document
-      #   Update.new('xgen', 'users', {:name => 'Tyler'}, {:name => 'Bob'})
-      #
-      # @example Perform a multi update
-      #   Update.new('xgen', 'users',
-      #     {:age => 20}, {:age => 21}, :flags => [:multi_update])
-      #
-      # @example Perform an upsert
-      #   Update.new('xgen', 'users', {:name => 'Tyler'}, :flags => [:upsert])
-      #
-      # @param database [String, Symbol]  The database to update.
-      # @param collection [String, Symbol] The collection to update.
-      # @param selector [Hash] The update selector.
-      # @param update [Hash] The update to perform.
-      # @param options [Hash] The additional query options.
-      #
-      # @option options :flags [Array] The flags for the update message.
-      #
-      #   Supported flags: +:upsert+, +:multi_update+
-      def initialize(database, collection, selector, update, options = {})
-        @database    = database
-        @collection  = collection
-        @namespace   = "#{database}.#{collection}"
-        @selector    = selector
-        @update      = update
-        @flags       = options[:flags] || []
-        @upconverter = Upconverter.new(collection, selector, update, flags)
-        super
-      end
-
-      # Return the event payload for monitoring.
-      #
-      # @example Return the event payload.
-      #   message.payload
-      #
-      # @return [ BSON::Document ] The event payload.
-      #
-      # @since 2.1.0
-      def payload
-        BSON::Document.new(
-          command_name: 'update',
-          database_name: @database,
-          command: upconverter.command,
-          request_id: request_id
-        )
-      end
-
-      protected
-
-      attr_reader :upconverter
-
-      private
-
-      # The operation code required to specify an Update message.
-      # @return [Fixnum] the operation code.
-      #
-      # @since 2.5.0
-      OP_CODE = 2001
-
-      # Available flags for an Update message.
-      FLAGS = [:upsert, :multi_update]
-
-      # Field representing Zero encoded as an Int32.
-      field :zero, Zero
-
-      # @!attribute
-      # @return [String] The namespace for this Update message.
-      field :namespace, CString
-
-      # @!attribute
-      # @return [Array<Symbol>] The flags for this Update message.
-      field :flags, BitVector.new(FLAGS)
-
-      # @!attribute
-      # @return [Hash] The selector for this Update message.
-      field :selector, Document
-
-      # @!attribute
-      # @return [Hash] The update for this Delete message.
-      field :update, Document
-
-      # Converts legacy update messages to the appropriare OP_COMMAND style
-      # message.
-      #
-      # @since 2.1.0
-      class Upconverter
-
-        # The multi constant.
-        #
-        # @since 2.2.0
-        MULTI = 'multi'.freeze
-
-        # The u constant.
-        #
-        # @since 2.2.0
-        U = 'u'.freeze
-
-        # The update constant.
-        #
-        # @since 2.2.0
-        UPDATE = 'update'.freeze
-
-        # The updates constant.
-        #
-        # @since 2.2.0
-        UPDATES = 'updates'.freeze
-
-        # The upsert constant.
-        #
-        # @since 2.2.0
-        UPSERT = 'upsert'.freeze
-
-        # @return [ String ] collection The name of the collection.
-        attr_reader :collection
-
-        # @return [ Hash ] filter The filter.
-        attr_reader :filter
-
-        # @return [ Hash ] update The update.
-        attr_reader :update
-
-        # @return [ Array<Symbol> ] flags The flags.
-        attr_reader :flags
-
-        # Instantiate the upconverter.
-        #
-        # @example Instantiate the upconverter.
-        #   Upconverter.new(
-        #     'users',
-        #     { name: 'test' },
-        #     { '$set' => { 'name' => 't' }},
-        #     []
-        #   )
-        #
-        # @param [ String ] collection The name of the collection.
-        # @param [ Hash ] filter The filter.
-        # @param [ Hash ] update The update.
-        # @param [ Array<Symbol> ] flags The flags.
-        #
-        # @since 2.1.0
-        def initialize(collection, filter, update, flags)
-          @collection = collection
-          @filter = filter
-          @update = update
-          @flags = flags
-        end
-
-        # Get the upconverted command.
-        #
-        # @example Get the command.
-        #   upconverter.command
-        #
-        # @return [ BSON::Document ] The upconverted command.
-        #
-        # @since 2.1.0
-        def command
-          document = BSON::Document.new
-          updates = BSON::Document.new
-          updates.store(Message::Q, filter)
-          updates.store(U, update)
-          if flags.include?(:multi_update)
-            updates.store(MULTI, true)
-          end
-          if flags.include?(:upsert)
-            updates.store(UPSERT, true)
-          end
-          document.store(UPDATE, collection)
-          document.store(Message::ORDERED, true)
-          document.store(UPDATES, [ updates ])
-          document
-        end
-      end
-
-      Registry.register(OP_CODE, self)
-    end
-  end
-end

data/spec/mongo/operation/delete/command_spec.rb

@@ -1,115 +0,0 @@
-# frozen_string_literal: true
-# encoding: utf-8
-
-require 'spec_helper'
-
-describe Mongo::Operation::Delete::Command do
-
-  let(:write_concern) do
-    Mongo::WriteConcern.get(w: :majority)
-  end
-
-  let(:session) { nil }
-  let(:deletes) { [{:q => { :foo => 1 }, :limit => 1}] }
-  let(:spec) do
-    { :deletes       => deletes,
-      :db_name       => authorized_collection.database.name,
-      :coll_name     => authorized_collection.name,
-      :write_concern => write_concern,
-      :ordered       => true,
-      :session       => session
-    }
-  end
-
-  let(:op) { described_class.new(spec) }
-
-  describe '#initialize' do
-
-    context 'spec' do
-
-      it 'sets the spec' do
-        expect(op.spec).to eq(spec)
-      end
-    end
-  end
-
-  describe '#==' do
-
-    context 'spec' do
-
-      context 'when two ops have the same specs' do
-        let(:other) { described_class.new(spec) }
-
-        it 'returns true' do
-          expect(op).to eq(other)
-        end
-      end
-
-      context 'when two ops have different specs' do
-        let(:other_deletes) { [{:q => { :bar => 1 }, :limit => 1}] }
-        let(:other_spec) do
-          { :deletes       => other_deletes,
-            :db_name       => authorized_collection.database.name,
-            :coll_name     => authorized_collection.name,
-            :write_concern => write_concern,
-            :ordered       => true
-          }
-        end
-        let(:other) { described_class.new(other_spec) }
-
-        it 'returns false' do
-          expect(op).not_to eq(other)
-        end
-      end
-    end
-  end
-
-  describe 'write concern' do
-
-    context 'when write concern is not specified' do
-
-      let(:spec) do
-        { :deletes       => deletes,
-          :db_name       => authorized_collection.database.name,
-          :coll_name     => authorized_collection.name,
-          :ordered       => true
-        }
-      end
-
-      it 'does not include write concern in the selector' do
-        expect(op.send(:command, double('server'))[:writeConcern]).to be_nil
-      end
-    end
-
-    context 'when write concern is specified' do
-
-      it 'includes write concern in the selector' do
-        expect(op.send(:command, double('server'))[:writeConcern]).to eq(write_concern.options)
-      end
-    end
-  end
-
-  describe '#message' do
-
-    context 'when the server does not support OP_MSG' do
-      max_server_version '3.4'
-
-      let(:expected_selector) do
-        {
-            :delete        => authorized_collection.name,
-            :deletes       => deletes,
-            :writeConcern => write_concern.options,
-            :ordered       => true
-        }
-      end
-
-      it 'creates the correct query wire protocol message' do
-        expect(Mongo::Protocol::Query).to receive(:new).with(authorized_collection.database.name,
-                                                             '$cmd',
-                                                             expected_selector,
-                                                             { limit: -1 } )
-        op.send(:message, authorized_primary)
-      end
-    end
-  end
-end

data/spec/mongo/operation/find/legacy_spec.rb

@@ -1,131 +0,0 @@
-# frozen_string_literal: true
-# encoding: utf-8
-
-require 'spec_helper'
-
-describe Mongo::Operation::Find::Legacy do
-  max_server_version '3.4'
-
-  let(:context) { Mongo::Operation::Context.new }
-
-  let(:selector) { { foo: 1 } }
-  let(:query_options) { {} }
-  let(:spec) do
-    { :selector  => selector,
-      :options      => query_options,
-      :db_name   => authorized_collection.database.name,
-      :coll_name => authorized_collection.name,
-      :read => Mongo::ServerSelector.get
-    }
-  end
-  let(:op) { described_class.new(spec) }
-
-  describe '#initialize' do
-
-    context 'query spec' do
-      it 'sets the query spec' do
-        expect(op.spec).to be(spec)
-      end
-    end
-  end
-
-  describe '#==' do
-
-    context 'when two ops have different specs' do
-      let(:other_spec) do
-        { :selector  => { :a => 1 },
-          :options      => query_options,
-          :db_name   => authorized_collection.database.name,
-          :coll_name => authorized_collection.name
-        }
-      end
-      let(:other) { described_class.new(other_spec) }
-
-      it 'returns false' do
-        expect(op).not_to eq(other)
-      end
-    end
-  end
-
-  describe '#message' do
-
-    let(:query_options) do
-      { :flags => [ :no_cursor_timeout ]}
-    end
-
-    let(:query) do
-      described_class.new(spec)
-    end
-
-    let(:cluster_single) do
-      double('cluster').tap do |c|
-        allow(c).to receive(:single?).and_return(true)
-      end
-    end
-
-    let(:message) do
-      authorized_primary.with_connection do |connection|
-        query.send(:message, connection)
-      end
-    end
-
-    it 'applies the correct flags' do
-      expect(message.flags).to eq(query_options[:flags])
-    end
-
-    context 'when the server is a secondary' do
-      let(:description) do
-        double('description').tap do |description|
-          expect(description).to receive(:mongos?) { false }
-          expect(description).to receive(:standalone?) { false }
-          #expect(description).to receive(:load_balancer?) { false }
-        end
-      end
-
-      let(:connection) do
-        double('connection').tap do |conn|
-          expect(conn).to receive(:description).at_least(:once).and_return(description)
-        end
-      end
-
-      let(:secondary_server_single) do
-        double('secondary_server').tap do |server|
-          expect(server).to receive(:with_connection).and_yield(connection)
-          expect(server).to receive(:cluster) { cluster_single }
-        end
-      end
-
-      let(:message) do
-        secondary_server_single.with_connection do |connection|
-          query.send(:message, connection)
-        end
-      end
-
-      before do
-        allow(connection).to receive(:server) { secondary_server_single }
-      end
-
-      it 'applies the correct flags' do
-        expect(message.flags).to eq([ :no_cursor_timeout, :secondary_ok ])
-      end
-    end
-
-    context "when the document contains an 'ok' field" do
-
-      before do
-        authorized_collection.insert_one(ok: false)
-      end
-
-      after do
-        authorized_collection.delete_many
-      end
-
-      it 'does not raise an exception' do
-        authorized_primary.with_connection do |connection|
-          expect(op.execute(connection, context: context)).to be_a(Mongo::Operation::Find::Legacy::Result)
-        end
-      end
-    end
-  end
-end
-

data/spec/mongo/operation/get_more_spec.rb

@@ -1,63 +0,0 @@
-# frozen_string_literal: true
-# encoding: utf-8
-
-require 'spec_helper'
-
-describe Mongo::Operation::GetMore::Legacy do
-
-  let(:to_return) do
-    50
-  end
-
-  let(:cursor_id) do
-    1
-  end
-
-  let(:spec) do
-    { :db_name   => SpecConfig.instance.test_db,
-      :coll_name => TEST_COLL,
-      :to_return => to_return,
-      :cursor_id => cursor_id }
-  end
-
-  let(:op) { described_class.new(spec) }
-
-  let(:context) { Mongo::Operation::Context.new }
-
-  describe '#initialize' do
-
-    it 'sets the spec' do
-      expect(op.spec).to be(spec)
-    end
-  end
-
-  describe '#==' do
-
-    context ' when two ops have different specs' do
-      let(:other_spec) do
-        { :db_name   => 'test_db',
-          :coll_name => 'test_coll',
-          :to_return => 50,
-          :cursor_id => 2 }
-      end
-      let(:other) { described_class.new(other_spec) }
-
-      it 'returns false' do
-        expect(op).not_to eq(other)
-      end
-    end
-  end
-
-  describe '#message' do
-
-    it 'creates a get more wire protocol message with correct specs' do
-      expect(Mongo::Protocol::GetMore).to receive(:new).with(SpecConfig.instance.test_db, TEST_COLL, to_return, cursor_id).and_call_original
-      begin
-        authorized_primary.with_connection do |connection|
-          op.execute(connection, context: context)
-        end
-      rescue
-      end
-    end
-  end
-end