mongo 1.12.5 → 2.0.0.beta

data/spec/support/server_selection/selection/ReplicaSetWithPrimary/read/Nearest.yml

@@ -0,0 +1,41 @@
+---
+topology_description:
+  type: ReplicaSetWithPrimary
+  servers:
+  - &1
+    address: b:27017
+    avg_rtt_ms: 5
+    type: RSSecondary
+    tag_sets:
+    - data_center: nyc
+  - &2
+    address: c:27017
+    avg_rtt_ms: 100
+    type: RSSecondary
+    tag_sets:
+    - data_center: nyc
+  - &3
+    address: a:27017
+    avg_rtt_ms: 26
+    type: RSPrimary
+    tag_sets:
+    - data_center: nyc
+operation: read
+read_preference:
+  mode: Nearest
+  tag_sets:
+  - data_center: nyc
+candidate_servers:
+- *1
+- *2
+- *3
+eligible_servers:
+- *1
+- *2
+- *3
+suitable_servers:
+- *1
+- *3
+- *2
+in_latency_window:
+- *1

data/spec/support/server_selection/selection/ReplicaSetWithPrimary/read/Nearest_non_matching.yml

@@ -0,0 +1,34 @@
+---
+topology_description:
+  type: ReplicaSetWithPrimary
+  servers:
+  - &1
+    address: b:27017
+    avg_rtt_ms: 5
+    type: RSSecondary
+    tag_sets:
+    - data_center: nyc
+  - &2
+    address: c:27017
+    avg_rtt_ms: 100
+    type: RSSecondary
+    tag_sets:
+    - data_center: nyc
+  - &3
+    address: a:27017
+    avg_rtt_ms: 26
+    type: RSPrimary
+    tag_sets:
+    - data_center: nyc
+operation: read
+read_preference:
+  mode: Nearest
+  tag_sets:
+  - data_center: sf
+candidate_servers:
+- *1
+- *2
+- *3
+eligible_servers: []
+suitable_servers: []
+in_latency_window: []

data/spec/support/server_selection/selection/ReplicaSetWithPrimary/read/Primary.yml

@@ -0,0 +1,33 @@
+---
+topology_description:
+  type: ReplicaSetWithPrimary
+  servers:
+  - address: b:27017
+    avg_rtt_ms: 5
+    type: RSSecondary
+    tag_sets:
+    - data_center: nyc
+  - address: c:27017
+    avg_rtt_ms: 100
+    type: RSSecondary
+    tag_sets:
+    - data_center: nyc
+  - &1
+    address: a:27017
+    avg_rtt_ms: 26
+    type: RSPrimary
+    tag_sets:
+    - data_center: nyc
+operation: read
+read_preference:
+  mode: Primary
+  tag_sets:
+  - {}
+candidate_servers:
+- *1
+eligible_servers:
+- *1
+suitable_servers:
+- *1
+in_latency_window:
+- *1

data/spec/support/server_selection/selection/ReplicaSetWithPrimary/read/PrimaryPreferred.yml

@@ -0,0 +1,39 @@
+---
+topology_description:
+  type: ReplicaSetWithPrimary
+  servers:
+  - &1
+    address: b:27017
+    avg_rtt_ms: 5
+    type: RSSecondary
+    tag_sets:
+    - data_center: nyc
+  - &2
+    address: c:27017
+    avg_rtt_ms: 100
+    type: RSSecondary
+    tag_sets:
+    - data_center: nyc
+  - &3
+    address: a:27017
+    avg_rtt_ms: 26
+    type: RSPrimary
+    tag_sets:
+    - data_center: nyc
+operation: read
+read_preference:
+  mode: PrimaryPreferred
+  tag_sets:
+  - {}
+candidate_servers:
+- *1
+- *2
+- *3
+eligible_servers:
+- *1
+- *2
+- *3
+suitable_servers:
+- *3
+in_latency_window:
+- *3

data/spec/support/server_selection/selection/ReplicaSetWithPrimary/read/PrimaryPreferred_non_matching.yml

@@ -0,0 +1,36 @@
+---
+topology_description:
+  type: ReplicaSetWithPrimary
+  servers:
+  - &1
+    address: b:27017
+    avg_rtt_ms: 5
+    type: RSSecondary
+    tag_sets:
+    - data_center: nyc
+  - &2
+    address: c:27017
+    avg_rtt_ms: 100
+    type: RSSecondary
+    tag_sets:
+    - data_center: nyc
+  - &3
+    address: a:27017
+    avg_rtt_ms: 26
+    type: RSPrimary
+    tag_sets:
+    - data_center: nyc
+operation: read
+read_preference:
+  mode: PrimaryPreferred
+  tag_sets:
+  - data_center: sf
+candidate_servers:
+- *1
+- *2
+- *3
+eligible_servers: []
+suitable_servers:
+- *3
+in_latency_window:
+- *3

data/spec/support/server_selection/selection/Sharded/read/SecondaryPreferred.yml

@@ -0,0 +1,32 @@
+---
+topology_description:
+  type: Sharded
+  servers:
+  - &1
+    address: g:27017
+    avg_rtt_ms: 5
+    type: Mongos
+    tag_sets:
+    - data_center: nyc
+  - &2
+    address: h:27017
+    avg_rtt_ms: 35
+    type: Mongos
+    tag_sets:
+    - data_center: dc
+operation: read
+read_preference:
+  mode: SecondaryPreferred
+  tag_sets:
+  - data_center: nyc
+candidate_servers:
+- *1
+- *2
+eligible_servers:
+- *1
+- *2
+suitable_servers:
+- *1
+- *2
+in_latency_window:
+- *1

data/spec/support/server_selection/selection/Single/read/SecondaryPreferred.yml

@@ -0,0 +1,23 @@
+---
+topology_description:
+  type: Single
+  servers:
+  - &1
+    address: a:27017
+    avg_rtt_ms: 5
+    type: Standalone
+    tag_sets:
+    - data_center: dc
+operation: read
+read_preference:
+  mode: SecondaryPreferred
+  tag_sets:
+  - data_center: nyc
+candidate_servers:
+- *1
+eligible_servers:
+- *1
+suitable_servers:
+- *1
+in_latency_window:
+- *1

data/spec/support/server_selection/selection/Unknown/read/SecondaryPreferred.yml

@@ -0,0 +1,13 @@
+---
+topology_description:
+  type: Unknown
+  servers: []
+operation: read
+read_preference:
+  mode: SecondaryPreferred
+  tag_sets:
+  - data_center: nyc
+candidate_servers: []
+eligible_servers: []
+suitable_servers: []
+in_latency_window: []

data/spec/support/server_selection_rtt.rb

@@ -0,0 +1,41 @@
+module Mongo
+  module ServerSelection
+    module RTT
+
+      # Represents a specification.
+      #
+      # @since 2.0.0
+      class Spec
+
+        # @return [ String ] description The spec description.
+        attr_reader :description
+
+        # @return [ Float ] avg_rtt_ms The starting average round trip time.
+        attr_reader :avg_rtt_ms
+
+        # @return [ Float ] new_rtt_ms The new round trip time for ismaster.
+        attr_reader :new_rtt_ms
+
+        # @return [ Float ] new_avg_rtt The newly calculated moving average round trip time.
+        attr_reader :new_avg_rtt
+
+        # Instantiate the new spec.
+        #
+        # @example Create the spec.
+        #   Spec.new(file)
+        #
+        # @param [ String ] file The name of the file.
+        #
+        # @since 2.0.0
+        def initialize(file)
+          @test = YAML.load(ERB.new(File.new(file).read).result)
+          @description = "avg_rtt_ms: #{@test['avg_rtt_ms']}, new_rtt_ms: #{@test['new_rtt_ms']}," +
+                           " new_avg_rtt: #{@test['new_avg_rtt']}"
+          @avg_rtt_ms = @test['avg_rtt_ms'] == 'NULL' ? nil : @test['avg_rtt_ms'].to_f
+          @new_rtt_ms = @test['new_rtt_ms'].to_f
+          @new_avg_rtt = @test['new_avg_rtt'].to_f
+        end
+      end
+    end
+  end
+end

data/spec/support/shared/bulk_write.rb

@@ -0,0 +1,498 @@
+shared_examples 'a bulk write object' do
+
+  context 'when no operations are provided' do
+
+    let(:operations) {[]}
+
+    it 'raises an error' do
+      expect {
+        bulk.execute
+      }.to raise_error(Mongo::Error::EmptyBatch)
+    end
+  end
+
+  context 'when invalid operations are provided' do
+
+    let(:operations) do
+      [{ :not_an_op => {}}]
+    end
+
+    it 'raises an error' do
+      expect {
+        bulk.execute
+      }.to raise_error(Mongo::Error::InvalidBulkOperation)
+    end
+  end
+
+  context 'when an insert_one operation is provided' do
+
+    context 'when a document is provided' do
+
+      let(:operations) do
+        [{ insert_one: { name: 'test' }}]
+      end
+
+      it 'returns nInserted of 1' do
+        expect(bulk.execute['nInserted']).to eq(1)
+      end
+
+      it 'only inserts that document' do
+        bulk.execute
+        expect(authorized_collection.find.first['name']).to eq('test')
+      end
+    end
+
+    context 'when an invalid object is provided' do
+
+      let(:operations) do
+        [{ insert_one: [] }]
+      end
+
+      it 'raises an exception' do
+        expect {
+          bulk.execute
+        }.to raise_error(Mongo::Error::InvalidDocument)
+      end
+    end
+  end
+
+  context 'delete_one' do
+
+    let(:docs) do
+      [ { a: 1 }, { a: 1 } ]
+    end
+
+    let(:expected) do
+      [{ 'a' => 1 }]
+    end
+
+    before do
+      authorized_collection.insert_many(docs)
+    end
+
+    let(:operations) do
+      [{ delete_one: { a: 1 }}]
+    end
+
+    context 'when no selector is specified' do
+
+      let(:operations) do
+        [{ delete_one: nil }]
+      end
+
+      it 'raises an exception' do
+        expect {
+          bulk.execute
+        }.to raise_exception(Mongo::Error::InvalidDocument)
+      end
+    end
+
+    context 'when multiple documents match delete selector' do
+
+      it 'reports nRemoved correctly' do
+        expect(bulk.execute['nRemoved']).to eq(1)
+      end
+
+      it 'deletes only matching documents' do
+        bulk.execute
+        expect(authorized_collection.find.projection(_id: 0).to_a).to eq(expected)
+      end
+    end
+  end
+
+  context 'when a delete_many operation is provided' do
+
+    let(:docs) do
+      [{ a: 1 }, { a: 1 }]
+    end
+
+    before do
+      authorized_collection.insert_many(docs)
+    end
+
+    let(:operations) do
+      [{ delete_many: { a: 1 }}]
+    end
+
+    context 'when no selector is specified' do
+
+      let(:operations) do
+        [{ delete_many: nil }]
+      end
+
+      it 'raises an exception' do
+        expect {
+          bulk.execute
+        }.to raise_exception(Mongo::Error::InvalidDocument)
+      end
+    end
+
+    context 'when a selector is specified' do
+
+      context 'when multiple documents match delete selector' do
+
+        it 'reports nRemoved correctly' do
+          expect(bulk.execute['nRemoved']).to eq(2)
+        end
+
+        it 'deletes all matching documents' do
+          bulk.execute
+          expect(authorized_collection.find.to_a).to be_empty
+        end
+      end
+
+      context 'when only one document matches delete selector' do
+
+        let(:docs) do
+          [{ a: 1 }, { a: 2 }]
+        end
+
+        let(:expected) do
+          [{ 'a' => 2 }]
+        end
+
+        it 'reports nRemoved correctly' do
+          expect(bulk.execute['nRemoved']).to eq(1)
+        end
+
+        it 'deletes all matching documents' do
+          bulk.execute
+          expect(authorized_collection.find.projection(_id: 0).to_a).to eq(expected)
+        end
+      end
+    end
+  end
+
+  context 'when a replace_one operation is provided' do
+
+    let(:docs) do
+      [{ a: 1 }, { a: 1 }]
+    end
+
+    let(:expected) do
+      [{ 'a' => 2 }, { 'a' => 1 }]
+    end
+
+    before do
+      authorized_collection.insert_many(docs)
+    end
+
+    let(:replacement) do
+      { a: 2 }
+    end
+
+    let(:operations) do
+      [{ replace_one: [{ a: 1 }, replacement ]}]
+    end
+
+    context 'when a replace document is not specified' do
+
+      let(:operations) do
+        [{ replace_one: [{ a: 1 }]}]
+      end
+
+      it 'raises an exception' do
+        expect {
+          bulk.execute
+        }.to raise_exception(ArgumentError)
+      end
+    end
+
+    context 'when there are $-operator top-level keys' do
+
+      let(:replacement) do
+        { :$set => { a: 3 }}
+      end
+
+      it 'raises an exception' do
+        expect {
+          bulk.execute
+        }.to raise_exception(Mongo::Error::InvalidReplacementDocument)
+      end
+
+    end
+
+    context 'when a replace document is specified' do
+
+      it 'applies the replacement to only one matching document' do
+        bulk.execute
+        expect(authorized_collection.find(replacement).count).to eq(1)
+      end
+
+      it 'reports nMatched correctly' do
+        expect(bulk.execute['nMatched']).to eq(1)
+      end
+
+      it 'only applies the replacement to one matching document' do
+        bulk.execute
+        expect(authorized_collection.find.projection(_id: 0).to_a).to eq(expected)
+      end
+
+      context 'when upsert is true' do
+
+        let(:operations) do
+          [{ replace_one: [{ a: 4 }, replacement, { :upsert => true }]}]
+        end
+
+        let(:expected) do
+          [{ 'a' => 1 }, { 'a' => 1 }, { 'a' => 2 }]
+        end
+
+        it 'upserts the replacement document' do
+          bulk.execute
+          expect(authorized_collection.find(replacement).count).to eq(1)
+        end
+
+        it 'reports nMatched correctly' do
+          expect(bulk.execute['nMatched']).to eq(0)
+        end
+
+        it 'does not replace any documents' do
+          bulk.execute
+          expect(authorized_collection.find.projection(_id: 0).to_a).to eq(expected)
+        end
+      end
+    end
+  end
+
+  context 'when an update_one operation is provided' do
+
+    let(:docs) do
+      [{ a: 1 }, { a: 1 }]
+    end
+
+    let(:update) do
+      { :$set => { a: 2 }}
+    end
+
+    let(:operations) do
+      [{ update_one: [{ a: 1 }, update ]}]
+    end
+
+    before do
+      authorized_collection.insert_many(docs)
+    end
+
+    let(:expected) do
+      [{ 'a' => 2 },  { 'a' => 1 }]
+    end
+
+    context 'when an update document is not specified' do
+
+      let(:operations) do
+        [{ update_one: [{ a: 1 }]}]
+      end
+
+      it 'raises an exception' do
+        expect {
+          bulk.execute
+        }.to raise_exception(ArgumentError)
+      end
+    end
+
+    context 'when an invalid update document is specified' do
+
+      let(:update) do
+        { a: 2 }
+      end
+
+      it 'raises an exception' do
+        expect {
+          bulk.execute
+        }.to raise_exception(Mongo::Error::InvalidUpdateDocument)
+      end
+    end
+
+    context 'when a valid update document is specified' do
+
+      it 'reports nModified correctly', if: write_command_enabled?  do
+        expect(bulk.execute['nModified']).to eq(1)
+      end
+
+      it 'reports nModified correctly', unless: write_command_enabled?  do
+        expect(bulk.execute['nModified']).to eq(nil)
+      end
+
+      it 'reports nUpserted correctly' do
+        expect(bulk.execute['nUpserted']).to eq(0)
+      end
+
+      it 'reports nMatched correctly' do
+        expect(bulk.execute['nMatched']).to eq(1)
+      end
+
+      it 'applies the correct writes' do
+        bulk.execute
+        expect(authorized_collection.find.projection(_id: 0).to_a).to eq(expected)
+      end
+
+      context 'when upsert is true' do
+
+        let(:operations) do
+          [{ update_one: [{ a: 3 }, update, { upsert: true }]}]
+        end
+
+        let(:expected) do
+          [{ 'a' => 1 },  { 'a' => 1 }, { 'a' => 2 }]
+        end
+
+        it 'reports nModified correctly', if: write_command_enabled?  do
+          expect(bulk.execute['nModified']).to eq(0)
+        end
+
+        it 'reports nModified correctly', unless: write_command_enabled?  do
+          expect(bulk.execute['nModified']).to eq(nil)
+        end
+
+        it 'reports nUpserted correctly' do
+          expect(bulk.execute['nUpserted']).to eq(1)
+        end
+
+        it 'reports nMatched correctly' do
+          expect(bulk.execute['nMatched']).to eq(0)
+        end
+
+        it 'applies the correct writes' do
+          bulk.execute
+          expect(authorized_collection.find.projection(_id: 0).to_a).to eq(expected)
+        end
+      end
+    end
+  end
+
+  context 'when an update_many operation is provided' do
+
+    let(:docs) do
+      [{ a: 1 }, { a: 1 }]
+    end
+
+    let(:update) do
+      { :$set => { a: 2 }}
+    end
+
+    let(:operations) do
+      [{ update_many: [{ a: 1 }, update ]}]
+    end
+
+    let(:expected) do
+      [{ 'a' => 2 },  { 'a' => 2 }]
+    end
+
+    before do
+      authorized_collection.insert_many(docs)
+    end
+
+    context 'when an update document is not specified' do
+
+      let(:operations) do
+        [{ update_many: [{ a: 1 }]}]
+      end
+
+      it 'raises an exception' do
+        expect do
+          bulk.execute
+        end.to raise_exception(ArgumentError)
+      end
+    end
+
+    context 'when an invalid update document is specified' do
+
+      let(:update) do
+        { a: 2 }
+      end
+
+      it 'raises an exception' do
+        expect do
+          bulk.execute
+        end.to raise_exception(Mongo::Error::InvalidUpdateDocument)
+      end
+    end
+
+    context 'when a valid update document is specified' do
+
+      it 'reports nModified correctly', if: write_command_enabled?  do
+        expect(bulk.execute['nModified']).to eq(2)
+      end
+
+      it 'reports nModified correctly', unless: write_command_enabled?  do
+        expect(bulk.execute['nModified']).to eq(nil)
+      end
+
+      it 'reports nUpserted correctly' do
+        expect(bulk.execute['nUpserted']).to eq(0)
+      end
+
+      it 'reports nMatched correctly' do
+        expect(bulk.execute['nMatched']).to eq(2)
+      end
+
+      it 'applies the correct writes' do
+        bulk.execute
+        expect(authorized_collection.find.projection(_id: 0).to_a).to eq(expected)
+      end
+
+      context 'when upsert is true' do
+
+        let(:operations) do
+          [{ update_one: [{ a: 3 }, update, { upsert: true }]}]
+        end
+
+        let(:expected) do
+          [ { 'a' => 1 },  { 'a' => 1 }, { 'a' => 2 } ]
+        end
+
+        it 'reports nModified correctly', if: write_command_enabled?  do
+          expect(bulk.execute['nModified']).to eq(0)
+        end
+
+        it 'reports nModified correctly', unless: write_command_enabled?  do
+          expect(bulk.execute['nModified']).to eq(nil)
+        end
+
+        it 'reports nUpserted correctly' do
+          expect(bulk.execute['nUpserted']).to eq(1)
+        end
+
+        it 'reports nMatched correctly' do
+          expect(bulk.execute['nMatched']).to eq(0)
+        end
+
+        it 'applies the correct writes' do
+          bulk.execute
+          expect(authorized_collection.find.projection(_id: 0).to_a).to eq(expected)
+        end
+      end
+    end
+  end
+
+  context 'when the operations need to be split' do
+
+    before do
+      authorized_collection.find.remove_many
+      6000.times do |i|
+        authorized_collection.insert_one(x: i)
+      end
+    end
+
+    let(:operations) do
+      [].tap do |ops|
+        3000.times do |i|
+          ops << { :update_one => [ { x: i },
+                                    {'$set' => { x: 6000-i } }]
+                 }
+        end
+        ops << { :insert_one => { test: 'emily' } }
+        3000.times do |i|
+          ops << { :update_one => [ { x: 3000+i },
+                                    {'$set' => { x: 3000-i } }]
+                 }
+        end
+      end
+    end
+
+    it 'completes all operations' do
+      bulk.execute
+      expect(authorized_collection.find(x: { '$lte' => 3000 }).to_a.size).to eq(3000)
+    end
+  end
+end