mongo 1.12.5 → 2.0.0.beta

data/spec/mongo/collection_spec.rb

@@ -0,0 +1,362 @@
+require 'spec_helper'
+
+describe Mongo::Collection do
+
+  describe '#==' do
+
+    let(:database) do
+      Mongo::Database.new(authorized_client, :test)
+    end
+
+    let(:collection) do
+      described_class.new(database, :users)
+    end
+
+    context 'when the names are the same' do
+
+      context 'when the databases are the same' do
+
+        let(:other) do
+          described_class.new(database, :users)
+        end
+
+        it 'returns true' do
+          expect(collection).to eq(other)
+        end
+      end
+
+      context 'when the databases are not the same' do
+
+        let(:other_db) do
+          Mongo::Database.new(authorized_client, :testing)
+        end
+
+        let(:other) do
+          described_class.new(other_db, :users)
+        end
+
+        it 'returns false' do
+          expect(collection).to_not eq(other)
+        end
+      end
+
+      context 'when the options are the same' do
+
+        let(:other) do
+          described_class.new(database, :users)
+        end
+
+        it 'returns true' do
+          expect(collection).to eq(other)
+        end
+      end
+
+      context 'when the options are not the same' do
+
+        let(:other) do
+          described_class.new(database, :users, :capped => true)
+        end
+
+        it 'returns false' do
+          expect(collection).to_not eq(other)
+        end
+      end
+    end
+
+    context 'when the names are not the same' do
+
+      let(:other) do
+        described_class.new(database, :sounds)
+      end
+
+      it 'returns false' do
+        expect(collection).to_not eq(other)
+      end
+    end
+
+    context 'when the object is not a collection' do
+
+      it 'returns false' do
+        expect(collection).to_not eq('test')
+      end
+    end
+  end
+
+  describe '#capped?' do
+
+    let(:database) do
+      authorized_client.database
+    end
+
+    context 'when the collection is capped' do
+
+      let(:collection) do
+        described_class.new(database, :specs, :capped => true, :size => 1024)
+      end
+
+      before do
+        collection.create
+      end
+
+      after do
+        collection.drop
+      end
+
+      it 'returns true' do
+        expect(collection).to be_capped
+      end
+    end
+
+    context 'when the collection is not capped' do
+
+      let(:collection) do
+        described_class.new(database, :specs)
+      end
+
+      before do
+        collection.create
+      end
+
+      after do
+        collection.drop
+      end
+
+      it 'returns false' do
+        expect(collection).to_not be_capped
+      end
+    end
+  end
+
+  describe '#create' do
+
+    let(:database) do
+      authorized_client.database
+    end
+
+    context 'when the collection has no options' do
+
+      let(:collection) do
+        described_class.new(database, :specs)
+      end
+
+      let!(:response) do
+        collection.create
+      end
+
+      after do
+        collection.drop
+      end
+
+      it 'executes the command' do
+        expect(response).to be_successful
+      end
+
+      it 'creates the collection in the database' do
+        expect(database.collection_names).to include('specs')
+      end
+    end
+
+    context 'when the collection has options' do
+
+      context 'when the collection is capped' do
+
+        shared_examples 'a capped collection command' do
+
+          let!(:response) do
+            collection.create
+          end
+
+          after do
+            collection.drop
+          end
+
+          it 'executes the command' do
+            expect(response).to be_successful
+          end
+
+          it 'sets the collection as capped' do
+            expect(collection).to be_capped
+          end
+
+          it 'creates the collection in the database' do
+            expect(database.collection_names).to include('specs')
+          end
+        end
+
+        context 'when instantiating a collection directly' do
+
+          let(:collection) do
+            described_class.new(database, :specs, :capped => true, :size => 1024)
+          end
+
+          it_behaves_like 'a capped collection command'
+        end
+
+        context 'when instantiating a collection through the database' do
+
+          let(:collection) do
+            authorized_client[:specs, :capped => true, :size => 1024]
+          end
+
+          it_behaves_like 'a capped collection command'
+        end
+      end
+    end
+  end
+
+  describe '#drop' do
+
+    let(:database) do
+      authorized_client.database
+    end
+
+    let(:collection) do
+      described_class.new(database, :specs)
+    end
+
+    before do
+      collection.create
+    end
+
+    let!(:response) do
+      collection.drop
+    end
+
+    it 'executes the command' do
+      expect(response).to be_successful
+    end
+
+    it 'drops the collection from the database' do
+      expect(database.collection_names).to_not include('specs')
+    end
+  end
+
+  describe '#find' do
+
+    context 'when provided a selector' do
+
+      let(:view) do
+        authorized_collection.find(name: 1)
+      end
+
+      it 'returns a authorized_collection view for the selector' do
+        expect(view.selector).to eq(name: 1)
+      end
+    end
+
+    context 'when provided no selector' do
+
+      let(:view) do
+        authorized_collection.find
+      end
+
+      it 'returns a authorized_collection view with an empty selector' do
+        expect(view.selector).to be_empty
+      end
+    end
+
+    context 'when iterating the authorized_collection view' do
+
+      before do
+        authorized_collection.insert_many([{ field: 'test1' }, { field: 'test2' }])
+      end
+
+      after do
+        authorized_collection.find.remove_many
+      end
+
+      let(:view) do
+        authorized_collection.find
+      end
+
+      it 'iterates over the documents' do
+        view.each do |document|
+          expect(document).to_not be_nil
+        end
+      end
+    end
+  end
+
+  describe '#insert_many' do
+
+    after do
+      authorized_collection.find.remove_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)
+    end
+  end
+
+  describe '#insert_one' do
+
+    after do
+      authorized_collection.find.remove_many
+    end
+
+    let(:result) do
+      authorized_collection.insert_one({ name: 'testing' })
+    end
+
+    it 'inserts the document into the collection', if: write_command_enabled? do
+      expect(result.written_count).to eq(1)
+    end
+
+    it 'inserts the document into the collection', unless: write_command_enabled? do
+      expect(result.written_count).to eq(0)
+    end
+  end
+
+  describe '#inspect' do
+
+    it 'includes the object id' do
+      expect(authorized_collection.inspect).to include(authorized_collection.object_id.to_s)
+    end
+
+    it 'includes the namespace' do
+      expect(authorized_collection.inspect).to include(authorized_collection.namespace)
+    end
+  end
+
+  describe '#indexes' do
+
+    let(:index_spec) do
+      { name: 1 }
+    end
+
+    let(:batch_size) { nil }
+
+    let(:index_names) do
+      authorized_collection.indexes(batch_size: batch_size).collect { |i| i['name'] }
+    end
+
+    before do
+      authorized_collection.indexes.create(index_spec, unique: true)
+    end
+
+    after do
+      authorized_collection.indexes.drop(index_spec)
+    end
+
+    it 'returns a list of indexes' do
+      expect(index_names).to include(*'name_1', '_id_')
+    end
+
+    context 'when batch size is specified' do
+
+      let(:batch_size) { 1 }
+
+      it 'returns a list of indexes' do
+        expect(index_names).to include(*'name_1', '_id_')
+      end
+    end
+  end
+end

data/spec/mongo/cursor_spec.rb

@@ -0,0 +1,295 @@
+require 'spec_helper'
+
+describe Mongo::Cursor do
+
+  describe '#each' do
+
+    let(:reply) do
+      Mongo::Operation::Read::Query.new(query_spec).execute(authorized_primary.context)
+    end
+
+    let(:cursor) do
+      described_class.new(view, reply, authorized_primary)
+    end
+
+    context 'when no options are provided to the view' do
+
+      let(:view) do
+        Mongo::Collection::View.new(authorized_collection)
+      end
+
+      let(:query_spec) do
+        { :selector => {}, :options => {}, :db_name => TEST_DB, :coll_name => TEST_COLL }
+      end
+
+      context 'when the initial query retieves all documents' do
+
+        let(:documents) do
+          (1..10).map{ |i| { field: "test#{i}" }}
+        end
+
+        before do
+          authorized_collection.insert_many(documents)
+        end
+
+        after do
+          authorized_collection.find.remove_many
+        end
+
+        it 'returns the correct amount' do
+          expect(cursor.to_a.count).to eq(10)
+        end
+
+        it 'iterates the documents' do
+          cursor.each do |doc|
+            expect(doc).to have_key('field')
+          end
+        end
+      end
+
+      context 'when the initial query does not retrieve all documents' do
+
+        let(:documents) do
+          (1..102).map{ |i| { field: "test#{i}" }}
+        end
+
+        before do
+          authorized_collection.insert_many(documents)
+        end
+
+        after do
+          authorized_collection.find.remove_many
+        end
+
+        it 'returns the correct amount' do
+          expect(cursor.to_a.count).to eq(102)
+        end
+
+        it 'iterates the documents' do
+          cursor.each do |doc|
+            expect(doc).to have_key('field')
+          end
+        end
+      end
+    end
+
+    context 'when options are provided to the view' do
+
+      let(:documents) do
+        (1..10).map{ |i| { field: "test#{i}" }}
+      end
+
+      before do
+        authorized_collection.insert_many(documents)
+      end
+
+      after do
+        authorized_collection.find.remove_many
+      end
+
+      context 'when a limit is provided' do
+
+        context 'when no batch size is provided' do
+
+          context 'when the limit is positive' do
+
+            let(:view) do
+              Mongo::Collection::View.new(authorized_collection, {}, :limit => 2)
+            end
+
+            let(:query_spec) do
+              {
+                :selector => {},
+                :options => { :limit => 2 },
+                :db_name => TEST_DB,
+                :coll_name => TEST_COLL
+              }
+            end
+
+            it 'returns the correct amount' do
+              expect(cursor.to_a.count).to eq(2)
+            end
+
+            it 'iterates the documents' do
+              cursor.each do |doc|
+                expect(doc).to have_key('field')
+              end
+            end
+          end
+
+          context 'when the limit is negative' do
+
+            let(:view) do
+              Mongo::Collection::View.new(authorized_collection, {}, :limit => -2)
+            end
+
+            let(:query_spec) do
+              {
+                :selector => {},
+                :options => { :limit => -2 },
+                :db_name => TEST_DB,
+                :coll_name => TEST_COLL
+              }
+            end
+
+            it 'returns the positive number of documents' do
+              expect(cursor.to_a.count).to eq(2)
+            end
+
+            it 'iterates the documents' do
+              cursor.each do |doc|
+                expect(doc).to have_key('field')
+              end
+            end
+          end
+
+          context 'when the limit is zero' do
+
+            let(:view) do
+              Mongo::Collection::View.new(authorized_collection, {}, :limit => 0)
+            end
+
+            let(:query_spec) do
+              {
+                :selector => {},
+                :options => { :limit => 0 },
+                :db_name => TEST_DB,
+                :coll_name => TEST_COLL
+              }
+            end
+
+            it 'returns all documents' do
+              expect(cursor.to_a.count).to eq(10)
+            end
+
+            it 'iterates the documents' do
+              cursor.each do |doc|
+                expect(doc).to have_key('field')
+              end
+            end
+          end
+        end
+
+        context 'when a batch size is provided' do
+
+          context 'when the batch size is less than the limit' do
+
+            let(:view) do
+              Mongo::Collection::View.new(
+                authorized_collection,
+                {},
+                :limit => 5, :batch_size => 3
+              )
+            end
+
+            let(:query_spec) do
+              {
+                :selector => {},
+                :options => { :limit => 5, :batch_size => 3 },
+                :db_name => TEST_DB,
+                :coll_name => TEST_COLL
+              }
+            end
+
+            it 'returns the limited number of documents' do
+              expect(cursor.to_a.count).to eq(5)
+            end
+
+            it 'iterates the documents' do
+              cursor.each do |doc|
+                expect(doc).to have_key('field')
+              end
+            end
+          end
+
+          context 'when the batch size is more than the limit' do
+
+            let(:view) do
+              Mongo::Collection::View.new(
+                authorized_collection,
+                {},
+                :limit => 5, :batch_size => 7
+              )
+            end
+
+            let(:query_spec) do
+              {
+                :selector => {},
+                :options => { :limit => 5, :batch_size => 7 },
+                :db_name => TEST_DB,
+                :coll_name => TEST_COLL
+              }
+            end
+
+            it 'returns the limited number of documents' do
+              expect(cursor.to_a.count).to eq(5)
+            end
+
+            it 'iterates the documents' do
+              cursor.each do |doc|
+                expect(doc).to have_key('field')
+              end
+            end
+          end
+
+          context 'when the batch size is the same as the limit' do
+
+            let(:view) do
+              Mongo::Collection::View.new(
+                authorized_collection,
+                {},
+                :limit => 5, :batch_size => 5
+              )
+            end
+
+            let(:query_spec) do
+              {
+                :selector => {},
+                :options => { :limit => 5, :batch_size => 5 },
+                :db_name => TEST_DB,
+                :coll_name => TEST_COLL
+              }
+            end
+
+            it 'returns the limited number of documents' do
+              expect(cursor.to_a.count).to eq(5)
+            end
+
+            it 'iterates the documents' do
+              cursor.each do |doc|
+                expect(doc).to have_key('field')
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+
+  describe '#inspect' do
+
+    let(:view) do
+      Mongo::Collection::View.new(authorized_collection)
+    end
+
+    let(:query_spec) do
+      { :selector => {}, :options => {}, :db_name => TEST_DB, :coll_name => TEST_COLL }
+    end
+
+    let(:reply) do
+      Mongo::Operation::Read::Query.new(query_spec)
+    end
+
+    let(:cursor) do
+      described_class.new(view, reply, authorized_primary)
+    end
+
+    it 'returns a string' do
+      expect(cursor.inspect).to be_a(String)
+    end
+
+    it 'returns a string containing the collection view inspect string' do
+      expect(cursor.inspect).to match(/.*#{view.inspect}.*/)
+    end
+  end
+end