mongo 1.12.5 → 2.0.0

data/spec/mongo/server/connection_spec.rb

@@ -0,0 +1,312 @@
+require 'spec_helper'
+
+describe Mongo::Server::Connection do
+
+  let(:address) do
+    Mongo::Address.new(DEFAULT_ADDRESS)
+  end
+
+  let(:server) do
+    Mongo::Server.new(address, Mongo::Event::Listeners.new)
+  end
+
+  describe '#connect!' do
+
+    context 'when no socket exists' do
+
+      let(:connection) do
+        described_class.new(server)
+      end
+
+      let!(:result) do
+        connection.connect!
+      end
+
+      let(:socket) do
+        connection.send(:socket)
+      end
+
+      it 'returns true' do
+        expect(result).to be true
+      end
+
+      it 'creates a socket' do
+        expect(socket).to_not be_nil
+      end
+
+      it 'connects the socket' do
+        expect(socket).to be_alive
+      end
+    end
+
+    context 'when a socket exists' do
+
+      let(:connection) do
+        described_class.new(server)
+      end
+
+      before do
+        connection.connect!
+        connection.connect!
+      end
+
+      let(:socket) do
+        connection.send(:socket)
+      end
+
+      it 'keeps the socket alive' do
+        expect(socket).to be_alive
+      end
+    end
+
+    context 'when user credentials exist' do
+
+      context 'when the user is not authorized' do
+
+        let(:connection) do
+          described_class.new(
+            server,
+            :user => 'notauser',
+            :password => 'password',
+            :database => TEST_DB
+          )
+        end
+
+        it 'raises an error' do
+          expect {
+            connection.connect!
+          }.to raise_error(Mongo::Auth::Unauthorized)
+        end
+      end
+
+      describe 'when the user is authorized' do
+
+        let(:connection) do
+          described_class.new(
+            server,
+            :user => TEST_USER.name,
+            :password => TEST_USER.password,
+            :database => TEST_DB
+          )
+        end
+
+        before do
+          connection.connect!
+        end
+
+        it 'sets the connection as authenticated' do
+          expect(connection).to be_authenticated
+        end
+      end
+    end
+  end
+
+  describe '#disconnect!' do
+
+    context 'when a socket is not connected' do
+
+      let(:connection) do
+        described_class.new(server)
+      end
+
+      it 'does not raise an error' do
+        expect(connection.disconnect!).to be true
+      end
+    end
+
+    context 'when a socket is connected' do
+
+      let(:connection) do
+        described_class.new(server)
+      end
+
+      before do
+        connection.connect!
+        connection.disconnect!
+      end
+
+      it 'disconnects the socket' do
+        expect(connection.send(:socket)).to be_nil
+      end
+    end
+  end
+
+  describe '#dispatch' do
+
+    let!(:connection) do
+      described_class.new(
+        server,
+        :user => TEST_USER.name,
+        :password => TEST_USER.password,
+        :database => TEST_DB
+      )
+    end
+
+    let(:documents) do
+      [{ 'name' => 'testing' }]
+    end
+
+    let(:insert) do
+      Mongo::Protocol::Insert.new(TEST_DB, TEST_COLL, documents)
+    end
+
+    let(:query) do
+      Mongo::Protocol::Query.new(TEST_DB, TEST_COLL, { 'name' => 'testing' })
+    end
+
+    context 'when providing a single message' do
+
+      let(:reply) do
+        connection.dispatch([ insert, query ])
+      end
+
+      after do
+        authorized_collection.find.delete_many
+      end
+
+      it 'it dispatchs the message to the socket' do
+        expect(reply.documents.first['name']).to eq('testing')
+      end
+    end
+
+    context 'when providing multiple messages' do
+
+      let(:selector) do
+        { :getlasterror => 1 }
+      end
+
+      let(:command) do
+        Mongo::Protocol::Query.new(TEST_DB, '$cmd', selector, :limit => -1)
+      end
+
+      let(:reply) do
+        connection.dispatch([ insert, command ])
+      end
+
+      after do
+        authorized_collection.find.delete_many
+      end
+
+      it 'it dispatchs the message to the socket' do
+        expect(reply.documents.first['ok']).to eq(1.0)
+      end
+    end
+
+    context 'when a network or socket error occurs' do
+
+      let(:socket) do
+        connection.connect!
+        connection.instance_variable_get(:@socket)
+      end
+
+      before do
+        expect(socket).to receive(:write).and_raise(Mongo::Error::SocketError)
+      end
+
+      it 'disconnects and raises the exception' do
+        expect {
+          connection.dispatch([ insert ])
+        }.to raise_error(Mongo::Error::SocketError)
+        expect(connection).to_not be_connected
+      end
+    end
+
+    context 'when the process is forked' do
+
+      let(:insert) do
+        Mongo::Protocol::Insert.new(TEST_DB, TEST_COLL, documents)
+      end
+
+      before do
+        expect(Process).to receive(:pid).at_least(:once).and_return(1)
+      end
+
+      after do
+        authorized_collection.find.delete_many
+      end
+
+      it 'disconnects the connection' do
+        expect(connection).to receive(:disconnect!).and_call_original
+        connection.dispatch([ insert ])
+      end
+
+      it 'sets a new pid' do
+        connection.dispatch([ insert ])
+        expect(connection.pid).to eq(1)
+      end
+    end
+  end
+
+  describe '#initialize' do
+
+    context 'when host and port are provided' do
+
+      let(:connection) do
+        described_class.new(server)
+      end
+
+      it 'sets the address' do
+        expect(connection.address).to eq(server.address)
+      end
+
+      it 'sets the socket to nil' do
+        expect(connection.send(:socket)).to be_nil
+      end
+
+      it 'sets the timeout to the default' do
+        expect(connection.timeout).to eq(5)
+      end
+    end
+
+    context 'when timeout options are provided' do
+
+      let(:connection) do
+        described_class.new(server, socket_timeout: 10)
+      end
+
+      it 'sets the timeout' do
+        expect(connection.timeout).to eq(10)
+      end
+    end
+
+    context 'when ssl options are provided' do
+
+      let(:ssl_options) do
+        { :ssl => true, :ssl_key => 'file', :ssl_key_pass_phrase => 'iamaphrase' }
+      end
+
+      let(:connection) do
+        described_class.new(server, ssl_options)
+      end
+
+      it 'sets the ssl options' do
+        expect(connection.send(:ssl_options)).to eq(ssl_options)
+      end
+    end
+
+    context 'when authentication options are provided' do
+
+      let(:connection) do
+        described_class.new(
+          server,
+          :user => TEST_USER.name,
+          :password => TEST_USER.password,
+          :database => TEST_DB,
+          :auth_mech => :mongodb_cr
+        )
+      end
+
+      let(:user) do
+        Mongo::Auth::User.new(
+          database: TEST_DB,
+          user: TEST_USER.name,
+          password: TEST_USER.password
+        )
+      end
+
+      it 'sets the authentication strategy for the connection' do
+        expect(connection.authenticator.user).to eq(user)
+      end
+    end
+  end
+end

data/spec/mongo/server/description/features_spec.rb

@@ -0,0 +1,138 @@
+require 'spec_helper'
+
+describe Mongo::Server::Description::Features do
+
+  describe '#initialize' do
+
+    context 'when the server wire version range is the same' do
+
+      let(:features) do
+        described_class.new(0..3)
+      end
+
+      it 'sets the server wire version range' do
+        expect(features.server_wire_versions).to eq(0..3)
+      end
+    end
+
+    context 'when the server wire version range is higher' do
+
+      it 'raises an exception' do
+        expect {
+          described_class.new(0..4)
+        }.to raise_error(Mongo::Error::UnsupportedFeatures)
+      end
+    end
+
+    context 'when the server wire version range is lower' do
+
+      let(:features) do
+        described_class.new(0..2)
+      end
+
+      it 'sets the server wire version range' do
+        expect(features.server_wire_versions).to eq(0..2)
+      end
+    end
+  end
+
+  describe '#list_collections_enabled?' do
+
+    context 'when the wire range includes 3' do
+
+      let(:features) do
+        described_class.new(0..3)
+      end
+
+      it 'returns true' do
+        expect(features).to be_list_collections_enabled
+      end
+    end
+
+    context 'when the wire range does not include 3' do
+
+      let(:features) do
+        described_class.new(0..2)
+      end
+
+      it 'returns false' do
+        expect(features).to_not be_list_collections_enabled
+      end
+    end
+  end
+
+  describe '#list_indexes_enabled?' do
+
+    context 'when the wire range includes 3' do
+
+      let(:features) do
+        described_class.new(0..3)
+      end
+
+      it 'returns true' do
+        expect(features).to be_list_indexes_enabled
+      end
+    end
+
+    context 'when the wire range does not include 3' do
+
+      let(:features) do
+        described_class.new(0..2)
+      end
+
+      it 'returns false' do
+        expect(features).to_not be_list_indexes_enabled
+      end
+    end
+  end
+
+  describe '#write_command_enabled?' do
+
+    context 'when the wire range includes 2' do
+
+      let(:features) do
+        described_class.new(0..3)
+      end
+
+      it 'returns true' do
+        expect(features).to be_write_command_enabled
+      end
+    end
+
+    context 'when the wire range does not include 2' do
+
+      let(:features) do
+        described_class.new(0..1)
+      end
+
+      it 'returns false' do
+        expect(features).to_not be_write_command_enabled
+      end
+    end
+  end
+
+  describe '#scram_sha_1_enabled?' do
+
+    context 'when the wire range includes 3' do
+
+      let(:features) do
+        described_class.new(0..3)
+      end
+
+      it 'returns true' do
+        expect(features).to be_scram_sha_1_enabled
+      end
+    end
+
+    context 'when the wire range does not include 3' do
+
+      let(:features) do
+        described_class.new(0..2)
+      end
+
+      it 'returns false' do
+        expect(features).to_not be_scram_sha_1_enabled
+      end
+    end
+  end
+end

data/spec/mongo/server/description/inspector/primary_elected_spec.rb

@@ -0,0 +1,94 @@
+require 'spec_helper'
+
+describe Mongo::Server::Description::Inspector::PrimaryElected do
+
+  let(:listeners) do
+    Mongo::Event::Listeners.new
+  end
+
+  let(:inspection) do
+    described_class.new(listeners)
+  end
+
+  let(:address) do
+    Mongo::Address.new('127.0.0.1:27017')
+  end
+
+  describe '#run' do
+
+    let(:config) do
+      {
+        'ismaster' => false,
+        'secondary' => true,
+        'hosts' => [ '127.0.0.1:27018', '127.0.0.1:27019' ],
+        'setName' => 'test'
+      }
+    end
+
+    let(:description) do
+      Mongo::Server::Description.new(address, config, listeners)
+    end
+
+    let(:updated) do
+      Mongo::Server::Description.new(address, new_config, listeners)
+    end
+
+    let(:listener) do
+      double('listener')
+    end
+
+    before do
+      listeners.add_listener(Mongo::Event::PRIMARY_ELECTED, listener)
+    end
+
+    context 'when the server becomes primary' do
+
+      let(:new_config) do
+        {
+          'ismaster' => true,
+          'secondary' => false,
+          'hosts' => [ '127.0.0.1:27018', '127.0.0.1:27019' ],
+          'setName' => 'test'
+        }
+      end
+
+      it 'fires a primary elected event' do
+        expect(listener).to receive(:handle).with(updated)
+        inspection.run(description, updated)
+      end
+    end
+
+    context 'when the server stays the same' do
+
+      let(:new_config) do
+        {
+          'ismaster' => false,
+          'secondary' => true,
+          'hosts' => [ '127.0.0.1:27018', '127.0.0.1:27019' ],
+          'setName' => 'test'
+        }
+      end
+
+      it 'fires no event' do
+        expect(listener).to_not receive(:handle)
+        inspection.run(description, updated)
+      end
+    end
+
+    context 'when the server becomes mongos' do
+
+      let(:new_config) do
+        {
+          'ismaster' => true,
+          'secondary' => false,
+          'msg' => 'isdbgrid'
+        }
+      end
+
+      it 'fires a primary elected event' do
+        expect(listener).to receive(:handle).with(updated)
+        inspection.run(description, updated)
+      end
+    end
+  end
+end

data/spec/mongo/server/description/inspector/server_added_spec.rb

@@ -0,0 +1,92 @@
+require 'spec_helper'
+
+describe Mongo::Server::Description::Inspector::ServerAdded do
+
+  let(:listeners) do
+    Mongo::Event::Listeners.new
+  end
+
+  let(:inspection) do
+    described_class.new(listeners)
+  end
+
+  let(:address) do
+    Mongo::Address.new('127.0.0.1:27017')
+  end
+
+  describe '#run' do
+
+    let(:config) do
+      {
+        'ismaster' => true,
+        'secondary' => false,
+        'hosts' => [ '127.0.0.1:27018', '127.0.0.1:27019' ],
+        'setName' => 'test'
+      }
+    end
+
+    let(:description) do
+      Mongo::Server::Description.new(address, config, listeners)
+    end
+
+    let(:updated) do
+      Mongo::Server::Description.new(address, new_config, listeners)
+    end
+
+    let(:listener) do
+      double('listener')
+    end
+
+    before do
+      listeners.add_listener(Mongo::Event::SERVER_ADDED, listener)
+    end
+
+    context 'when a host is added' do
+
+      let(:new_config) do
+        { 'hosts' => [ '127.0.0.1:27019', '127.0.0.1:27020' ] }
+      end
+
+      it 'fires a server added event' do
+        expect(listener).to receive(:handle).with('127.0.0.1:27020')
+        inspection.run(description, updated)
+      end
+    end
+
+    context 'when an arbiter is added' do
+
+      let(:new_config) do
+        { 'arbiters' => [ '127.0.0.1:27020' ] }
+      end
+
+      it 'fires a server added event' do
+        expect(listener).to receive(:handle).with('127.0.0.1:27020')
+        inspection.run(description, updated)
+      end
+    end
+
+    context 'when a passive is added' do
+
+      let(:new_config) do
+        { 'passives' => [ '127.0.0.1:27020' ] }
+      end
+
+      it 'fires a server added event' do
+        expect(listener).to receive(:handle).with('127.0.0.1:27020')
+        inspection.run(description, updated)
+      end
+    end
+
+    context 'when no server is added' do
+
+      let(:new_config) do
+        { 'hosts' => [ '127.0.0.1:27018', '127.0.0.1:27019' ] }
+      end
+
+      it 'fires no event' do
+        expect(listener).to_not receive(:handle)
+        inspection.run(description, updated)
+      end
+    end
+  end
+end