mongo 1.12.5 → 2.0.0.beta

data/spec/mongo/server/connection_pool_spec.rb

@@ -0,0 +1,120 @@
+require 'spec_helper'
+
+describe Mongo::Server::ConnectionPool do
+
+  let(:address) do
+    Mongo::Address.new('127.0.0.1:27017')
+  end
+
+  describe '#checkin' do
+
+    let(:server) do
+      Mongo::Server.new(address, Mongo::Event::Listeners.new)
+    end
+
+    let(:pool) do
+      described_class.get(server)
+    end
+
+    context 'when a connection is checked out on the thread' do
+
+      let!(:connection) do
+        pool.checkout
+      end
+
+      before do
+        pool.checkin(connection)
+      end
+
+      let(:queue) do
+        pool.send(:queue).queue
+      end
+
+      it 'returns the connection to the queue' do
+        expect(queue.size).to eq(1)
+      end
+    end
+  end
+
+  describe '#checkout' do
+
+    let(:server) do
+      Mongo::Server.new(address, Mongo::Event::Listeners.new)
+    end
+
+    let(:pool) do
+      described_class.get(server)
+    end
+
+    context 'when no connection is checked out on the same thread' do
+
+      let!(:connection) do
+        pool.checkout
+      end
+
+      it 'returns a new connection' do
+        expect(connection.address).to eq(server.address)
+      end
+    end
+
+    context 'when a connection is checked out on the same thread' do
+
+      before do
+        pool.checkout
+      end
+
+      it 'returns the threads connection' do
+        expect(pool.checkout.address).to eq(server.address)
+      end
+    end
+
+    context 'when a connection is checked out on a different thread' do
+
+      let!(:connection) do
+        Thread.new { pool.checkout }.value
+      end
+
+      it 'returns a new connection' do
+        expect(pool.checkout.address).to eq(server.address)
+      end
+
+      it 'does not return the same connection instance' do
+        expect(pool.checkout).to_not eql(connection)
+      end
+    end
+  end
+
+  describe '.get' do
+
+    let(:server) do
+      Mongo::Server.new(address, Mongo::Event::Listeners.new)
+    end
+
+    let(:pool) do
+      described_class.get(server)
+    end
+
+    it 'returns the pool for the server' do
+      expect(pool).to eql(described_class.get(server))
+    end
+  end
+
+  describe '#inspect' do
+
+    let(:server) do
+      Mongo::Server.new(address, Mongo::Event::Listeners.new)
+    end
+
+    let(:pool) do
+      described_class.get(server)
+    end
+
+    it 'includes the object id' do
+      expect(pool.inspect).to include(pool.object_id.to_s)
+    end
+
+    it 'includes the queue inspection' do
+      expect(pool.inspect).to include(pool.__send__(:queue).inspect)
+    end
+  end
+end

data/spec/mongo/server/connection_spec.rb

@@ -0,0 +1,289 @@
+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
+
+    let(:delete) do
+      Mongo::Protocol::Delete.new(TEST_DB, TEST_COLL, {})
+    end
+
+    context 'when providing a single message' do
+
+      let(:reply) do
+        connection.dispatch([ insert, query ])
+      end
+
+      # @todo: Can remove this once we have more implemented with global hooks.
+      after do
+        connection.dispatch([ delete ])
+      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
+
+      # @todo: Can remove this once we have more implemented with global hooks.
+      after do
+        connection.dispatch([ delete ])
+      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
+  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(:connection) do
+        described_class.new(server, :ssl => true)
+      end
+
+      it 'sets the ssl options' do
+        expect(connection.send(:ssl_options)).to eq(:ssl => true)
+      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