mongo 2.0.6 → 2.1.0

data/spec/mongo/cluster/topology/replica_set_spec.rb

@@ -6,22 +6,30 @@ describe Mongo::Cluster::Topology::ReplicaSet do
     Mongo::Address.new('127.0.0.1:27017')
   end
 
+  let(:listeners) do
+    Mongo::Event::Listeners.new
+  end
+
+  let(:monitoring) do
+    Mongo::Monitoring.new
+  end
+
   describe '#servers' do
 
     let(:mongos) do
-      Mongo::Server.new(address, double('cluster'), Mongo::Event::Listeners.new, TEST_OPTIONS)
+      Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
     end
 
     let(:standalone) do
-      Mongo::Server.new(address, double('cluster'), Mongo::Event::Listeners.new, TEST_OPTIONS)
+      Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
     end
 
     let(:replica_set) do
-      Mongo::Server.new(address, double('cluster'), Mongo::Event::Listeners.new, TEST_OPTIONS)
+      Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
     end
 
     let(:replica_set_two) do
-      Mongo::Server.new(address, double('cluster'), Mongo::Event::Listeners.new, TEST_OPTIONS)
+      Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
     end
 
     let(:mongos_description) do
@@ -102,11 +110,11 @@ describe Mongo::Cluster::Topology::ReplicaSet do
   describe '#add_hosts?' do
 
     let(:primary) do
-      Mongo::Server.new(address, double('cluster'), Mongo::Event::Listeners.new, TEST_OPTIONS)
+      Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
     end
 
     let(:secondary) do
-      Mongo::Server.new(address, double('cluster'), Mongo::Event::Listeners.new, TEST_OPTIONS)
+      Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
     end
 
     let(:primary_description) do
@@ -183,7 +191,7 @@ describe Mongo::Cluster::Topology::ReplicaSet do
   describe '#remove_hosts?' do
 
     let(:primary) do
-      Mongo::Server.new(address, double('cluster'), Mongo::Event::Listeners.new, TEST_OPTIONS)
+      Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
     end
 
     let(:primary_description) do
@@ -231,6 +239,7 @@ describe Mongo::Cluster::Topology::ReplicaSet do
         double('description').tap do |d|
           allow(d).to receive(:config).and_return({ 'ismaster' => true })
           allow(d).to receive(:primary?).and_return(false)
+          allow(d).to receive(:me_mismatch?).and_return(false)
           allow(d).to receive(:hosts).and_return([])
         end
       end
@@ -249,6 +258,7 @@ describe Mongo::Cluster::Topology::ReplicaSet do
           allow(d).to receive(:hosts).and_return([ primary ])
           allow(d).to receive(:replica_set_name).and_return('test')
           allow(d).to receive(:replica_set_member?).and_return(true)
+          allow(d).to receive(:me_mismatch?).and_return(false)
         end
       end
 
@@ -262,7 +272,7 @@ describe Mongo::Cluster::Topology::ReplicaSet do
   describe '#remove_server?' do
 
     let(:secondary) do
-      Mongo::Server.new(address, double('cluster'), Mongo::Event::Listeners.new, TEST_OPTIONS)
+      Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
     end
 
     let(:secondary_description) do
@@ -346,6 +356,5 @@ describe Mongo::Cluster::Topology::ReplicaSet do
         expect(topology.remove_server?(description, secondary)).to eq(false)
       end
     end
-
   end
 end

data/spec/mongo/cluster/topology/sharded_spec.rb

@@ -10,17 +10,24 @@ describe Mongo::Cluster::Topology::Sharded do
     described_class.new({})
   end
 
+  let(:monitoring) do
+    Mongo::Monitoring.new
+  end
+
+  let(:listeners) do
+    Mongo::Event::Listeners.new
+  end
 
   let(:mongos) do
-    Mongo::Server.new(address, double('cluster'), Mongo::Event::Listeners.new, TEST_OPTIONS)
+    Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
   end
 
   let(:standalone) do
-    Mongo::Server.new(address, double('cluster'), Mongo::Event::Listeners.new, TEST_OPTIONS)
+    Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
   end
 
   let(:replica_set) do
-    Mongo::Server.new(address, double('cluster'), Mongo::Event::Listeners.new, TEST_OPTIONS)
+    Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
   end
 
   let(:mongos_description) do
@@ -99,6 +106,7 @@ describe Mongo::Cluster::Topology::Sharded do
       let(:description) do
         double('description').tap do |d|
           allow(d).to receive(:mongos?).and_return(false)
+          allow(d).to receive(:unknown?).and_return(false)
           allow(d).to receive(:is_server?).and_return(true)
         end
       end

data/spec/mongo/cluster/topology/single_spec.rb

@@ -10,22 +10,30 @@ describe Mongo::Cluster::Topology::Single do
     described_class.new({})
   end
 
+  let(:monitoring) do
+    Mongo::Monitoring.new
+  end
+
+  let(:listeners) do
+    Mongo::Event::Listeners.new
+  end
+
   describe '.servers' do
 
     let(:mongos) do
-      Mongo::Server.new(address, double('cluster'), Mongo::Event::Listeners.new, TEST_OPTIONS)
+      Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
     end
 
     let(:standalone) do
-      Mongo::Server.new(address, double('cluster'), Mongo::Event::Listeners.new, TEST_OPTIONS)
+      Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
     end
 
     let(:standalone_two) do
-      Mongo::Server.new(address, double('cluster'), Mongo::Event::Listeners.new, TEST_OPTIONS)
+      Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
     end
 
     let(:replica_set) do
-      Mongo::Server.new(address, double('cluster'), Mongo::Event::Listeners.new, TEST_OPTIONS)
+      Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
     end
 
     let(:mongos_description) do

data/spec/mongo/cluster_spec.rb

@@ -2,8 +2,12 @@ require 'spec_helper'
 
 describe Mongo::Cluster do
 
+  let(:monitoring) do
+    Mongo::Monitoring.new
+  end
+
   let(:cluster) do
-    described_class.new(ADDRESSES, TEST_OPTIONS)
+    described_class.new(ADDRESSES, monitoring, TEST_OPTIONS)
   end
 
   describe '#==' do
@@ -15,7 +19,7 @@ describe Mongo::Cluster do
         context 'when the options are the same' do
 
           let(:other) do
-            described_class.new(ADDRESSES, TEST_OPTIONS)
+            described_class.new(ADDRESSES, monitoring, TEST_OPTIONS)
           end
 
           it 'returns true' do
@@ -26,7 +30,7 @@ describe Mongo::Cluster do
         context 'when the options are not the same' do
 
           let(:other) do
-            described_class.new([ '127.0.0.1:27017' ], TEST_OPTIONS.merge(:replica_set => 'test'))
+            described_class.new([ '127.0.0.1:27017' ], monitoring, TEST_OPTIONS.merge(:replica_set => 'test'))
           end
 
           it 'returns false' do
@@ -38,7 +42,7 @@ describe Mongo::Cluster do
       context 'when the addresses are not the same' do
 
         let(:other) do
-          described_class.new([ '127.0.0.1:27018' ], TEST_OPTIONS)
+          described_class.new([ '127.0.0.1:27018' ], monitoring, TEST_OPTIONS)
         end
 
         it 'returns false' do
@@ -76,8 +80,11 @@ describe Mongo::Cluster do
     context 'when the option is provided' do
 
       let(:cluster) do
-        described_class.new([ '127.0.0.1:27017' ], TEST_OPTIONS.merge(:connect => :replica_set,
-                                                                      :replica_set => 'testing'))
+        described_class.new(
+          [ '127.0.0.1:27017' ],
+          monitoring,
+          TEST_OPTIONS.merge(:connect => :replica_set, :replica_set => 'testing')
+        )
       end
 
       it 'returns the name' do
@@ -88,7 +95,7 @@ describe Mongo::Cluster do
     context 'when the option is not provided' do
 
       let(:cluster) do
-        described_class.new([ '127.0.0.1:27017' ], TEST_OPTIONS.dup.delete_if { |k| k == :replica_set })
+        described_class.new([ '127.0.0.1:27017' ], monitoring, TEST_OPTIONS.dup.delete_if { |k| k == :replica_set })
       end
 
       it 'returns nil' do
@@ -210,6 +217,36 @@ describe Mongo::Cluster do
     end
   end
 
+  describe '#disconnect!' do
+
+    let(:known_servers) do
+      cluster.instance_variable_get(:@servers)
+    end
+
+    before do
+      known_servers.each do |server|
+        expect(server).to receive(:disconnect!).and_call_original
+      end
+    end
+
+    it 'disconnects each server and returns true' do
+      expect(cluster.disconnect!).to be(true)
+    end
+  end
+
+  describe '#reconnect!' do
+
+    before do
+      cluster.servers.each do |server|
+        expect(server).to receive(:reconnect!).and_call_original
+      end
+    end
+
+    it 'reconnects each server and returns true' do
+      expect(cluster.reconnect!).to be(true)
+    end
+  end
+
   describe '#remove' do
 
     let(:address_a) do
@@ -220,12 +257,16 @@ describe Mongo::Cluster do
       Mongo::Address.new('127.0.0.1:27018')
     end
 
+    let(:monitoring) do
+      Mongo::Monitoring.new
+    end
+
     let(:server_a) do
-      Mongo::Server.new(address_a, cluster, Mongo::Event::Listeners.new)
+      Mongo::Server.new(address_a, cluster, monitoring, Mongo::Event::Listeners.new)
     end
 
     let(:server_b) do
-      Mongo::Server.new(address_b, cluster, Mongo::Event::Listeners.new)
+      Mongo::Server.new(address_b, cluster, monitoring, Mongo::Event::Listeners.new)
     end
 
     let(:servers) do
@@ -309,8 +350,12 @@ describe Mongo::Cluster do
       Mongo::Address.new('127.0.0.1:27017')
     end
 
+    let(:monitoring) do
+      Mongo::Monitoring.new
+    end
+
     let(:server) do
-      Mongo::Server.new(address, cluster, listeners)
+      Mongo::Server.new(address, cluster, monitoring, listeners)
     end
 
     let(:servers) do

data/spec/mongo/collection/view/aggregation_spec.rb

@@ -61,7 +61,7 @@ describe Mongo::Collection::View::Aggregation do
     end
 
     after do
-      authorized_collection.find.delete_many
+      authorized_collection.delete_many
     end
 
     context 'when a block is provided' do
@@ -176,6 +176,193 @@ describe Mongo::Collection::View::Aggregation do
       it 'includes the option in the spec' do
         expect(aggregation.send(:aggregate_spec)[:selector][:allowDiskUse]).to eq(true)
       end
+
+      context 'when allow_disk_use is specified as an option' do
+
+        let(:options) do
+          { :allow_disk_use => true }
+        end
+
+        let(:aggregation) do
+          described_class.new(view, pipeline, options)
+        end
+
+        it 'includes the option in the spec' do
+          expect(aggregation.send(:aggregate_spec)[:selector][:allowDiskUse]).to eq(true)
+        end
+
+        context 'when #allow_disk_use is also called' do
+
+          let(:options) do
+            { :allow_disk_use => true }
+          end
+
+          let(:aggregation) do
+            described_class.new(view, pipeline, options).allow_disk_use(false)
+          end
+
+          it 'overrides the first option with the second' do
+            expect(aggregation.send(:aggregate_spec)[:selector][:allowDiskUse]).to eq(false)
+          end
+        end
+      end
+    end
+
+    context 'when max_time_ms is an option' do
+
+      let(:options) do
+        { :max_time_ms => 100 }
+      end
+
+      it 'includes the option in the spec' do
+        expect(aggregation.send(:aggregate_spec)[:selector][:maxTimeMS]).to eq(options[:max_time_ms])
+      end
+    end
+
+    context 'when batch_size is set' do
+
+      context 'when batch_size is set on the view' do
+
+        let(:view_options) do
+          { :batch_size => 10 }
+        end
+
+        it 'uses the batch_size on the view' do
+          expect(aggregation.send(:aggregate_spec)[:selector][:cursor][:batchSize]).to eq(view_options[:batch_size])
+        end
+      end
+
+      context 'when batch_size is provided in the options' do
+
+        let(:options) do
+          { :batch_size => 20 }
+        end
+
+        it 'includes the option in the spec' do
+          expect(aggregation.send(:aggregate_spec)[:selector][:cursor][:batchSize]).to eq(options[:batch_size])
+        end
+
+        context 'when  batch_size is also set on the view' do
+
+          let(:view_options) do
+            { :batch_size => 10 }
+          end
+
+          it 'overrides the view batch_size with the option batch_size' do
+            expect(aggregation.send(:aggregate_spec)[:selector][:cursor][:batchSize]).to eq(options[:batch_size])
+          end
+        end
+      end
+    end
+
+    context 'when use_cursor is set' do
+
+      context 'when use_cursor is true' do
+
+        context 'when batch_size is set' do
+
+          let(:options) do
+            { :use_cursor => true,
+              :batch_size => 10
+            }
+          end
+
+          it 'sets a batch size document in the spec' do
+            expect(aggregation.send(:aggregate_spec)[:selector][:cursor][:batchSize]).to eq(options[:batch_size])
+          end
+        end
+
+        context 'when batch_size is not set' do
+
+          let(:options) do
+            { :use_cursor => true }
+          end
+
+          it 'sets an empty document in the spec' do
+            expect(aggregation.send(:aggregate_spec)[:selector][:cursor]).to eq({})
+          end
+        end
+
+      end
+
+      context 'when use_cursor is false' do
+
+        let(:options) do
+          { :use_cursor => false }
+        end
+
+        context 'when batch_size is set' do
+
+          it 'does not set the cursor option in the spec' do
+            expect(aggregation.send(:aggregate_spec)[:selector][:cursor]).to be_nil
+          end
+        end
+      end
+    end
+  end
+
+  context 'when $out is in the pipeline', if: write_command_enabled? do
+
+    let(:pipeline) do
+      [{
+           "$group" => {
+               "_id" => "$city",
+               "totalpop" => { "$sum" => "$pop" }
+           }
+       },
+       {
+           '$out' => 'output_collection'
+       }
+      ]
+    end
+
+    after do
+      authorized_client['output_collection'].delete_many
+    end
+
+    context 'when $out is a string' do
+
+      it 'does not allow the operation on a secondary' do
+        expect(aggregation.send(:secondary_ok?)).to be(false)
+      end
+    end
+
+    context 'when $out is a symbol' do
+
+      let(:pipeline) do
+        [{
+             "$group" => {
+                 "_id" => "$city",
+                 "totalpop" => { "$sum" => "$pop" }
+             }
+         },
+         {
+             :$out => 'output_collection'
+         }
+        ]
+      end
+
+      it 'does not allow the operation on a secondary' do
+        expect(aggregation.send(:secondary_ok?)).to be(false)
+      end
+    end
+
+
+    context 'when the context is not a valid server for writing' do
+
+     it 'reroutes the operation to a primary' do
+       allow(aggregation).to receive(:valid_server?).and_return(false)
+       expect(Mongo::Logger.logger).to receive(:warn?).and_call_original
+       aggregation.to_a
+     end
+    end
+
+    context 'when the context is a valid server for writing' do
+
+     it 'does not reroute the operation to a primary' do
+       expect(Mongo::Logger.logger).not_to receive(:warn?)
+       aggregation.to_a
+     end
     end
   end
 end

data/spec/mongo/collection/view/explainable_spec.rb

@@ -15,7 +15,7 @@ describe Mongo::Collection::View::Explainable do
   end
 
   after do
-    authorized_collection.find.delete_many
+    authorized_collection.delete_many
   end
 
   describe '#explain' do

data/spec/mongo/collection/view/immutable_spec.rb

@@ -0,0 +1,103 @@
+require 'spec_helper'
+
+describe Mongo::Collection::View::Immutable do
+
+  let(:selector) do
+    {}
+  end
+
+  let(:options) do
+    {}
+  end
+
+  let(:view) do
+    Mongo::Collection::View.new(authorized_collection, selector, options)
+  end
+
+  after do
+    authorized_collection.delete_many
+  end
+
+  describe '#configure' do
+
+    context 'when the options has a modifiers document' do
+
+      let(:options) do
+        { modifiers: { :$maxTimeMS => 500 } }
+      end
+
+      let(:new_view) do
+        view.projection(_id: 1)
+      end
+
+      it 'returns a new view' do
+        expect(view).not_to be(new_view)
+      end
+
+      it 'creates a new options hash' do
+        expect(view.options).not_to be(new_view.options)
+      end
+
+      it 'keeps the modifier fields already in the options hash' do
+        expect(new_view.modifiers[:$maxTimeMS]).to eq(500)
+      end
+
+      it 'sets the option' do
+        expect(new_view.projection).to eq(_id: 1)
+      end
+
+      it 'creates a new modifiers document' do
+        expect(view.modifiers).not_to be(new_view.modifiers)
+      end
+    end
+  end
+
+  describe '#configure_modifier' do
+
+    let(:new_view) do
+      view.sort('x' => Mongo::Index::ASCENDING)
+    end
+
+    context 'when the options does not have a modifiers document' do
+
+      it 'returns a new view' do
+        expect(view).not_to be(new_view)
+      end
+
+      it 'returns a new view with the modifiers document containing the option' do
+        expect(new_view.modifiers[:$orderby]).to eq({ 'x' => Mongo::Index::ASCENDING })
+      end
+    end
+
+    context 'when the options has a modifiers document' do
+
+      let(:options) do
+        { modifiers: { :$maxTimeMS => 500 } }
+      end
+
+      it 'returns a new view' do
+        expect(view).not_to be(new_view)
+      end
+
+      it 'creates a new options hash' do
+        expect(view.options).not_to be(new_view.options)
+      end
+
+      it 'keeps the fields already in the options hash and merges in the new one' do
+        expect(new_view.modifiers[:$maxTimeMS]).to eq(500)
+      end
+
+      it 'sets the new value in the new view modifier document' do
+        expect(new_view.modifiers[:$orderby]).to eq('x' => Mongo::Index::ASCENDING)
+      end
+
+      it 'returns that value when the corresponding option method is called' do
+        expect(new_view.sort).to eq({ 'x' => Mongo::Index::ASCENDING })
+      end
+
+      it 'creates a new modifiers document' do
+        expect(view.modifiers).not_to be(new_view.modifiers)
+      end
+    end
+  end
+end