mongo 2.0.6 → 2.1.0.beta

data/spec/mongo/collection_spec.rb

@@ -2,6 +2,10 @@ require 'spec_helper'
 
 describe Mongo::Collection do
 
+  after do
+    authorized_collection.delete_many
+  end
+
   describe '#==' do
 
     let(:database) do
@@ -282,7 +286,7 @@ describe Mongo::Collection do
       end
 
       after do
-        authorized_collection.find.delete_many
+        authorized_collection.delete_many
       end
 
       let(:view) do
@@ -295,24 +299,148 @@ describe Mongo::Collection do
         end
       end
     end
+
+    context 'when provided options' do
+
+      let(:view) do
+        authorized_collection.find({}, options)
+      end
+
+      context 'when provided :allow_partial_results' do
+
+        let(:options) do
+          { allow_partial_results: true }
+        end
+
+        it 'returns a view with :allow_partial_results set' do
+          expect(view.options[:allow_partial_results]).to be(options[:allow_partial_results])
+        end
+      end
+
+      context 'when provided :batch_size' do
+
+        let(:options) do
+          { batch_size: 100 }
+        end
+
+        it 'returns a view with :batch_size set' do
+          expect(view.options[:batch_size]).to be(options[:batch_size])
+        end
+      end
+
+      context 'when provided :comment' do
+
+        let(:options) do
+          { comment: 'slow query' }
+        end
+
+        it 'returns a view with :comment set' do
+          expect(view.options[:comment]).to be(options[:comment])
+        end
+      end
+
+      context 'when provided :cursor_type' do
+
+        let(:options) do
+          { cursor_type: :tailable }
+        end
+
+        it 'returns a view with :cursor_type set' do
+          expect(view.options[:cursor_type]).to be(options[:cursor_type])
+        end
+      end
+
+      context 'when provided :max_time_ms' do
+
+        let(:options) do
+          { max_time_ms: 500 }
+        end
+
+        it 'returns a view with :max_time_ms set' do
+          expect(view.options[:max_time_ms]).to be(options[:max_time_ms])
+        end
+      end
+
+      context 'when provided :modifiers' do
+
+        let(:options) do
+          { modifiers: { :$orderby => Mongo::Index::ASCENDING } }
+        end
+
+        it 'returns a view with modifiers set' do
+          expect(view.options[:modifiers]).to be(options[:modifiers])
+        end
+      end
+
+      context 'when provided :no_cursor_timeout' do
+
+        let(:options) do
+          { no_cursor_timeout: true }
+        end
+
+        it 'returns a view with :no_cursor_timeout set' do
+          expect(view.options[:no_cursor_timeout]).to be(options[:no_cursor_timeout])
+        end
+      end
+
+      context 'when provided :oplog_replay' do
+
+        let(:options) do
+          { oplog_replay: false }
+        end
+
+        it 'returns a view with :oplog_replay set' do
+          expect(view.options[:oplog_replay]).to be(options[:oplog_replay])
+        end
+      end
+
+      context 'when provided :projection' do
+
+        let(:options) do
+          { projection:  { 'x' => 1 } }
+        end
+
+        it 'returns a view with :projection set' do
+          expect(view.options[:projection]).to be(options[:projection])
+        end
+      end
+
+      context 'when provided :skip' do
+
+        let(:options) do
+          { skip:  5 }
+        end
+
+        it 'returns a view with :skip set' do
+          expect(view.options[:skip]).to be(options[:skip])
+        end
+      end
+
+      context 'when provided :sort' do
+
+        let(:options) do
+          { sort:  { 'x' => Mongo::Index::ASCENDING } }
+        end
+
+        it 'returns a view with :sort set' do
+          expect(view.options[:sort]).to be(options[:sort])
+        end
+      end
+    end
   end
 
   describe '#insert_many' do
 
     after do
-      authorized_collection.find.delete_many
+      authorized_collection.delete_many
     end
 
     let(:result) do
       authorized_collection.insert_many([{ name: 'test1' }, { name: 'test2' }])
     end
 
-    it 'inserts the documents into the collection', if: write_command_enabled? do
-      expect(result.written_count).to eq(2)
-    end
-
-    it 'inserts the documents into the collection', unless: write_command_enabled? do
-      expect(result.written_count).to eq(0)
+    it 'inserts the documents into the collection' do
+      expect(result.inserted_count).to eq(2)
     end
 
     it 'contains the ids in the result' do
@@ -323,7 +451,7 @@ describe Mongo::Collection do
   describe '#insert_one' do
 
     after do
-      authorized_collection.find.delete_many
+      authorized_collection.delete_many
     end
 
     let(:result) do
@@ -387,4 +515,772 @@ describe Mongo::Collection do
       end
     end
   end
+
+  describe '#aggregate' do
+
+    it 'returns an Aggregation object' do
+      expect(authorized_collection.aggregate([])).to be_a(Mongo::Collection::View::Aggregation)
+    end
+
+    context 'when options are provided' do
+
+      let(:options) do
+        { :allow_disk_use => true }
+      end
+
+      it 'sets the options on the Aggregation object' do
+        expect(authorized_collection.aggregate([], options).options).to eq(options)
+      end
+    end
+  end
+
+  describe '#count' do
+
+    let(:documents) do
+      (1..10).map{ |i| { field: "test#{i}" }}
+    end
+
+    before do
+      authorized_collection.insert_many(documents)
+    end
+
+    after do
+      authorized_collection.delete_many
+    end
+
+    it 'returns an integer count' do
+      expect(authorized_collection.count).to eq(10)
+    end
+
+    context 'when options are provided' do
+
+      it 'passes the options to the count' do
+        expect(authorized_collection.count({}, limit: 5)).to eq(5)
+      end
+    end
+  end
+
+  describe '#distinct' do
+
+    let(:documents) do
+      (1..3).map{ |i| { field: "test#{i}" }}
+    end
+
+    before do
+      authorized_collection.insert_many(documents)
+    end
+
+    after do
+      authorized_collection.delete_many
+    end
+
+    it 'returns the distinct values' do
+      expect(authorized_collection.distinct(:field)).to eq([ 'test1', 'test2', 'test3' ])
+    end
+
+    context 'when a selector is provided' do
+
+      it 'returns the distinct values' do
+        expect(authorized_collection.distinct(:field, field: 'test1')).to eq([ 'test1' ])
+      end
+    end
+
+    context 'when options are provided' do
+
+      it 'passes the options to the distinct command' do
+        expect(authorized_collection.distinct(:field, {}, max_time_ms: 100)).to eq([ 'test1', 'test2', 'test3' ])
+      end
+    end
+  end
+
+  describe '#delete_one' do
+
+    context 'when a selector was provided' do
+
+      let(:selector) do
+        { field: 'test1' }
+      end
+
+      before do
+        authorized_collection.insert_many([
+                                              { field: 'test1' },
+                                              { field: 'test1' },
+                                              { field: 'test1' }
+                                          ])
+      end
+
+      after do
+        authorized_collection.delete_many
+      end
+
+      let(:response) do
+        authorized_collection.delete_one(selector)
+      end
+
+      it 'deletes the first matching document in the collection' do
+        expect(response.deleted_count).to eq(1)
+      end
+    end
+
+    context 'when no selector was provided' do
+
+      before do
+        authorized_collection.insert_many([{ field: 'test1' }, { field: 'test2' }])
+      end
+
+      let(:response) do
+        authorized_collection.delete_one
+      end
+
+      it 'deletes the first document in the collection' do
+        expect(response.deleted_count).to eq(1)
+      end
+    end
+  end
+
+  describe '#delete_many' do
+
+    before do
+      authorized_collection.insert_many([{ field: 'test1' }, { field: 'test2' }])
+    end
+
+    after do
+      authorized_collection.delete_many
+    end
+
+    context 'when a selector was provided' do
+
+      let(:selector) do
+        { field: 'test1' }
+      end
+
+      it 'deletes the matching documents in the collection' do
+        expect(authorized_collection.delete_many(selector).deleted_count).to eq(1)
+      end
+    end
+
+    context 'when no selector was provided' do
+
+      it 'deletes all the documents in the collection' do
+        expect(authorized_collection.delete_many.deleted_count).to eq(2)
+      end
+    end
+  end
+
+  describe '#replace_one' do
+
+    let(:selector) do
+      { field: 'test1' }
+    end
+
+    context 'when a selector was provided' do
+
+      before do
+        authorized_collection.insert_many([{ field: 'test1' }, { field: 'test1' }])
+      end
+
+      after do
+        authorized_collection.delete_many
+      end
+
+      let!(:response) do
+        authorized_collection.replace_one(selector, { field: 'testing' })
+      end
+
+      let(:updated) do
+        authorized_collection.find(field: 'testing').first
+      end
+
+      it 'updates the first matching document in the collection' do
+        expect(response.modified_count).to eq(1)
+      end
+
+      it 'updates the documents in the collection' do
+        expect(updated[:field]).to eq('testing')
+      end
+    end
+
+    context 'when upsert is false' do
+
+      let!(:response) do
+        authorized_collection.replace_one(selector, { field: 'test1' }, upsert: false)
+      end
+
+      let(:updated) do
+        authorized_collection.find(field: 'test1').to_a
+      end
+
+      it 'reports that no documents were written' do
+        expect(response.modified_count).to eq(0)
+      end
+
+      it 'does not insert the document' do
+        expect(updated).to be_empty
+      end
+    end
+
+    context 'when upsert is true' do
+
+      let!(:response) do
+        authorized_collection.replace_one(selector, { field: 'test1' }, upsert: true)
+      end
+
+      let(:updated) do
+        authorized_collection.find(field: 'test1').first
+      end
+
+      after do
+        authorized_collection.delete_many
+      end
+
+      it 'reports that a document was written' do
+        expect(response.written_count).to eq(1)
+      end
+
+      it 'inserts the document' do
+        expect(updated[:field]).to eq('test1')
+      end
+    end
+
+    context 'when upsert is not specified' do
+
+      let!(:response) do
+        authorized_collection.replace_one(selector, { field: 'test1' })
+      end
+
+      let(:updated) do
+        authorized_collection.find(field: 'test1').to_a
+      end
+
+      it 'reports that no documents were written' do
+        expect(response.modified_count).to eq(0)
+      end
+
+      it 'does not insert the document' do
+        expect(updated).to be_empty
+      end
+    end
+  end
+
+  describe '#update_many' do
+
+    let(:selector) do
+      { field: 'test' }
+    end
+
+    after do
+      authorized_collection.delete_many
+    end
+
+    context 'when a selector was provided' do
+
+      before do
+        authorized_collection.insert_many([{ field: 'test' }, { field: 'test' }])
+      end
+
+      let!(:response) do
+        authorized_collection.update_many(selector, '$set'=> { field: 'testing' })
+      end
+
+      let(:updated) do
+        authorized_collection.find(field: 'testing').to_a.last
+      end
+
+      it 'returns the number updated' do
+        expect(response.modified_count).to eq(2)
+      end
+
+      it 'updates the documents in the collection' do
+        expect(updated[:field]).to eq('testing')
+      end
+    end
+
+    context 'when upsert is false' do
+
+      let(:response) do
+        authorized_collection.update_many(selector, { '$set'=> { field: 'testing' } },
+                         upsert: false)
+      end
+
+      let(:updated) do
+        authorized_collection.find.to_a
+      end
+
+      it 'reports that no documents were updated' do
+        expect(response.modified_count).to eq(0)
+      end
+
+      it 'updates no documents in the collection' do
+        expect(updated).to be_empty
+      end
+    end
+
+    context 'when upsert is true' do
+
+      let!(:response) do
+        authorized_collection.update_many(selector, { '$set'=> { field: 'testing' } },
+                         upsert: true)
+      end
+
+      let(:updated) do
+        authorized_collection.find.to_a.last
+      end
+
+      it 'reports that a document was written' do
+        expect(response.written_count).to eq(1)
+      end
+
+      it 'inserts a document into the collection' do
+        expect(updated[:field]).to eq('testing')
+      end
+    end
+
+    context 'when upsert is not specified' do
+
+      let(:response) do
+        authorized_collection.update_many(selector, { '$set'=> { field: 'testing' } })
+      end
+
+      let(:updated) do
+        authorized_collection.find.to_a
+      end
+
+      it 'reports that no documents were updated' do
+        expect(response.modified_count).to eq(0)
+      end
+
+      it 'updates no documents in the collection' do
+        expect(updated).to be_empty
+      end
+    end
+  end
+
+  describe '#update_one' do
+
+    let(:selector) do
+      { field: 'test1' }
+    end
+
+    context 'when a selector was provided' do
+
+      before do
+        authorized_collection.insert_many([{ field: 'test1' }, { field: 'test1' }])
+      end
+
+      let!(:response) do
+        authorized_collection.update_one(selector, '$set'=> { field: 'testing' })
+      end
+
+      let(:updated) do
+        authorized_collection.find(field: 'testing').first
+      end
+
+      it 'updates the first matching document in the collection' do
+        expect(response.modified_count).to eq(1)
+      end
+
+      it 'updates the documents in the collection' do
+        expect(updated[:field]).to eq('testing')
+      end
+    end
+
+    context 'when upsert is false' do
+
+      let(:response) do
+        authorized_collection.update_one(selector, { '$set'=> { field: 'testing' } },
+                        upsert: false)
+      end
+
+      let(:updated) do
+        authorized_collection.find.to_a
+      end
+
+      it 'reports that no documents were updated' do
+        expect(response.modified_count).to eq(0)
+      end
+
+      it 'updates no documents in the collection' do
+        expect(updated).to be_empty
+      end
+    end
+
+    context 'when upsert is true' do
+
+      let!(:response) do
+        authorized_collection.update_one(selector, { '$set'=> { field: 'testing' } },
+                        upsert: true)
+      end
+
+      let(:updated) do
+        authorized_collection.find.first
+      end
+
+      it 'reports that a document was written' do
+        expect(response.written_count).to eq(1)
+      end
+
+      it 'inserts a document into the collection' do
+        expect(updated[:field]).to eq('testing')
+      end
+    end
+
+    context 'when upsert is not specified' do
+
+      let(:response) do
+        authorized_collection.update_one(selector, { '$set'=> { field: 'testing' } })
+      end
+
+      let(:updated) do
+        authorized_collection.find.to_a
+      end
+
+      it 'reports that no documents were updated' do
+        expect(response.modified_count).to eq(0)
+      end
+
+      it 'updates no documents in the collection' do
+        expect(updated).to be_empty
+      end
+    end
+  end
+
+  describe '#find_one_and_delete' do
+
+    before do
+      authorized_collection.insert_many([{ field: 'test1' }])
+    end
+
+    context 'when a matching document is found' do
+
+      let(:selector) do
+        { field: 'test1' }
+      end
+
+      context 'when no options are provided' do
+
+        let!(:document) do
+          authorized_collection.find_one_and_delete(selector)
+        end
+
+        it 'deletes the document from the database' do
+          expect(authorized_collection.find.to_a).to be_empty
+        end
+
+        it 'returns the document' do
+          expect(document['field']).to eq('test1')
+        end
+      end
+
+      context 'when a projection is provided' do
+
+        let!(:document) do
+          authorized_collection.find_one_and_delete(selector, projection: { _id: 1 })
+        end
+
+        it 'deletes the document from the database' do
+          expect(authorized_collection.find.to_a).to be_empty
+        end
+
+        it 'returns the document with limited fields' do
+          expect(document['field']).to be_nil
+          expect(document['_id']).to_not be_nil
+        end
+      end
+
+      context 'when a sort is provided' do
+
+        let!(:document) do
+          authorized_collection.find_one_and_delete(selector, sort: { field: 1 })
+        end
+
+        it 'deletes the document from the database' do
+          expect(authorized_collection.find.to_a).to be_empty
+        end
+
+        it 'returns the document with limited fields' do
+          expect(document['field']).to eq('test1')
+        end
+      end
+
+      context 'when max_time_ms is provided', if: write_command_enabled? do
+
+        it 'includes the max_time_ms value in the command' do
+          expect {
+            authorized_collection.find_one_and_delete(selector, max_time_ms: 0.1)
+          }.to raise_error(Mongo::Error::OperationFailure)
+        end
+      end
+    end
+
+    context 'when no matching document is found' do
+
+      let(:selector) do
+        { field: 'test5' }
+      end
+
+      let!(:document) do
+        authorized_collection.find_one_and_delete(selector)
+      end
+
+      it 'returns nil' do
+        expect(document).to be_nil
+      end
+    end
+  end
+
+  describe '#find_one_and_update' do
+
+    before do
+      authorized_collection.insert_many([{ field: 'test1' }])
+    end
+
+    let(:selector) do
+      { field: 'test1' }
+    end
+
+    context 'when a matching document is found' do
+
+      context 'when no options are provided' do
+
+        let(:document) do
+          authorized_collection.find_one_and_update(selector, { '$set' => { field: 'testing' }})
+        end
+
+        it 'returns the original document' do
+          expect(document['field']).to eq('test1')
+        end
+      end
+
+      context 'when no options are provided' do
+
+        let(:document) do
+          authorized_collection.find_one_and_update(selector, { '$set' => { field: 'testing' }})
+        end
+
+        it 'returns the original document' do
+          expect(document['field']).to eq('test1')
+        end
+      end
+
+      context 'when return_document options are provided' do
+
+        context 'when return_document is :after' do
+
+          let(:document) do
+            authorized_collection.find_one_and_update(selector, { '$set' => { field: 'testing' }}, :return_document => :after)
+          end
+
+          it 'returns the new document' do
+            expect(document['field']).to eq('testing')
+          end
+        end
+
+        context 'when return_document is :before' do
+
+          let(:document) do
+            authorized_collection.find_one_and_update(selector, { '$set' => { field: 'testing' }}, :return_document => :before)
+          end
+
+          it 'returns the original document' do
+            expect(document['field']).to eq('test1')
+          end
+        end
+      end
+
+      context 'when a projection is provided' do
+
+        let(:document) do
+          authorized_collection.find_one_and_update(selector, { '$set' => { field: 'testing' }}, projection: { _id: 1 })
+        end
+
+        it 'returns the document with limited fields' do
+          expect(document['field']).to be_nil
+          expect(document['_id']).to_not be_nil
+        end
+      end
+
+      context 'when a sort is provided' do
+
+        let(:document) do
+          authorized_collection.find_one_and_update(selector, { '$set' => { field: 'testing' }}, sort: { field: 1 })
+        end
+
+        it 'returns the original document' do
+          expect(document['field']).to eq('test1')
+        end
+      end
+    end
+
+    context 'when max_time_ms is provided' do
+
+      it 'includes the max_time_ms value in the command', if: write_command_enabled? do
+        expect {
+          authorized_collection.find_one_and_update(selector, { '$set' => { field: 'testing' }}, max_time_ms: 0.1)
+        }.to raise_error(Mongo::Error::OperationFailure)
+      end
+    end
+
+    context 'when no matching document is found' do
+
+      let(:selector) do
+        { field: 'test5' }
+      end
+
+      let(:document) do
+        authorized_collection.find_one_and_update(selector, { '$set' => { field: 'testing' }})
+      end
+
+      it 'returns nil' do
+        expect(document).to be_nil
+      end
+    end
+
+    context 'when no matching document is found' do
+
+      context 'when no upsert options are provided' do
+
+        let(:selector) do
+          { field: 'test5' }
+        end
+
+        let(:document) do
+          authorized_collection.find_one_and_update(selector, { '$set' => { field: 'testing' }})
+        end
+
+        it 'returns nil' do
+          expect(document).to be_nil
+        end
+      end
+
+      context 'when upsert options are provided' do
+
+        let(:selector) do
+          { field: 'test5' }
+        end
+
+        let(:document) do
+          authorized_collection.find_one_and_update(selector, { '$set' => { field: 'testing' }}, :upsert => true, :return_document => :after)
+        end
+
+        it 'returns the new document' do
+          expect(document['field']).to eq('testing')
+        end
+      end
+    end
+  end
+
+  describe '#find_one_and_replace' do
+
+    before do
+      authorized_collection.insert_many([{ field: 'test1', other: 'sth' }])
+    end
+
+    let(:selector) do
+      { field: 'test1' }
+    end
+
+    context 'when a matching document is found' do
+
+      context 'when no options are provided' do
+
+        let(:document) do
+          authorized_collection.find_one_and_replace(selector, { field: 'testing' })
+        end
+
+        it 'returns the original document' do
+          expect(document['field']).to eq('test1')
+        end
+      end
+
+      context 'when return_document options are provided' do
+
+        context 'when return_document is :after' do
+
+          let(:document) do
+            authorized_collection.find_one_and_replace(selector, { field: 'testing' }, :return_document => :after)
+          end
+
+          it 'returns the new document' do
+            expect(document['field']).to eq('testing')
+          end
+        end
+
+        context 'when return_document is :before' do
+
+          let(:document) do
+            authorized_collection.find_one_and_replace(selector, { field: 'testing' }, :return_document => :before)
+          end
+
+          it 'returns the original document' do
+            expect(document['field']).to eq('test1')
+          end
+        end
+      end
+
+      context 'when a projection is provided' do
+
+        let(:document) do
+          authorized_collection.find_one_and_replace(selector, { field: 'testing' }, projection: { _id: 1 })
+        end
+
+        it 'returns the document with limited fields' do
+          expect(document['field']).to be_nil
+          expect(document['_id']).to_not be_nil
+        end
+      end
+
+      context 'when a sort is provided' do
+
+        let(:document) do
+          authorized_collection.find_one_and_replace(selector, { field: 'testing' }, :sort => { field: 1 })
+        end
+
+        it 'returns the original document' do
+          expect(document['field']).to eq('test1')
+        end
+      end
+    end
+
+    context 'when no matching document is found' do
+
+      context 'when no upsert options are provided' do
+
+        let(:selector) do
+          { field: 'test5' }
+        end
+
+        let(:document) do
+          authorized_collection.find_one_and_replace(selector, { field: 'testing' })
+        end
+
+        it 'returns nil' do
+          expect(document).to be_nil
+        end
+      end
+
+      context 'when upsert options are provided' do
+
+        let(:selector) do
+          { field: 'test5' }
+        end
+
+        let(:document) do
+          authorized_collection.find_one_and_replace(selector, { field: 'testing' }, :upsert => true, :return_document => :after)
+        end
+
+        it 'returns the new document' do
+          expect(document['field']).to eq('testing')
+        end
+      end
+    end
+
+    context 'when max_time_ms is provided', if: write_command_enabled? do
+
+      it 'includes the max_time_ms value in the command' do
+        expect {
+          authorized_collection.find_one_and_replace(selector, { field: 'testing' }, max_time_ms: 0.1)
+        }.to raise_error(Mongo::Error::OperationFailure)
+      end
+    end
+  end
 end