mongo 1.8.6 → 1.12.5

data/lib/mongo.rb

@@ -1,3 +1,17 @@
+# Copyright (C) 2009-2013 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.
+
 module Mongo
   ASCENDING   =  1
   DESCENDING  = -1
@@ -36,45 +50,49 @@ module Mongo
     OP_QUERY_NO_CURSOR_TIMEOUT = 2 ** 4
     OP_QUERY_AWAIT_DATA        = 2 ** 5
     OP_QUERY_EXHAUST           = 2 ** 6
+    OP_QUERY_PARTIAL           = 2 ** 7
 
     REPLY_CURSOR_NOT_FOUND     = 2 ** 0
     REPLY_QUERY_FAILURE        = 2 ** 1
     REPLY_SHARD_CONFIG_STALE   = 2 ** 2
     REPLY_AWAIT_CAPABLE        = 2 ** 3
   end
+
+  module ErrorCode # MongoDB Core Server src/mongo/base/error_codes.err
+    BAD_VALUE                = 2
+    UNKNOWN_ERROR            = 8
+    INVALID_BSON             = 22
+    WRITE_CONCERN_FAILED     = 64
+    MULTIPLE_ERRORS_OCCURRED = 65
+    UNAUTHORIZED             = 13
+
+    # mongod/s 2.6 and above return code 59 when a command doesn't exist.
+    # mongod versions previous to 2.6 and mongos 2.4.x return no error code
+    # when a command does exist.
+    # mongos versions previous to 2.4.0 return code 13390 when a command
+    # does not exist.
+    COMMAND_NOT_FOUND_CODES  = [nil, 59, 13390]
+  end
 end
 
 require 'bson'
 
-require 'mongo/util/thread_local_variable_manager'
-require 'mongo/util/conversions'
-require 'mongo/util/support'
-require 'mongo/util/read_preference'
-require 'mongo/util/write_concern'
-require 'mongo/util/core_ext'
-require 'mongo/util/logging'
-require 'mongo/util/node'
-require 'mongo/util/pool'
-require 'mongo/util/pool_manager'
-require 'mongo/util/sharding_pool_manager'
-require 'mongo/util/server_version'
-require 'mongo/util/socket_util'
-require 'mongo/util/ssl_socket'
-require 'mongo/util/tcp_socket'
-require 'mongo/util/unix_socket'
-require 'mongo/util/uri_parser'
-
+require 'set'
+require 'thread'
+require 'monitor'
 
+require 'mongo/utils'
+require 'mongo/exception'
+require 'mongo/functional'
+require 'mongo/connection'
+require 'mongo/collection_writer'
+require 'mongo/collection'
+require 'mongo/bulk_write_collection_view'
+require 'mongo/cursor'
+require 'mongo/db'
+require 'mongo/gridfs'
 require 'mongo/networking'
 require 'mongo/mongo_client'
 require 'mongo/mongo_replica_set_client'
 require 'mongo/mongo_sharded_client'
 require 'mongo/legacy'
-require 'mongo/collection'
-require 'mongo/cursor'
-require 'mongo/db'
-require 'mongo/exceptions'
-require 'mongo/gridfs/grid_ext'
-require 'mongo/gridfs/grid'
-require 'mongo/gridfs/grid_io'
-require 'mongo/gridfs/grid_file_system'

data/mongo.gemspec

@@ -3,7 +3,7 @@ Gem::Specification.new do |s|
 
   s.version           = File.read(File.join(File.dirname(__FILE__), 'VERSION'))
   s.platform          = Gem::Platform::RUBY
-  s.authors           = ['Tyler Brock', 'Gary Murakami', 'Emily Stolfo', 'Brandon Black', 'Durran Jordan']
+  s.authors           = ['Emily Stolfo', 'Durran Jordan', 'Gary Murakami', 'Tyler Brock', 'Brandon Black']
   s.email             = 'mongodb-dev@googlegroups.com'
   s.homepage          = 'http://www.mongodb.org'
   s.summary           = 'Ruby driver for MongoDB'
@@ -27,5 +27,5 @@ Gem::Specification.new do |s|
   s.require_paths     = ['lib']
   s.has_rdoc          = 'yard'
 
-  s.add_dependency('bson', "~> #{s.version}")
+  s.add_dependency('bson', "#{s.version}")
 end

data/test/functional/authentication_test.rb

@@ -1,18 +1,39 @@
+# Copyright (C) 2009-2013 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.
+
 require 'test_helper'
-require 'shared/authentication'
+require 'shared/authentication/basic_auth_shared'
+require 'shared/authentication/sasl_plain_shared'
+require 'shared/authentication/bulk_api_auth_shared'
+require 'shared/authentication/gssapi_shared'
+require 'shared/authentication/scram_shared'
+
 
 class AuthenticationTest < Test::Unit::TestCase
   include Mongo
-  include AuthenticationTests
+  include BasicAuthTests
+  include SASLPlainTests
+  include BulkAPIAuthTests
+  include GSSAPITests
+  include SCRAMTests
 
   def setup
-    @client = MongoClient.new
-    @db     = @client[MONGO_TEST_DB]
-    init_auth
-  end
-
-  def test_authenticate_with_connection_uri
-    @db.add_user('eunice', 'uritest')
-    assert MongoClient.from_uri("mongodb://eunice:uritest@#{host_port}/#{@db.name}")
+    @client    = standard_connection
+    omit("auth not enabled on mongod") unless auth_enabled?(@client)
+    @admin     = @client['admin']
+    @version   = @client.server_version
+    @db        = @client['ruby-test']
+    @host_info = host_port
   end
 end

data/test/functional/bulk_api_stress_test.rb

@@ -0,0 +1,133 @@
+# Copyright (C) 2009-2013 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.
+
+require 'test_helper'
+
+class BulkApiStressTest < Test::Unit::TestCase
+
+  # Generate a large string of 'size' MB (estimated
+  # by a string of 'size' * 1024 * 1024 characters).
+  def generate_large_string(size)
+    s = "a" * (size * 1024 * 1024)
+  end
+
+  def setup
+    @client = standard_connection
+    @db = @client[TEST_DB]
+    @coll = @db["bulk-api-stress-tests"]
+    @coll.remove
+  end
+
+  def test_ordered_batch_large_inserts
+    bulk = @coll.initialize_ordered_bulk_op
+    s = generate_large_string(4)
+
+    for i in 0..5
+      bulk.insert({:_id => i, :msg => s})
+    end
+    bulk.insert({:_id => 3}) # error
+    bulk.insert({:_id => 100})
+
+    ex = assert_raise BulkWriteError do
+      bulk.execute
+    end
+
+    error_details = ex.result
+    assert_equal 6, error_details["nInserted"]
+    assert_equal 1, error_details["writeErrors"].length
+    error = error_details["writeErrors"][0]
+    assert_equal 11000, error["code"] # duplicate key error
+    assert error["errmsg"].kind_of? String
+    assert_equal 6, error["index"]
+    assert_equal 6, @coll.count()
+  end
+
+  def test_unordered_batch_large_inserts
+    bulk = @coll.initialize_unordered_bulk_op
+    s = generate_large_string(4)
+
+    for i in 0..5
+      bulk.insert({:_id => i, :msg => s})
+    end
+    bulk.insert({:_id => 3}) # error
+    bulk.insert({:_id => 100})
+
+    ex = assert_raise BulkWriteError do
+      bulk.execute
+    end
+
+    error_details = ex.result
+    assert_equal 7, error_details["nInserted"]
+    assert_equal 1, error_details["writeErrors"].length
+    error = error_details["writeErrors"][0]
+    assert_equal 11000, error["code"] # duplicate key error
+    assert error["errmsg"].kind_of? String
+    assert_equal 6, error["index"]
+    assert_equal 7, @coll.count()
+  end
+
+  def test_large_single_insert
+    bulk = @coll.initialize_unordered_bulk_op
+    s = generate_large_string(17)
+    bulk.insert({:a => s})
+# RUBY-730:
+#    ex = assert_raise BulkWriteError do
+#      bulk.execute
+#    end
+  end
+
+  def test_ordered_batch_large_batch
+    bulk = @coll.initialize_ordered_bulk_op
+
+    bulk.insert({:_id => 1600})
+    for i in 0..2000
+      bulk.insert({:_id => i})
+    end
+
+    ex = assert_raise BulkWriteError do
+      bulk.execute
+    end
+
+    error_details = ex.result
+    assert_equal 1601, error_details["nInserted"]
+    assert_equal 1, error_details["writeErrors"].length
+    error = error_details["writeErrors"][0]
+    assert_equal 11000, error["code"] # duplicate key error
+    assert error["errmsg"].kind_of? String
+    assert_equal 1601, error["index"]
+    assert_equal 1601, @coll.count()
+  end
+
+  def test_unordered_batch_large_batch
+    bulk = @coll.initialize_unordered_bulk_op
+
+    bulk.insert({:_id => 1600})
+    for i in 0..2000
+      bulk.insert({:_id => i})
+    end
+
+    ex = assert_raise BulkWriteError do
+      bulk.execute
+    end
+
+    error_details = ex.result
+    assert_equal 2001, error_details["nInserted"]
+    assert_equal 1, error_details["writeErrors"].length
+    error = error_details["writeErrors"][0]
+    assert_equal 11000, error["code"] # duplicate key error
+    assert error["errmsg"].kind_of? String
+    assert_equal 1601, error["index"]
+    assert_equal 2001, @coll.count()
+  end
+end