mongo 2.3.1 → 2.4.0.rc0

data/spec/mongo/client_spec.rb

@@ -260,6 +260,17 @@ describe Mongo::Client do
           expect(client.cluster.logger).to_not eq(Mongo::Logger.logger)
         end
       end
+
+      context 'when providing a heartbeat_frequency' do
+
+        let(:client) do
+          described_class.new(['127.0.0.1:27017'], :heartbeat_frequency => 2)
+        end
+
+        it 'sets the heartbeat frequency' do
+          expect(client.cluster.options[:heartbeat_frequency]).to eq(client.options[:heartbeat_frequency])
+        end
+      end
     end
 
     context 'when providing a connection string' do
@@ -557,6 +568,41 @@ describe Mongo::Client do
       end
     end
 
+    context 'when the app_name is changed' do
+
+      let(:client) do
+        authorized_client
+      end
+
+      let!(:original_options) do
+        client.options
+      end
+
+      let(:new_options) do
+        { app_name: 'reports' }
+      end
+
+      let!(:new_client) do
+        authorized_client.with(new_options)
+      end
+
+      it 'returns a new client' do
+        expect(new_client).not_to equal(client)
+      end
+
+      it 'replaces the existing options' do
+        expect(new_client.options).to eq(client.options.merge(new_options))
+      end
+
+      it 'does not modify the original client' do
+        expect(client.options).to eq(original_options)
+      end
+
+      it 'does not keep the same cluster' do
+        expect(new_client.cluster).not_to be(client.cluster)
+      end
+    end
+
     context 'when the write concern is not changed' do
 
       let(:client) do
@@ -595,7 +641,7 @@ describe Mongo::Client do
       end
 
       it 'keeps the same cluster' do
-        expect(new_client.cluster).to equal(client.cluster)
+        expect(new_client.cluster).to be(client.cluster)
       end
     end
 

data/spec/mongo/cluster/app_metadata_spec.rb

@@ -0,0 +1,104 @@
+require 'spec_helper'
+
+describe Mongo::Cluster::AppMetadata do
+
+  let(:app_metadata) do
+    described_class.new(cluster)
+  end
+
+  let(:cluster) do
+    authorized_client.cluster
+  end
+
+  describe '#initialize' do
+
+    context 'when the cluster has an app name option set' do
+
+      let(:cluster) do
+        authorized_client.with(app_name: :reports).cluster
+      end
+
+      it 'sets the app name' do
+        expect(app_metadata.send(:full_client_document)[:application][:name]).to eq(:reports)
+      end
+
+      context 'when the app name exceeds the max length of 128' do
+
+        let(:cluster) do
+          authorized_client.with(app_name: "\u3042"*43).cluster
+        end
+
+        it 'raises an error' do
+          expect {
+            app_metadata.send(:validate!)
+          }.to raise_exception(Mongo::Error::InvalidApplicationName)
+        end
+      end
+    end
+
+    context 'when the cluster does not have an app name option set' do
+
+      it 'does not set the app name' do
+        expect(app_metadata.send(:full_client_document)[:application]).to be(nil)
+      end
+    end
+
+    context 'when the client document exceeds the max of 512 bytes' do
+
+      context 'when the os.type length is too long' do
+
+        before do
+          allow(app_metadata).to receive(:type).and_return('x'*500)
+        end
+
+        it 'truncates the document' do
+          expect(app_metadata.send(:ismaster_bytes)).to be_a(String)
+        end
+      end
+
+      context 'when the os.name length is too long' do
+
+        before do
+          allow(app_metadata).to receive(:name).and_return('x'*500)
+        end
+
+        it 'truncates the document' do
+          expect(app_metadata.send(:ismaster_bytes)).to be_a(String)
+        end
+      end
+
+      context 'when the os.architecture length is too long' do
+
+        before do
+          allow(app_metadata).to receive(:architecture).and_return('x'*500)
+        end
+
+        it 'truncates the document' do
+          expect(app_metadata.send(:ismaster_bytes)).to be_a(String)
+        end
+      end
+
+      context 'when the platform length is too long' do
+
+        before do
+          allow(app_metadata).to receive(:platform).and_return('x'*500)
+        end
+
+        it 'truncates the document to be just an ismaster command' do
+          expect(app_metadata.send(:ismaster_bytes)).to be_a(String)
+        end
+      end
+
+      context 'when the driver info is too long' do
+
+        before do
+          allow(app_metadata).to receive(:driver_doc).and_return('x'*500)
+        end
+
+        it 'truncates the document to be just an ismaster command' do
+          expect(app_metadata.ismaster_bytes.length).to eq(Mongo::Server::Monitor::Connection::ISMASTER_BYTES.length)
+        end
+      end
+    end
+  end
+end

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

@@ -14,22 +14,28 @@ describe Mongo::Cluster::Topology::ReplicaSet do
     Mongo::Monitoring.new
   end
 
+  let(:cluster) do
+    double('cluster').tap do |cl|
+      allow(cl).to receive(:app_metadata).and_return(app_metadata)
+    end
+  end
+
   describe '#servers' do
 
     let(:mongos) do
-      Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
+      Mongo::Server.new(address, cluster, monitoring, listeners, TEST_OPTIONS)
     end
 
     let(:standalone) do
-      Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
+      Mongo::Server.new(address, cluster, monitoring, listeners, TEST_OPTIONS)
     end
 
     let(:replica_set) do
-      Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
+      Mongo::Server.new(address, cluster, monitoring, listeners, TEST_OPTIONS)
     end
 
     let(:replica_set_two) do
-      Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
+      Mongo::Server.new(address, cluster, monitoring, listeners, TEST_OPTIONS)
     end
 
     let(:mongos_description) do
@@ -110,11 +116,11 @@ describe Mongo::Cluster::Topology::ReplicaSet do
   describe '#add_hosts?' do
 
     let(:primary) do
-      Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
+      Mongo::Server.new(address, cluster, monitoring, listeners, TEST_OPTIONS)
     end
 
     let(:secondary) do
-      Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
+      Mongo::Server.new(address, cluster, monitoring, listeners, TEST_OPTIONS)
     end
 
     let(:primary_description) do
@@ -193,7 +199,7 @@ describe Mongo::Cluster::Topology::ReplicaSet do
   describe '#remove_hosts?' do
 
     let(:primary) do
-      Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
+      Mongo::Server.new(address, cluster, monitoring, listeners, TEST_OPTIONS)
     end
 
     let(:primary_description) do
@@ -274,7 +280,7 @@ describe Mongo::Cluster::Topology::ReplicaSet do
   describe '#remove_server?' do
 
     let(:secondary) do
-      Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
+      Mongo::Server.new(address, cluster, monitoring, listeners, TEST_OPTIONS)
     end
 
     let(:secondary_description) do

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

@@ -18,16 +18,22 @@ describe Mongo::Cluster::Topology::Sharded do
     Mongo::Event::Listeners.new
   end
 
+  let(:cluster) do
+    double('cluster').tap do |cl|
+      allow(cl).to receive(:app_metadata).and_return(app_metadata)
+    end
+  end
+
   let(:mongos) do
-    Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
+    Mongo::Server.new(address, cluster, monitoring, listeners, TEST_OPTIONS)
   end
 
   let(:standalone) do
-    Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
+    Mongo::Server.new(address, cluster, monitoring, listeners, TEST_OPTIONS)
   end
 
   let(:replica_set) do
-    Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
+    Mongo::Server.new(address, cluster, monitoring, listeners, TEST_OPTIONS)
   end
 
   let(:mongos_description) do

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

@@ -18,22 +18,28 @@ describe Mongo::Cluster::Topology::Single do
     Mongo::Event::Listeners.new
   end
 
+  let(:cluster) do
+    double('cluster').tap do |cl|
+      allow(cl).to receive(:app_metadata).and_return(app_metadata)
+    end
+  end
+
   describe '.servers' do
 
     let(:mongos) do
-      Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
+      Mongo::Server.new(address, cluster, monitoring, listeners, TEST_OPTIONS)
     end
 
     let(:standalone) do
-      Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
+      Mongo::Server.new(address, cluster, monitoring, listeners, TEST_OPTIONS)
     end
 
     let(:standalone_two) do
-      Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
+      Mongo::Server.new(address, cluster, monitoring, listeners, TEST_OPTIONS)
     end
 
     let(:replica_set) do
-      Mongo::Server.new(address, double('cluster'), monitoring, listeners, TEST_OPTIONS)
+      Mongo::Server.new(address, cluster, monitoring, listeners, TEST_OPTIONS)
     end
 
     let(:mongos_description) do

data/spec/mongo/cluster_spec.rb

@@ -463,4 +463,33 @@ describe Mongo::Cluster do
       expect(primary_candidates).to include(cluster.next_primary)
     end
   end
+
+  describe '#app_metadata' do
+
+    it 'returns an AppMetadata object' do
+      expect(cluster.app_metadata).to be_a(Mongo::Cluster::AppMetadata)
+    end
+
+    context 'when the client has an app_name set' do
+
+      let(:cluster) do
+        authorized_client.with(app_name: 'reports').cluster
+      end
+
+      it 'constructs an AppMetadata object with the app_name' do
+        expect(cluster.app_metadata.send(:full_client_document)[:application]).to eq('name' => 'reports')
+      end
+    end
+
+    context 'when the client does not have an app_name set' do
+
+      let(:cluster) do
+        authorized_client.cluster
+      end
+
+      it 'constructs an AppMetadata object with no app_name' do
+        expect(cluster.app_metadata.send(:full_client_document)[:application]).to be_nil
+      end
+    end
+  end
 end

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

@@ -26,6 +26,10 @@ describe Mongo::Collection::View::Aggregation do
     described_class.new(view, pipeline, options)
   end
 
+  after do
+    authorized_collection.delete_many
+  end
+
   describe '#allow_disk_use' do
 
     let(:new_agg) do
@@ -143,6 +147,31 @@ describe Mongo::Collection::View::Aggregation do
         expect(aggregation.to_a.size).to eq(2)
       end
     end
+
+    context 'when the view has a write concern' do
+
+      let(:collection) do
+        authorized_collection.with(write: { w: WRITE_CONCERN[:w]+1 })
+      end
+
+      let(:view) do
+        Mongo::Collection::View.new(collection, selector, view_options)
+      end
+
+      context 'when the server supports write concern on the aggregate command', if: collation_enabled? do
+
+        it 'does not apply the write concern' do
+          expect(aggregation.to_a.size).to eq(2)
+        end
+      end
+
+      context 'when the server does not support write concern on the aggregation command', unless: collation_enabled? do
+
+        it 'does not apply the write concern' do
+          expect(aggregation.to_a.size).to eq(2)
+        end
+      end
+    end
   end
 
   describe '#initialize' do
@@ -173,6 +202,41 @@ describe Mongo::Collection::View::Aggregation do
     it 'executes an explain' do
       expect(aggregation.explain).to_not be_empty
     end
+
+    context 'when a collation is specified' do
+
+      before do
+        authorized_collection.insert_many([ { name: 'bang' }, { name: 'bang' }])
+      end
+
+      let(:pipeline) do
+        [{ "$match" => { "name" => "BANG" } }]
+      end
+
+      let(:options) do
+        { collation: { locale: 'en_US', strength: 2 } }
+      end
+
+      let(:result) do
+        aggregation.explain['$cursor']['queryPlanner']['collation']['locale']
+      end
+
+      context 'when the server selected supports collations', if: collation_enabled? do
+
+        it 'applies the collation' do
+          expect(result).to eq('en_US')
+        end
+      end
+
+      context 'when the server selected does not support collations', unless: collation_enabled? do
+
+        it 'raises an exception' do
+          expect {
+            result
+          }.to raise_exception(Mongo::Error::UnsupportedCollation)
+        end
+      end
+    end
   end
 
   describe '#aggregate_spec' do
@@ -323,6 +387,41 @@ describe Mongo::Collection::View::Aggregation do
     end
   end
 
+  context 'when the aggregation has a collation defined' do
+
+    before do
+      authorized_collection.insert_many([ { name: 'bang' }, { name: 'bang' }])
+    end
+
+    let(:pipeline) do
+      [{ "$match" => { "name" => "BANG" } }]
+    end
+
+    let(:options) do
+      { collation: { locale: 'en_US', strength: 2 } }
+    end
+
+    let(:result) do
+      aggregation.collect { |doc| doc['name']}
+    end
+
+    context 'when the server selected supports collations', if: collation_enabled? do
+
+      it 'applies the collation' do
+        expect(result).to eq(['bang', 'bang'])
+      end
+    end
+
+    context 'when the server selected does not support collations', unless: collation_enabled? do
+
+      it 'raises an exception' do
+        expect {
+          result
+        }.to raise_exception(Mongo::Error::UnsupportedCollation)
+      end
+    end
+  end
+
   context 'when $out is in the pipeline', if: write_command_enabled? do
 
     let(:pipeline) do
@@ -385,6 +484,46 @@ describe Mongo::Collection::View::Aggregation do
        expect(Mongo::Logger.logger).not_to receive(:warn?)
        aggregation.to_a
      end
+
+     context 'when the view has a write concern' do
+
+       let(:collection) do
+         authorized_collection.with(write: { w: WRITE_CONCERN[:w]+1 })
+       end
+
+       let(:view) do
+         Mongo::Collection::View.new(collection, selector, view_options)
+       end
+
+       context 'when the server supports write concern on the aggregate command', if: collation_enabled? do
+
+         it 'uses the write concern' do
+           expect {
+             aggregation.to_a
+           }.to raise_exception(Mongo::Error::OperationFailure)
+         end
+       end
+
+       context 'when the server does not support write concern on the aggregation command', unless: collation_enabled? do
+
+         let(:documents) do
+           [
+             { city: "Berlin", pop: 18913, neighborhood: "Kreuzberg" },
+             { city: "Berlin", pop: 84143, neighborhood: "Mitte" },
+             { city: "New York", pop: 40270, neighborhood: "Brooklyn" }
+           ]
+         end
+
+         before do
+           authorized_collection.insert_many(documents)
+           aggregation.to_a
+         end
+
+         it 'does not apply the write concern' do
+           expect(authorized_client['output_collection'].find.count).to eq(2)
+         end
+       end
+     end
     end
   end
 end