mongo 1.12.5 → 2.0.0.beta

data/spec/mongo/write_concern/acknowledged_spec.rb

@@ -0,0 +1,44 @@
+require 'spec_helper'
+
+describe Mongo::WriteConcern::Acknowledged do
+
+  describe '#get_last_error' do
+
+    let(:get_last_error) do
+      concern.get_last_error
+    end
+
+    context 'when the options are symbols' do
+
+      let(:concern) do
+        described_class.new(:w => :majority)
+      end
+
+      it 'converts the values to strings' do
+        expect(get_last_error).to eq(:getlasterror => 1, :w => 'majority')
+      end
+    end
+
+    context 'when the options are strings' do
+
+      let(:concern) do
+        described_class.new(:w => 'majority')
+      end
+
+      it 'keeps the values as strings' do
+        expect(get_last_error).to eq(:getlasterror => 1, :w => 'majority')
+      end
+    end
+
+    context 'when the options are numbers' do
+
+      let(:concern) do
+        described_class.new(:w => 3)
+      end
+
+      it 'keeps the values as numbers' do
+        expect(get_last_error).to eq(:getlasterror => 1, :w => 3)
+      end
+    end
+  end
+end

data/spec/mongo/write_concern/unacknowledged_spec.rb

@@ -0,0 +1,15 @@
+require 'spec_helper'
+
+describe Mongo::WriteConcern::Unacknowledged do
+
+  let(:concern) do
+    described_class.new(:w => 0)
+  end
+
+  describe '#get_last_error' do
+
+    it 'returns nil' do
+      expect(concern.get_last_error).to be_nil
+    end
+  end
+end

data/spec/mongo_orchestration_spec.rb

@@ -0,0 +1,70 @@
+require 'spec_helper'
+
+#describe MongoOrchestration, if: mongo_orchestration_available? do
+#
+#  context 'when creating a standalone' do
+#
+#    context 'when the mongo orchestration service is available' do
+#
+#      before do
+#        initialize_mo_standalone!
+#      end
+#
+#      after do
+#        stop_mo_standalone!
+#      end
+#
+#      it 'sets up a standalone object with an id' do
+#        expect(@standalone.id).to_not be_nil
+#      end
+#
+#      it 'sets up a standalone object with a client' do
+#        expect(@standalone.client).to_not be_nil
+#      end
+#
+#      it 'sets the correct cluster topology' do
+#        expect(@standalone.client.cluster.topology).to eq(Mongo::Cluster::Topology::Standalone)
+#      end
+#
+#      it 'sets the path' do
+#        expect(@standalone.path).to eq(MongoOrchestration::DEFAULT_BASE_URI)
+#      end
+#    end
+#
+#    context 'when the mongo orchestration service is not available' do
+#
+#      it 'raises an error' do
+#        expect do
+#          initialize_mo_standalone!('http://localhost:1')
+#        end.to raise_exception(MongoOrchestration::ServiceNotAvailable)
+#      end
+#    end
+#  end
+#
+#  context 'when stopping a standalone' do
+#
+#    before do
+#      initialize_mo_standalone!
+#      stop_mo_standalone!
+#    end
+#
+#    it 'stops the standalone' do
+#      expect(@standalone.alive?).to eq(false)
+#    end
+#  end
+#
+#  context 'when making a direct request' do
+#
+#    before do
+#      initialize_mo_standalone!
+#    end
+#
+#    after do
+#      stop_mo_standalone!
+#    end
+#
+#    it 'returns the response' do
+#      expect(@standalone.request('GET', 'servers').class).to be(HTTParty::Response)
+#    end
+#  end
+#end

data/spec/spec_helper.rb

@@ -0,0 +1,148 @@
+if RUBY_VERSION > '1.9' && RUBY_VERSION < '2.2'
+  require 'simplecov'
+  require 'coveralls'
+
+  SimpleCov.formatter = SimpleCov::Formatter::MultiFormatter[
+    SimpleCov::Formatter::HTMLFormatter,
+    Coveralls::SimpleCov::Formatter
+  ]
+
+  SimpleCov.start do
+    # report groups
+    add_group 'Wire Protocol', 'lib/mongo/protocol'
+    # filters
+    add_filter 'tasks'
+    add_filter 'spec'
+    add_filter 'bin'
+  end
+end
+
+require 'mongo'
+
+require 'support/travis'
+require 'support/matchers'
+require 'support/authorization'
+require 'support/mongo_orchestration'
+require 'support/server_discovery_and_monitoring'
+require 'support/server_selection_rtt'
+require 'support/server_selection'
+
+Mongo::Logger.logger = Logger.new($stdout)
+Mongo::Logger.logger.level = Logger::INFO
+
+RSpec.configure do |config|
+  config.color     = true
+  config.fail_fast = true unless ENV['CI'] || ENV['JENKINS_HOME']
+  config.formatter = 'documentation'
+  config.include(Authorization)
+
+  config.before(:suite) do
+
+    begin
+      # Create the root user administrator as the first user to be added to the
+      # database. This user will need to be authenticated in order to add any
+      # more users to any other databases.
+      ADMIN_UNAUTHORIZED_CLIENT.database.users.create(ROOT_USER)
+    rescue Exception => e
+    end
+    begin
+      # Adds the test user to the test database with permissions on all
+      # databases that will be used in the test suite.
+      ADMIN_AUTHORIZED_CLIENT.database.users.create(TEST_USER)
+    rescue Exception => e
+      unless write_command_enabled?
+        # If we are on versions less than 2.6, we need to create a user for
+        # each database, since the users are not stored in the admin database
+        # but in the system.users collection on the datbases themselves. Also,
+        # roles in versions lower than 2.6 can only be strings, not hashes.
+        begin ROOT_AUTHORIZED_CLIENT.database.users.create(TEST_READ_WRITE_USER); rescue; end
+      end
+    end
+  end
+end
+
+TEST_SET = 'ruby-driver-rs'
+COVERAGE_MIN = 90
+CURRENT_PATH = File.expand_path(File.dirname(__FILE__))
+SERVER_DISCOVERY_TESTS = Dir.glob("#{CURRENT_PATH}/support/sdam/**/*.yml")
+SERVER_SELECTION_RTT_TESTS = Dir.glob("#{CURRENT_PATH}/support/server_selection/rtt/*.yml")
+SERVER_SELECTION_TESTS = Dir.glob("#{CURRENT_PATH}/support/server_selection/selection/**/*.yml")
+
+# Determine whether the test clients are connecting to a standlone.
+#
+# @since 2.0.0
+def standalone?
+  $mongo_client ||= initialize_scanned_client!
+  $standalone ||= $mongo_client.cluster.standalone?
+end
+
+# Determine whether the test clients are connecting to a replica set.
+#
+# @since 2.0.0
+def replica_set?
+  $mongo_client ||= initialize_scanned_client!
+  $replica_set ||= $mongo_client.cluster.replica_set?
+end
+
+# Determine whether the test clients are connecting to a sharded cluster.
+#
+# @since 2.0.0
+def sharded?
+  $mongo_client ||= initialize_scanned_client!
+  $sharded ||= $mongo_client.cluster.sharded?
+end
+
+# For instances where behaviour is different on different versions, we need to
+# determine in the specs if we are 2.6 or higher.
+#
+# @since 2.0.0
+def write_command_enabled?
+  $mongo_client ||= initialize_scanned_client!
+  $write_command_enabled ||= $mongo_client.cluster.servers.first.features.write_command_enabled?
+end
+
+# For instances where behaviour is different on different versions, we need to
+# determine in the specs if we are 2.7 or higher.
+#
+# @since 2.0.0
+def list_command_enabled?
+  $mongo_client ||= initialize_scanned_client!
+  $list_command_enabled ||= $mongo_client.cluster.servers.first.features.list_indexes_enabled?
+end
+
+alias :scram_sha_1_enabled? :list_command_enabled?
+
+# Depending on whether write commands are enabled, there are different documents that
+# are guaranteed to cause a delete failure.
+#
+# @since 2.0.0
+def failing_delete_doc
+  write_command_enabled? ? { q: { '$set' => { a: 1 } }, limit: 0 } :
+                           { que: { field: 'test' } }
+end
+
+# Initializes a basic scanned client to do an ismaster check.
+#
+# @since 2.0.0
+def initialize_scanned_client!
+  Mongo::Client.new(ADDRESSES, database: TEST_DB, connect: CONNECT)
+end
+
+def initialize_mo_standalone!(path = nil)
+  $mongo_standalone ||= MongoOrchestration.get(:standalone, path: path)
+end
+
+def stop_mo_standalone!
+  $mongo_standalone.stop if $mongo_standalone
+end
+
+def mongo_orchestration_available?(path = nil)
+  begin
+    MongoOrchestration.get(:standalone, path: path)
+  rescue MongoOrchestration::ServiceNotAvailable
+    return false
+  end
+end
+
+# require all shared examples
+Dir['./spec/support/shared/*.rb'].sort.each { |file| require file }

data/spec/support/authorization.rb

@@ -0,0 +1,245 @@
+# Copyright (C) 2009-2014 MongoDB, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# The default test database for all specs.
+#
+# @since 2.0.0
+TEST_DB = 'ruby-driver'.freeze
+
+# The default test collection.
+#
+# @since 2.0.0
+TEST_COLL = 'test'.freeze
+
+# The seed addresses to be used when creating a client.
+#
+# @since 2.0.0
+ADDRESSES = ENV['MONGODB_ADDRESSES'] ? ENV['MONGODB_ADDRESSES'].split(',').freeze :
+              [ '127.0.0.1:27017' ].freeze
+
+# A default address to use in tests.
+#
+# @since 2.0.0
+DEFAULT_ADDRESS = ADDRESSES.first
+
+# The topology type.
+#
+# @since 2.0.0
+CONNECT = ENV['RS_ENABLED'] == 'true' ? :replica_set.freeze :
+            ENV['SHARDED_ENABLED'] == 'true' ? :sharded.freeze :
+            :direct.freeze
+
+# The root user name.
+#
+# @since 2.0.0
+ROOT_USER_NAME = ENV['ROOT_USER_NAME'] || 'root-user'
+
+# The root user password.
+#
+# @since 2.0.0
+ROOT_USER_PWD = ENV['ROOT_USER_PWD'] || 'password'
+
+# Gets the root system administrator user.
+#
+# @since 2.0.0
+ROOT_USER = Mongo::Auth::User.new(
+  database: Mongo::Database::ADMIN,
+  user: ROOT_USER_NAME,
+  password: ROOT_USER_PWD,
+  roles: [
+    Mongo::Auth::Roles::USER_ADMIN_ANY_DATABASE,
+    Mongo::Auth::Roles::DATABASE_ADMIN_ANY_DATABASE,
+    Mongo::Auth::Roles::READ_WRITE_ANY_DATABASE,
+    Mongo::Auth::Roles::HOST_MANAGER
+  ]
+)
+
+# Get the default test user for the suite on versions 2.6 and higher.
+#
+# @since 2.0.0
+TEST_USER = Mongo::Auth::User.new(
+  database: Mongo::Database::ADMIN,
+  user: 'test-user',
+  password: 'password',
+  roles: [
+    { role: Mongo::Auth::Roles::READ_WRITE, db: TEST_DB },
+    { role: Mongo::Auth::Roles::DATABASE_ADMIN, db: TEST_DB }
+  ]
+)
+
+# MongoDB 2.4 and lower does not allow hashes as roles, so we need to create a
+# user on those versions for each database permission in order to ensure the
+# legacy roles work with users. The following users are those.
+
+# Gets the default test user for the suite on 2.4 and lower.
+#
+# @since 2.0.
+TEST_READ_WRITE_USER = Mongo::Auth::User.new(
+  database: TEST_DB,
+  user: 'test-user',
+  password: 'password',
+  roles: [ Mongo::Auth::Roles::READ_WRITE, Mongo::Auth::Roles::DATABASE_ADMIN ]
+)
+
+# The write concern to use in the tests.
+#
+# @since 2.0.0
+WRITE_CONCERN = CONNECT == :direct || :sharded ? { w: 1 } : { w: 'majority' }
+
+# Provides an authorized mongo client on the default test database for the
+# default test user.
+#
+# @since 2.0.0
+AUTHORIZED_CLIENT = Mongo::Client.new(
+  ADDRESSES,
+  database: TEST_DB,
+  user: TEST_USER.name,
+  password: TEST_USER.password,
+  max_pool_size: 1,
+  write: WRITE_CONCERN,
+  connect: CONNECT
+)
+
+# Provides an authorized mongo client on the default test database for the
+# default root system administrator.
+#
+# @since 2.0.0
+ROOT_AUTHORIZED_CLIENT = Mongo::Client.new(
+  ADDRESSES,
+  auth_source: Mongo::Database::ADMIN,
+  database: TEST_DB,
+  user: ROOT_USER.name,
+  password: ROOT_USER.password,
+  max_pool_size: 1,
+  write: WRITE_CONCERN,
+  connect: CONNECT
+)
+
+# Provides an unauthorized mongo client on the default test database.
+#
+# @since 2.0.0
+UNAUTHORIZED_CLIENT = Mongo::Client.new(
+  ADDRESSES,
+  database: TEST_DB,
+  max_pool_size: 1,
+  write: WRITE_CONCERN,
+  connect: CONNECT
+)
+
+# Provides an unauthorized mongo client on the admin database, for use in
+# setting up the first admin root user.
+#
+# @since 2.0.0
+ADMIN_UNAUTHORIZED_CLIENT = Mongo::Client.new(
+  ADDRESSES,
+  database: Mongo::Database::ADMIN,
+  max_pool_size: 1,
+  write: WRITE_CONCERN,
+  connect: CONNECT
+)
+
+# Get an authorized client on the admin database logged in as the admin
+# root user.
+#
+# @since 2.0.0
+ADMIN_AUTHORIZED_CLIENT = ADMIN_UNAUTHORIZED_CLIENT.with(
+  user: ROOT_USER.name,
+  password: ROOT_USER.password
+)
+
+module Authorization
+
+  # On inclusion provides helpers for use with testing with and without
+  # authorization.
+  #
+  #
+  # @since 2.0.0
+  def self.included(context)
+
+    # Gets the root system administrator user.
+    #
+    # @since 2.0.0
+    context.let(:root_user) { ROOT_USER }
+
+    # Get the default test user for the suite.
+    #
+    # @since 2.0.0
+    context.let(:test_user) { TEST_USER }
+
+    # Provides an authorized mongo client on the default test database for the
+    # default test user.
+    #
+    # @since 2.0.0
+    context.let(:authorized_client) { AUTHORIZED_CLIENT }
+
+    # Provides an authorized mongo client on the default test database for the
+    # default root system administrator.
+    #
+    # @since 2.0.0
+    context.let(:root_authorized_client) { ROOT_AUTHORIZED_CLIENT }
+
+    # Provides an unauthorized mongo client on the default test database.
+    #
+    # @since 2.0.0
+    context.let!(:unauthorized_client) { UNAUTHORIZED_CLIENT }
+
+    # Provides an unauthorized mongo client on the admin database, for use in
+    # setting up the first admin root user.
+    #
+    # @since 2.0.0
+    context.let!(:admin_unauthorized_client) { ADMIN_UNAUTHORIZED_CLIENT }
+
+    # Get an authorized client on the admin database logged in as the admin
+    # root user.
+    #
+    # @since 2.0.0
+    context.let!(:admin_authorized_client) { ADMIN_AUTHORIZED_CLIENT }
+
+    # Gets the default test collection from the authorized client.
+    #
+    # @since 2.0.0
+    context.let(:authorized_collection) do
+      authorized_client[TEST_COLL]
+    end
+
+    # Gets the default test collection from the unauthorized client.
+    #
+    # @since 2.0.0
+    context.let(:unauthorized_collection) do
+      unauthorized_client[TEST_COLL]
+    end
+
+    # Gets a primary server for the default authorized client.
+    #
+    # @since 2.0.0
+    context.let(:authorized_primary) do
+      authorized_client.cluster.next_primary
+    end
+
+    # Get a primary server for the client authorized as the root system
+    # administrator.
+    #
+    # @since 2.0.0
+    context.let(:root_authorized_primary) do
+      root_authorized_client.cluster.next_primary
+    end
+
+    # Get a primary server from the unauthorized client.
+    #
+    # @since 2.0.0
+    context.let(:unauthorized_primary) do
+      authorized_client.cluster.next_primary
+    end
+  end
+end