mongo 2.6.4 → 2.7.0.rc0

data/spec/mongo/cursor/builder/get_more_command_spec.rb

@@ -50,7 +50,7 @@ describe Mongo::Cursor::Builder::GetMoreCommand do
     shared_examples_for 'a getMore command builder' do
 
       it 'includes the database name' do
-        expect(specification[:db_name]).to eq(TEST_DB)
+        expect(specification[:db_name]).to eq(SpecConfig.instance.test_db)
       end
 
       it 'includes getMore with cursor id' do

data/spec/mongo/cursor/builder/op_get_more_spec.rb

@@ -42,7 +42,7 @@ describe Mongo::Cursor::Builder::OpGetMore do
     end
 
     it 'includes the database name' do
-      expect(specification[:db_name]).to eq(TEST_DB)
+      expect(specification[:db_name]).to eq(SpecConfig.instance.test_db)
     end
 
     it 'includes the collection name' do

data/spec/mongo/cursor_spec.rb

@@ -1,6 +1,54 @@
 require 'spec_helper'
 
 describe Mongo::Cursor do
+  let(:authorized_collection) do
+    authorized_client['cursor_spec_collection']
+  end
+
+  before do
+    authorized_collection.drop
+  end
+
+  describe '#initialize' do
+    let(:server) do
+      view.send(:server_selector).select_server(authorized_client.cluster)
+    end
+
+    let(:reply) do
+      view.send(:send_initial_query, server)
+    end
+
+    let(:cursor) do
+      described_class.new(view, reply, server)
+    end
+
+    before do
+      documents = [{test: 1}] * 10
+      authorized_collection.insert_many(documents)
+    end
+
+    context 'cursor exhausted by initial result' do
+      let(:view) do
+        Mongo::Collection::View.new(authorized_collection)
+      end
+
+      it 'does not schedule the finalizer' do
+        expect(ObjectSpace).not_to receive(:define_finalizer)
+        cursor
+      end
+    end
+
+    context 'cursor not exhausted by initial result' do
+      let(:view) do
+        Mongo::Collection::View.new(authorized_collection, {}, batch_size: 2)
+      end
+
+      it 'schedules the finalizer' do
+        expect(ObjectSpace).to receive(:define_finalizer)
+        cursor
+      end
+    end
+  end
 
   describe '#each' do
 
@@ -32,10 +80,6 @@ describe Mongo::Cursor do
           authorized_collection.insert_many(documents)
         end
 
-        after do
-          authorized_collection.delete_many
-        end
-
         it 'returns the correct amount' do
           expect(cursor.to_a.count).to eq(10)
         end
@@ -57,10 +101,6 @@ describe Mongo::Cursor do
           authorized_collection.insert_many(documents)
         end
 
-        after do
-          authorized_collection.delete_many
-        end
-
         context 'when a getMore gets a socket error' do
 
           let(:op) do
@@ -102,13 +142,10 @@ describe Mongo::Cursor do
       end
 
       before do
+        authorized_collection.drop
         authorized_collection.insert_many(documents)
       end
 
-      after do
-        authorized_collection.delete_many
-      end
-
       context 'when a limit is provided' do
 
         context 'when no batch size is provided' do
@@ -250,10 +287,6 @@ describe Mongo::Cursor do
                                      cursor.instance_variable_get(:@server))
       end
 
-      after do
-        authorized_collection.delete_many
-      end
-
       let(:view) do
         Mongo::Collection::View.new(
             authorized_collection,
@@ -290,11 +323,8 @@ describe Mongo::Cursor do
       end
 
       before do
-        authorized_collection.insert_many(documents)
-      end
-
-      after do
         authorized_collection.delete_many
+        authorized_collection.insert_many(documents)
       end
 
       let(:view) do
@@ -336,10 +366,6 @@ describe Mongo::Cursor do
       collection.insert_many(documents)
     end
 
-    after do
-      collection.delete_many
-    end
-
     let(:cursor) do
       view.instance_variable_get(:@cursor)
     end
@@ -412,7 +438,7 @@ describe Mongo::Cursor do
     end
 
     let(:query_spec) do
-      { :selector => {}, :options => {}, :db_name => TEST_DB, :coll_name => TEST_COLL }
+      { :selector => {}, :options => {}, :db_name => SpecConfig.instance.test_db, :coll_name => TEST_COLL }
     end
 
     let(:reply) do

data/spec/mongo/database_spec.rb

@@ -5,13 +5,13 @@ describe Mongo::Database do
   describe '#==' do
 
     let(:database) do
-      described_class.new(authorized_client, TEST_DB)
+      described_class.new(authorized_client, SpecConfig.instance.test_db)
     end
 
     context 'when the names are the same' do
 
       let(:other) do
-        described_class.new(authorized_client, TEST_DB)
+        described_class.new(authorized_client, SpecConfig.instance.test_db)
       end
 
       it 'returns true' do
@@ -41,7 +41,7 @@ describe Mongo::Database do
   describe '#[]' do
 
     let(:database) do
-      described_class.new(authorized_client, TEST_DB)
+      described_class.new(authorized_client, SpecConfig.instance.test_db)
     end
 
     context 'when providing a valid name' do
@@ -67,7 +67,7 @@ describe Mongo::Database do
     context 'when the client has options' do
 
       let(:client) do
-        Mongo::Client.new([default_address.host], TEST_OPTIONS.merge(read: { mode: :secondary }))
+        new_local_client([default_address.host], SpecConfig.instance.test_options.merge(read: { mode: :secondary }))
       end
 
       let(:database) do
@@ -88,15 +88,12 @@ describe Mongo::Database do
   describe '#collection_names' do
 
     let(:database) do
-      described_class.new(authorized_client, TEST_DB)
+      described_class.new(authorized_client, SpecConfig.instance.test_db)
     end
 
     before do
-      database[:users].create
-    end
-
-    after do
       database[:users].drop
+      database[:users].create
     end
 
     it 'returns the stripped names of the collections' do
@@ -131,13 +128,10 @@ describe Mongo::Database do
 
       before do
         2.times do |i|
-          database["#{i}_dalmatians"].create
+          database["#{i}_dalmatians"].drop
         end
-      end
-
-      after do
         2.times do |i|
-          database["#{i}_dalmatians"].drop
+          database["#{i}_dalmatians"].create
         end
       end
 
@@ -151,7 +145,7 @@ describe Mongo::Database do
   describe '#list_collections' do
 
     let(:database) do
-      described_class.new(authorized_client, TEST_DB)
+      described_class.new(authorized_client, SpecConfig.instance.test_db)
     end
 
     let(:result) do
@@ -161,11 +155,8 @@ describe Mongo::Database do
     end
 
     before do
-      database[:users].create
-    end
-
-    after do
       database[:users].drop
+      database[:users].create
     end
 
     it 'returns a list of the collections info', if: list_command_enabled?  do
@@ -173,7 +164,7 @@ describe Mongo::Database do
     end
 
     it 'returns a list of the collections info', unless: list_command_enabled?  do
-      expect(result).to include("#{TEST_DB}.users")
+      expect(result).to include("#{SpecConfig.instance.test_db}.users")
     end
   end
 
@@ -182,7 +173,7 @@ describe Mongo::Database do
     context 'when the database exists' do
 
       let(:database) do
-        described_class.new(authorized_client, TEST_DB)
+        described_class.new(authorized_client, SpecConfig.instance.test_db)
       end
 
       let(:collection) do
@@ -190,11 +181,8 @@ describe Mongo::Database do
       end
 
       before do
-        database[:users].create
-      end
-
-      after do
         database[:users].drop
+        database[:users].create
       end
 
       it 'returns collection objects for each name' do
@@ -220,7 +208,7 @@ describe Mongo::Database do
     context 'when the user is not authorized', if: auth_enabled? do
 
       let(:database) do
-        described_class.new(unauthorized_client, TEST_DB)
+        described_class.new(unauthorized_client, SpecConfig.instance.test_db)
       end
 
       it 'raises an exception' do
@@ -234,7 +222,7 @@ describe Mongo::Database do
   describe '#command' do
 
     let(:database) do
-      described_class.new(authorized_client, TEST_DB)
+      described_class.new(authorized_client, SpecConfig.instance.test_db)
     end
 
     it 'sends the query command to the cluster' do
@@ -330,7 +318,7 @@ describe Mongo::Database do
       end
 
       let(:database) do
-        described_class.new(client, TEST_DB, client.options)
+        described_class.new(client, SpecConfig.instance.test_db, client.options)
       end
 
       before do
@@ -353,7 +341,7 @@ describe Mongo::Database do
       end
 
       let(:database) do
-        described_class.new(client, TEST_DB, client.options)
+        described_class.new(client, SpecConfig.instance.test_db, client.options)
       end
 
       before do
@@ -374,7 +362,7 @@ describe Mongo::Database do
       end
 
       let(:database) do
-        described_class.new(client, TEST_DB, client.options)
+        described_class.new(client, SpecConfig.instance.test_db, client.options)
       end
 
       it 'uses the client server_selection_timeout' do
@@ -413,7 +401,7 @@ describe Mongo::Database do
       end
 
       let(:database) do
-        described_class.new(authorized_client.with(client_options), TEST_DB)
+        described_class.new(authorized_client.with(client_options), SpecConfig.instance.test_db)
       end
 
       context 'when a write concern is not in the command selector' do
@@ -452,7 +440,7 @@ describe Mongo::Database do
   describe '#drop' do
 
     let(:database) do
-      described_class.new(authorized_client, TEST_DB)
+      described_class.new(authorized_client, SpecConfig.instance.test_db)
     end
 
     it 'drops the database' do
@@ -512,11 +500,11 @@ describe Mongo::Database do
     context 'when provided a valid name' do
 
       let(:database) do
-        described_class.new(authorized_client, TEST_DB)
+        described_class.new(authorized_client, SpecConfig.instance.test_db)
       end
 
       it 'sets the name as a string' do
-        expect(database.name).to eq(TEST_DB)
+        expect(database.name).to eq(SpecConfig.instance.test_db)
       end
 
       it 'sets the client' do
@@ -537,7 +525,7 @@ describe Mongo::Database do
   describe '#inspect' do
 
     let(:database) do
-      described_class.new(authorized_client, TEST_DB)
+      described_class.new(authorized_client, SpecConfig.instance.test_db)
     end
 
     it 'includes the object id' do
@@ -552,7 +540,7 @@ describe Mongo::Database do
   describe '#fs', unless: sharded? do
 
     let(:database) do
-      described_class.new(authorized_client, TEST_DB)
+      described_class.new(authorized_client, SpecConfig.instance.test_db)
     end
 
     shared_context 'a GridFS database' do
@@ -567,7 +555,7 @@ describe Mongo::Database do
           Mongo::Grid::File.new('Hello!', :filename => 'test.txt')
         end
 
-        after do
+        before do
           fs.files_collection.delete_many
           fs.chunks_collection.delete_many
         end

data/spec/mongo/error/no_server_available_spec.rb

@@ -0,0 +1,22 @@
+require 'spec_helper'
+
+describe Mongo::Error::NoServerAvailable do
+  describe 'message' do
+    let(:selector) do
+      Mongo::ServerSelector::Primary.new
+    end
+
+    let(:cluster) do
+      Mongo::Cluster.new(['127.0.0.1:27017'],
+        Mongo::Monitoring.new, monitoring_io: false)
+    end
+
+    let(:error) do
+      Mongo::Error::NoServerAvailable.new(selector, cluster)
+    end
+
+    it 'is correct' do
+      expect(error.message).to eq('No primary server is available in cluster: #<Cluster topology=Unknown[127.0.0.1:27017] servers=[]> with timeout=30, LT=0.015')
+    end
+  end
+end

data/spec/mongo/error/operation_failure_spec.rb

@@ -149,4 +149,74 @@ describe Mongo::Error::OperationFailure do
       end
     end
   end
+
+  describe '#labels' do
+
+    context 'when the result is nil' do
+
+      subject do
+        described_class.new('not master (10107)', nil,
+          :code => 10107, :code_name => 'NotMaster')
+      end
+
+      it 'has no labels' do
+        expect(subject.labels).to eq([])
+      end
+    end
+
+    context 'when the result is not nil' do
+
+      let(:reply_document) do
+        {
+            'code' => 251,
+            'codeName' => 'NoSuchTransaction',
+            'errorLabels' => labels,
+        }
+      end
+
+      let(:reply) do
+        Mongo::Protocol::Reply.new.tap do |r|
+          # Because this was not created by Mongo::Protocol::Reply::deserialize, we need to manually
+          # initialize the fields.
+          r.instance_variable_set(:@documents, [reply_document])
+          r.instance_variable_set(:@flags, [])
+        end
+      end
+
+      let(:result) do
+        Mongo::Operation::Result.new(reply)
+      end
+
+      subject do
+        begin
+          result.send(:raise_operation_failure)
+        rescue => e
+          e
+        end
+      end
+
+      context 'when the error has no labels' do
+
+        let(:labels) do
+          []
+        end
+
+        it 'has the correct labels' do
+          expect(subject.labels).to eq(labels)
+        end
+      end
+
+
+      context 'when the error has labels' do
+
+        let(:labels) do
+          [ Mongo::Error::TRANSIENT_TRANSACTION_ERROR_LABEL ]
+        end
+
+        it 'has the correct labels' do
+          expect(subject.labels).to eq(labels)
+        end
+      end
+    end
+  end
 end