mongo 2.1.0.beta → 2.1.0.rc0

data/spec/mongo/collection_spec.rb

@@ -6,6 +6,10 @@ describe Mongo::Collection do
     authorized_collection.delete_many
   end
 
+  let(:collection_invalid_write_concern) do
+    authorized_collection.client.with(write: { w: (WRITE_CONCERN[:w] + 1) })[authorized_collection.name]
+  end
+
   describe '#==' do
 
     let(:database) do
@@ -86,6 +90,172 @@ describe Mongo::Collection do
     end
   end
 
+  describe '#with' do
+
+    let(:client) do
+      Mongo::Client.new(ADDRESSES)
+    end
+
+    let(:database) do
+      Mongo::Database.new(client, :test)
+    end
+
+    let(:collection) do
+      database.collection(:users)
+    end
+
+    let(:new_collection) do
+      collection.with(new_options)
+    end
+
+    context 'when new read options are provided' do
+
+      let(:new_options) do
+        { read: { mode: :secondary } }
+      end
+
+      it 'returns a new collection' do
+        expect(new_collection).not_to be(collection)
+      end
+
+      it 'sets the new read options on the new collection' do
+        expect(new_collection.read_preference).to eq(Mongo::ServerSelector.get(new_options[:read]))
+      end
+
+      context 'when the client has a server selection timeout setting' do
+
+        let(:client) do
+          Mongo::Client.new(ADDRESSES, server_selection_timeout: 2)
+        end
+
+        let(:server_selection_timeout) do
+          new_collection.read_preference.server_selection_timeout
+        end
+
+        it 'keeps the server_selection_timeout setting from client' do
+          expect(server_selection_timeout).to eq(client.options[:server_selection_timeout])
+        end
+      end
+
+      context 'when the client has a read preference set' do
+
+        let(:client) do
+          Mongo::Client.new(ADDRESSES, read: { mode: :primary_preferred })
+        end
+
+        it 'sets the new read options on the new collection' do
+          expect(new_collection.read_preference).to eq(Mongo::ServerSelector.get(new_options[:read]))
+        end
+      end
+
+      context 'when the client has a read preference and server selection timeout set' do
+
+        let(:client) do
+          Mongo::Client.new(ADDRESSES, read: { mode: :primary_preferred }, server_selection_timeout: 2)
+        end
+
+        let(:server_selection_timeout) do
+          new_collection.read_preference.server_selection_timeout
+        end
+
+        it 'sets the new read options on the new collection' do
+          expect(new_collection.read_preference).to eq(Mongo::ServerSelector.get(new_options[:read]))
+        end
+
+        it 'keeps the server_selection_timeout setting from client' do
+          expect(server_selection_timeout).to eq(client.options[:server_selection_timeout])
+        end
+      end
+    end
+
+    context 'when new write options are provided' do
+
+      let(:new_options) do
+        { write: { w: 5 } }
+      end
+
+      it 'returns a new collection' do
+        expect(new_collection).not_to be(collection)
+      end
+
+      it 'sets the new write options on the new collection' do
+        expect(new_collection.write_concern.options).to eq(Mongo::WriteConcern.get(new_options[:write]).options)
+      end
+
+      context 'when the client has a write concern set' do
+
+        let(:client) do
+          Mongo::Client.new(ADDRESSES, write: { w: 10 })
+        end
+
+        it 'sets the new write options on the new collection' do
+          expect(new_collection.write_concern.options).to eq(Mongo::WriteConcern.get(new_options[:write]).options)
+        end
+      end
+    end
+
+    context 'when new read and write options are provided' do
+
+      let(:new_options) do
+        {
+          read: { mode: :secondary },
+          write: { w: 4}
+        }
+      end
+
+      it 'returns a new collection' do
+        expect(new_collection).not_to be(collection)
+      end
+
+      it 'sets the new read options on the new collection' do
+        expect(new_collection.read_preference).to eq(Mongo::ServerSelector.get(new_options[:read]))
+      end
+
+      it 'sets the new write options on the new collection' do
+        expect(new_collection.write_concern.options).to eq(Mongo::WriteConcern.get(new_options[:write]).options)
+      end
+
+      context 'when the client has a server selection timeout setting' do
+
+        let(:client) do
+          Mongo::Client.new(ADDRESSES, server_selection_timeout: 2)
+        end
+
+        let(:server_selection_timeout) do
+          new_collection.read_preference.server_selection_timeout
+        end
+
+        it 'keeps the server_selection_timeout setting from client' do
+          expect(server_selection_timeout).to eq(client.options[:server_selection_timeout])
+        end
+      end
+
+      context 'when the client has a read preference set' do
+
+        let(:client) do
+          Mongo::Client.new(ADDRESSES, read: { mode: :primary_preferred })
+        end
+
+        it 'sets the new read options on the new collection' do
+          expect(new_collection.read_preference).to eq(Mongo::ServerSelector.get(new_options[:read]))
+        end
+      end
+    end
+
+    context 'when neither read nor write options are provided' do
+
+      let(:new_options) do
+        { some_option: 'invalid' }
+      end
+
+      it 'raises an error' do
+        expect {
+          new_collection
+        }.to raise_exception(Mongo::Error::UnchangeableCollectionOption)
+      end
+    end
+  end
+
   describe '#capped?' do
 
     let(:database) do
@@ -324,7 +494,7 @@ describe Mongo::Collection do
         end
 
         it 'returns a view with :batch_size set' do
-          expect(view.options[:batch_size]).to be(options[:batch_size])
+          expect(view.options[:batch_size]).to eq(options[:batch_size])
         end
       end
 
@@ -335,7 +505,7 @@ describe Mongo::Collection do
         end
 
         it 'returns a view with :comment set' do
-          expect(view.options[:comment]).to be(options[:comment])
+          expect(view.modifiers[:$comment]).to eq(options[:comment])
         end
       end
 
@@ -346,10 +516,12 @@ describe Mongo::Collection do
         end
 
         it 'returns a view with :cursor_type set' do
-          expect(view.options[:cursor_type]).to be(options[:cursor_type])
+          expect(view.options[:cursor_type]).to eq(options[:cursor_type])
         end
       end
 
+      #limit
+
       context 'when provided :max_time_ms' do
 
         let(:options) do
@@ -357,7 +529,7 @@ describe Mongo::Collection do
         end
 
         it 'returns a view with :max_time_ms set' do
-          expect(view.options[:max_time_ms]).to be(options[:max_time_ms])
+          expect(view.modifiers[:$maxTimeMS]).to eq(options[:max_time_ms])
         end
       end
 
@@ -368,7 +540,11 @@ describe Mongo::Collection do
         end
 
         it 'returns a view with modifiers set' do
-          expect(view.options[:modifiers]).to be(options[:modifiers])
+          expect(view.modifiers).to eq(options[:modifiers])
+        end
+
+        it 'dups the modifiers hash' do
+          expect(view.modifiers).not_to be(options[:modifiers])
         end
       end
 
@@ -379,7 +555,7 @@ describe Mongo::Collection do
         end
 
         it 'returns a view with :no_cursor_timeout set' do
-          expect(view.options[:no_cursor_timeout]).to be(options[:no_cursor_timeout])
+          expect(view.options[:no_cursor_timeout]).to eq(options[:no_cursor_timeout])
         end
       end
 
@@ -390,7 +566,7 @@ describe Mongo::Collection do
         end
 
         it 'returns a view with :oplog_replay set' do
-          expect(view.options[:oplog_replay]).to be(options[:oplog_replay])
+          expect(view.options[:oplog_replay]).to eq(options[:oplog_replay])
         end
       end
 
@@ -401,7 +577,7 @@ describe Mongo::Collection do
         end
 
         it 'returns a view with :projection set' do
-          expect(view.options[:projection]).to be(options[:projection])
+          expect(view.options[:projection]).to eq(options[:projection])
         end
       end
 
@@ -412,7 +588,7 @@ describe Mongo::Collection do
         end
 
         it 'returns a view with :skip set' do
-          expect(view.options[:skip]).to be(options[:skip])
+          expect(view.options[:skip]).to eq(options[:skip])
         end
       end
 
@@ -423,7 +599,7 @@ describe Mongo::Collection do
         end
 
         it 'returns a view with :sort set' do
-          expect(view.options[:sort]).to be(options[:sort])
+          expect(view.modifiers[:$orderby]).to eq(options[:sort])
         end
       end
     end
@@ -446,6 +622,19 @@ describe Mongo::Collection do
     it 'contains the ids in the result' do
       expect(result.inserted_ids.size).to eq(2)
     end
+
+    context 'when the inserts fail' do
+
+      let(:result) do
+        authorized_collection.insert_many([{ _id: 1 }, { _id: 1 }])
+      end
+
+      it 'raises an BulkWriteError' do
+        expect {
+          result
+        }.to raise_exception(Mongo::Error::BulkWriteError)
+      end
+    end
   end
 
   describe '#insert_one' do
@@ -469,6 +658,20 @@ describe Mongo::Collection do
     it 'contains the id in the result' do
       expect(result.inserted_id).to_not be_nil
     end
+
+    context 'when the insert fails' do
+
+      let(:result) do
+        authorized_collection.insert_one(_id: 1)
+        authorized_collection.insert_one(_id: 1)
+      end
+
+      it 'raises an OperationFailure' do
+        expect {
+          result
+        }.to raise_exception(Mongo::Error::OperationFailure)
+      end
+    end
   end
 
   describe '#inspect' do
@@ -636,6 +839,19 @@ describe Mongo::Collection do
         expect(response.deleted_count).to eq(1)
       end
     end
+
+    context 'when the delete fails', if: standalone? do
+
+      let(:result) do
+        collection_invalid_write_concern.delete_one
+      end
+
+      it 'raises an OperationFailure' do
+        expect {
+          result
+        }.to raise_exception(Mongo::Error::OperationFailure)
+      end
+    end
   end
 
   describe '#delete_many' do
@@ -665,6 +881,54 @@ describe Mongo::Collection do
         expect(authorized_collection.delete_many.deleted_count).to eq(2)
       end
     end
+
+    context 'when the deletes fail', if: standalone?  do
+
+      let(:result) do
+        collection_invalid_write_concern.delete_many
+      end
+
+      it 'raises an OperationFailure' do
+        expect {
+          result
+        }.to raise_exception(Mongo::Error::OperationFailure)
+      end
+    end
+  end
+
+  describe '#parallel_scan', unless: sharded? do
+
+    let(:documents) do
+      (1..200).map do |i|
+        { name: "testing-scan-#{i}" }
+      end
+    end
+
+    before do
+      authorized_collection.insert_many(documents)
+    end
+
+    let(:cursors) do
+      authorized_collection.parallel_scan(2)
+    end
+
+    it 'returns an array of cursors', if: write_command_enabled? do
+      cursors.each do |cursor|
+        expect(cursor.class).to be(Mongo::Cursor)
+      end
+    end
+
+    it 'returns the correct number of documents', if: write_command_enabled? do
+      expect(
+        cursors.reduce(0) { |total, cursor| total + cursor.to_a.size }
+      ).to eq(200)
+    end
+
+    it 'raises an error', unless: write_command_enabled? do
+      expect {
+        cursors
+      }.to raise_error(Mongo::Error::OperationFailure)
+    end
   end
 
   describe '#replace_one' do
@@ -760,6 +1024,19 @@ describe Mongo::Collection do
         expect(updated).to be_empty
       end
     end
+
+    context 'when the replace fails' do
+
+      let(:result) do
+        authorized_collection.replace_one(selector, { '$s' => 'test1' })
+      end
+
+      it 'raises an OperationFailure' do
+        expect {
+          result
+        }.to raise_exception(Mongo::Error::OperationFailure)
+      end
+    end
   end
 
   describe '#update_many' do
@@ -853,6 +1130,19 @@ describe Mongo::Collection do
         expect(updated).to be_empty
       end
     end
+
+    context 'when the updates fail' do
+
+      let(:result) do
+        authorized_collection.update_many(selector, { '$s'=> { field: 'testing' } })
+      end
+
+      it 'raises an OperationFailure' do
+        expect {
+          result
+        }.to raise_exception(Mongo::Error::OperationFailure)
+      end
+    end
   end
 
   describe '#update_one' do
@@ -942,6 +1232,19 @@ describe Mongo::Collection do
         expect(updated).to be_empty
       end
     end
+
+    context 'when the update fails' do
+
+      let(:result) do
+        authorized_collection.update_one(selector, { '$s'=> { field: 'testing' } })
+      end
+
+      it 'raises an OperationFailure' do
+        expect {
+          result
+        }.to raise_exception(Mongo::Error::OperationFailure)
+      end
+    end
   end
 
   describe '#find_one_and_delete' do
@@ -950,11 +1253,11 @@ describe Mongo::Collection do
       authorized_collection.insert_many([{ field: 'test1' }])
     end
 
-    context 'when a matching document is found' do
+    let(:selector) do
+      { field: 'test1' }
+    end
 
-      let(:selector) do
-        { field: 'test1' }
-      end
+    context 'when a matching document is found' do
 
       context 'when no options are provided' do
 
@@ -1026,18 +1329,31 @@ describe Mongo::Collection do
         expect(document).to be_nil
       end
     end
-  end
 
-  describe '#find_one_and_update' do
+    context 'when the operation fails', if: write_command_enabled? do
 
-    before do
-      authorized_collection.insert_many([{ field: 'test1' }])
+      let(:result) do
+        authorized_collection.find_one_and_delete(selector, max_time_ms: 0.1)
+      end
+
+      it 'raises an OperationFailure' do
+        expect {
+          result
+        }.to raise_exception(Mongo::Error::OperationFailure)
+      end
     end
+  end
+
+  describe '#find_one_and_update' do
 
     let(:selector) do
       { field: 'test1' }
     end
 
+    before do
+      authorized_collection.insert_many([{ field: 'test1' }])
+    end
+
     context 'when a matching document is found' do
 
       context 'when no options are provided' do
@@ -1167,6 +1483,19 @@ describe Mongo::Collection do
         end
       end
     end
+
+    context 'when the operation fails', if: write_command_enabled? do
+
+      let(:result) do
+        authorized_collection.find_one_and_update(selector, { '$set' => { field: 'testing' }}, max_time_ms: 0.1)
+      end
+
+      it 'raises an OperationFailure' do
+        expect {
+          result
+        }.to raise_exception(Mongo::Error::OperationFailure)
+      end
+    end
   end
 
   describe '#find_one_and_replace' do
@@ -1282,5 +1611,18 @@ describe Mongo::Collection do
         }.to raise_error(Mongo::Error::OperationFailure)
       end
     end
+
+    context 'when the operation fails', if: write_command_enabled? do
+
+      let(:result) do
+        authorized_collection.find_one_and_replace(selector, { field: 'testing' }, max_time_ms: 0.1)
+      end
+
+      it 'raises an OperationFailure' do
+        expect {
+          result
+        }.to raise_exception(Mongo::Error::OperationFailure)
+      end
+    end
   end
 end