mongo 1.12.5 → 2.0.0

data/spec/mongo/database_spec.rb

@@ -0,0 +1,302 @@
+require 'spec_helper'
+
+describe Mongo::Database do
+
+  describe '#==' do
+
+    let(:database) do
+      described_class.new(authorized_client, TEST_DB)
+    end
+
+    context 'when the names are the same' do
+
+      let(:other) do
+        described_class.new(authorized_client, TEST_DB)
+      end
+
+      it 'returns true' do
+        expect(database).to eq(other)
+      end
+    end
+
+    context 'when the names are not the same' do
+
+      let(:other) do
+        described_class.new(authorized_client, :other)
+      end
+
+      it 'returns false' do
+        expect(database).to_not eq(other)
+      end
+    end
+
+    context 'when the object is not a database' do
+
+      it 'returns false' do
+        expect(database).to_not eq('test')
+      end
+    end
+  end
+
+  describe '#[]' do
+
+    let(:database) do
+      described_class.new(authorized_client, TEST_DB)
+    end
+
+    context 'when providing a valid name' do
+
+      let(:collection) do
+        database[:users]
+      end
+
+      it 'returns a new collection' do
+        expect(collection.name).to eq('users')
+      end
+    end
+
+    context 'when providing an invalid name' do
+
+      it 'raises an error' do
+        expect do
+          database[nil]
+        end.to raise_error(Mongo::Error::InvalidCollectionName)
+      end
+    end
+  end
+
+  describe '#collection_names' do
+
+    let(:database) do
+      described_class.new(authorized_client, TEST_DB)
+    end
+
+    before do
+      database[:users].create
+    end
+
+    after do
+      database[:users].drop
+    end
+
+    it 'returns the stripped names of the collections' do
+      expect(database.collection_names).to include('users')
+    end
+
+    it 'does not include system collections' do
+      expect(database.collection_names).to_not include('system.indexes')
+    end
+
+    context 'when specifying a batch size' do
+
+      it 'returns the stripped names of the collections' do
+        expect(database.collection_names(batch_size: 1).to_a).to include('users')
+      end
+    end
+  end
+
+  describe '#collections' do
+
+    context 'when the database exists' do
+
+      let(:database) do
+        described_class.new(authorized_client, TEST_DB)
+      end
+
+      let(:collection) do
+        Mongo::Collection.new(database, 'users')
+      end
+
+      before do
+        database[:users].create
+      end
+
+      after do
+        database[:users].drop
+      end
+
+      it 'returns collection objects for each name' do
+        expect(database.collections).to include(collection)
+      end
+    end
+
+    context 'when the database does not exist' do
+
+      let(:database) do
+        described_class.new(authorized_client, 'invalid_database')
+      end
+
+      it 'returns an empty list', if: write_command_enabled? do
+        expect(database.collections).to be_empty
+      end
+    end
+
+    context 'when the user is not authorized', unless: sharded? do
+
+      let(:database) do
+        described_class.new(unauthorized_client, TEST_DB)
+      end
+
+      it 'raises an exception' do
+        expect {
+          database.collections
+        }.to raise_error(Mongo::Error::OperationFailure)
+      end
+    end
+  end
+
+  describe '#command' do
+
+    let(:database) do
+      described_class.new(authorized_client, TEST_DB)
+    end
+
+    it 'sends the query command to the cluster' do
+      expect(database.command(:ismaster => 1).written_count).to eq(0)
+    end
+
+    context 'when an alternate read preference is specified' do
+
+      before do
+        allow(database.cluster).to receive(:standalone?).and_return(false)
+      end
+
+      let(:read) do
+        { :mode => :secondary, :tag_sets => [{ 'non' => 'existent' }] }
+      end
+
+      let(:client) do
+        authorized_client.with(server_selection_timeout: 2)
+      end
+
+      let(:database) do
+        described_class.new(client, TEST_DB, client.options)
+      end
+
+      it 'uses that read preference', unless: sharded? do
+        expect do
+          database.command({ ping: 1 }, { read: read })
+        end.to raise_error(Mongo::ServerSelector::NoServerAvailable)
+      end
+    end
+  end
+
+  describe '#drop' do
+
+    let(:database) do
+      described_class.new(authorized_client, TEST_DB)
+    end
+
+    it 'drops the database', if: write_command_enabled? do
+      expect(database.drop).to be_successful
+    end
+
+    it 'raises an exception', unless: write_command_enabled? do
+      expect {
+        database.drop
+      }.to raise_error(Mongo::Error::OperationFailure)
+    end
+  end
+
+  describe '#initialize' do
+
+    context 'when provided a valid name' do
+
+      let(:database) do
+        described_class.new(authorized_client, TEST_DB)
+      end
+
+      it 'sets the name as a string' do
+        expect(database.name).to eq(TEST_DB)
+      end
+
+      it 'sets the client' do
+        expect(database.client).to eq(authorized_client)
+      end
+    end
+
+    context 'when the name is nil' do
+
+      it 'raises an error' do
+        expect do
+          described_class.new(authorized_client, nil)
+        end.to raise_error(Mongo::Error::InvalidDatabaseName)
+      end
+    end
+  end
+
+  describe '#inspect' do
+
+    let(:database) do
+      described_class.new(authorized_client, TEST_DB)
+    end
+
+    it 'includes the object id' do
+      expect(database.inspect).to include(database.object_id.to_s)
+    end
+
+    it 'includes the name' do
+      expect(database.inspect).to include(database.name)
+    end
+  end
+
+  describe '#fs' do
+
+    let(:database) do
+      described_class.new(authorized_client, TEST_DB)
+    end
+
+    shared_context 'a GridFS database' do
+
+      it 'returns a Grid::FS for the db' do
+        expect(fs).to be_a(Mongo::Grid::FS)
+      end
+
+      context 'when operating on the fs' do
+
+        let(:file) do
+          Mongo::Grid::File.new('Hello!', :filename => 'test.txt')
+        end
+
+        before do
+          fs.insert_one(file)
+        end
+
+        after do
+          fs.files_collection.find.delete_many
+          fs.chunks_collection.find.delete_many
+        end
+
+        let(:from_db) do
+          fs.find_one(:filename => 'test.txt')
+        end
+
+        it 'returns the assembled file from the db' do
+          expect(from_db.filename).to eq(file.metadata.filename)
+        end
+      end
+    end
+
+    context  'when no options are provided' do
+
+      let(:fs) do
+        database.fs
+      end
+
+      it_behaves_like 'a GridFS database'
+    end
+
+    context 'when a custom prefix is provided' do
+
+      let(:fs) do
+        database.fs(:fs_name => 'grid')
+      end
+
+      it 'sets the custom prefix' do
+        expect(fs.prefix).to eq('grid')
+      end
+
+      it_behaves_like 'a GridFS database'
+    end
+  end
+end

data/spec/mongo/error/parser_spec.rb

@@ -0,0 +1,119 @@
+require 'spec_helper'
+
+describe Mongo::Error::Parser do
+
+  describe '#message' do
+
+    let(:parser) do
+      described_class.new(document)
+    end
+
+    context 'when the document contains no error message' do
+
+      let(:document) do
+        { 'ok' => 1 }
+      end
+
+      it 'returns an empty string' do
+        expect(parser.message).to be_empty
+      end
+    end
+
+    context 'when the document contains an errmsg' do
+
+      let(:document) do
+        { 'errmsg' => 'no such command: notacommand', 'code'=>59 }
+      end
+
+      it 'returns the message' do
+        expect(parser.message).to eq('no such command: notacommand (59)')
+      end
+    end
+
+    context 'when the document contains writeErrors' do
+
+      context 'when only a single error exists' do
+
+        let(:document) do
+          { 'writeErrors' => [{ 'code' => 9, 'errmsg' => 'Unknown modifier: $st' }]}
+        end
+
+        it 'returns the message' do
+          expect(parser.message).to eq('Unknown modifier: $st (9)')
+        end
+      end
+
+      context 'when multiple errors exist' do
+
+        let(:document) do
+          {
+            'writeErrors' => [
+              { 'code' => 9, 'errmsg' => 'Unknown modifier: $st' },
+              { 'code' => 9, 'errmsg' => 'Unknown modifier: $bl' }
+            ]
+          }
+        end
+
+        it 'returns the messages concatenated' do
+          expect(parser.message).to eq(
+            'Unknown modifier: $st (9), Unknown modifier: $bl (9)'
+          )
+        end
+      end
+    end
+
+    context 'when the document contains $err' do
+
+      let(:document) do
+        { '$err' => 'not authorized for query', 'code' => 13 }
+      end
+
+      it 'returns the message' do
+        expect(parser.message).to eq('not authorized for query (13)')
+      end
+    end
+
+    context 'when the document contains err' do
+
+      let(:document) do
+        { 'err' => 'not authorized for query', 'code' => 13 }
+      end
+
+      it 'returns the message' do
+        expect(parser.message).to eq('not authorized for query (13)')
+      end
+    end
+
+    context 'when the document contains writeConcernErrors' do
+
+      context 'when a single error exists' do
+
+        let(:document) do
+          { 'writeConcernError' => [{ 'errmsg' => 'not authorized for query', 'code' => 13 }]}
+        end
+
+        it 'returns the message' do
+          expect(parser.message).to eq('not authorized for query (13)')
+        end
+      end
+
+      context 'when multiple errors exist' do
+
+        let(:document) do
+          {
+            'writeConcernError' => [
+              { 'code' => 9, 'errmsg' => 'Unknown modifier: $st' },
+              { 'code' => 9, 'errmsg' => 'Unknown modifier: $bl' }
+            ]
+          }
+        end
+
+        it 'returns the messages concatenated' do
+          expect(parser.message).to eq(
+            'Unknown modifier: $st (9), Unknown modifier: $bl (9)'
+          )
+        end
+      end
+    end
+  end
+end

data/spec/mongo/event/publisher_spec.rb

@@ -0,0 +1,50 @@
+require 'spec_helper'
+
+describe Mongo::Event::Publisher do
+
+  describe '#publish' do
+
+    let(:listeners) do
+      Mongo::Event::Listeners.new
+    end
+
+    let(:klass) do
+      Class.new do
+        include Mongo::Event::Publisher
+
+        def initialize(listeners)
+          @event_listeners = listeners
+        end
+      end
+    end
+
+    let(:publisher) do
+      klass.new(listeners)
+    end
+
+    let(:listener) do
+      double('listener')
+    end
+
+    context 'when the event has listeners' do
+
+      before do
+        listeners.add_listener('test', listener)
+        listeners.add_listener('test', listener)
+      end
+
+      it 'handles the event for each listener' do
+        expect(listener).to receive(:handle).with('test').twice
+        publisher.publish('test', 'test')
+      end
+    end
+
+    context 'when the event has no listeners' do
+
+      it 'does not handle anything' do
+        expect(listener).to receive(:handle).never
+        publisher.publish('test', 'test')
+      end
+    end
+  end
+end

data/spec/mongo/event/subscriber_spec.rb

@@ -0,0 +1,34 @@
+require 'spec_helper'
+
+describe Mongo::Event::Subscriber do
+
+  let(:listeners) do
+    Mongo::Event::Listeners.new
+  end
+
+  let(:klass) do
+    Class.new do
+      include Mongo::Event::Subscriber
+
+      def initialize(listeners)
+        @event_listeners = listeners
+      end
+    end
+  end
+
+  describe '#subscribe_to' do
+
+    let(:listener) do
+      double('listener')
+    end
+
+    let(:subscriber) do
+      klass.new(listeners)
+    end
+
+    it 'adds subscribes the listener to the publisher' do
+      expect(listeners).to receive(:add_listener).with('test', listener)
+      subscriber.subscribe_to('test', listener)
+    end
+  end
+end