mongo 1.3.0 → 1.12.5

data/test/test_helper.rb

@@ -1,102 +1,61 @@
-$:.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
-require 'rubygems' if RUBY_VERSION < '1.9.0' && ENV['C_EXT']
-require 'mongo'
-require 'test/unit'
-
-def silently
-  warn_level = $VERBOSE
-  $VERBOSE = nil
-  result = yield
-  $VERBOSE = warn_level
-  result
-end
-
-begin
-  require 'rubygems' if RUBY_VERSION < "1.9.0" && !ENV['C_EXT']
-  silently { require 'shoulda' }
-  silently { require 'mocha' }
-rescue LoadError
-  puts <<MSG
-
-This test suite requires shoulda and mocha.
-You can install them as follows:
-  gem install shoulda
-  gem install mocha
-
-MSG
-
-  exit
-end
-
-require 'bson_ext/cbson' if !(RUBY_PLATFORM =~ /java/) && ENV['C_EXT']
-
-unless defined? MONGO_TEST_DB
-  MONGO_TEST_DB = 'ruby-test-db'
-end
-
-unless defined? TEST_PORT
-  TEST_PORT = ENV['MONGO_RUBY_DRIVER_PORT'] ? ENV['MONGO_RUBY_DRIVER_PORT'].to_i : Mongo::Connection::DEFAULT_PORT
-end
-
-unless defined? TEST_HOST
-  TEST_HOST = ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost'
-end
-
-class Test::Unit::TestCase
-  include Mongo
-  include BSON
-
-  def self.standard_connection(options={})
-    Connection.new(TEST_HOST, TEST_PORT, options)
-  end
-
-  def standard_connection(options={})
-    self.class.standard_connection(options)
-  end
-
-  def self.host_port
-    "#{mongo_host}:#{mongo_port}"
-  end
-
-  def self.mongo_host
-    TEST_HOST
-  end
-
-  def self.mongo_port
-    TEST_PORT
+# 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.
+
+# NOTE: on ruby <1.9 you need to run individual tests with 'bundle exec'
+
+unless RUBY_VERSION < '1.9' || ENV.key?('JENKINS_CI')
+  require 'simplecov'
+  require 'coveralls'
+
+  SimpleCov.formatter = SimpleCov::Formatter::MultiFormatter[
+    SimpleCov::Formatter::HTMLFormatter,
+    Coveralls::SimpleCov::Formatter
+  ]
+
+  SimpleCov.start do
+    add_group 'Driver', 'lib/mongo'
+    add_group 'BSON', 'lib/bson'
+
+    add_filter 'tasks'
+    add_filter 'test'
+    add_filter 'bin'
   end
+end
 
-  def host_port
-    self.class.host_port
-  end
+# required for at_exit, at_start hooks
+require 'test-unit'
 
-  def mongo_host
-    self.class.mongo_host
-  end
+require 'test/unit'
+require 'shoulda'
+require 'mocha/setup'
 
-  def mongo_port
-    self.class.mongo_port
-  end
+# cluster manager
+require 'tools/mongo_config'
 
-  def new_mock_socket(host='localhost', port=27017)
-    socket = Object.new
-    socket.stubs(:setsockopt).with(Socket::IPPROTO_TCP, Socket::TCP_NODELAY, 1)
-    socket.stubs(:close)
-    socket
-  end
+# For kerberos testing.
+begin
+  require 'mongo_kerberos'
+rescue LoadError; end
 
-  def new_mock_db
-    db = Object.new
-  end
+# test helpers
+require 'helpers/general'
+require 'helpers/test_unit'
 
-  def assert_raise_error(klass, message)
-    begin
-      yield
-    rescue => e
-      assert_equal klass, e.class
-      assert e.message.include?(message), "#{e.message} does not include #{message}."
-    else
-      flunk "Expected assertion #{klass} but none was raised."
-    end
-  end
+# optional development and debug utilities
+begin
+  require 'pry-rescue'
+  require 'pry-nav'
+rescue LoadError
+  # failed to load, skipping pry
 end

data/test/threading/basic_test.rb

@@ -0,0 +1,120 @@
+# 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 ThreadingTest < Test::Unit::TestCase
+
+  include Mongo
+
+  def setup
+    @client = standard_connection(:pool_size => 10, :pool_timeout => 30)
+    @db = @client.db(TEST_DB)
+    @coll = @db.collection('thread-test-collection')
+    @coll.drop
+
+    collections = ['duplicate', 'unique']
+
+    collections.each do |coll_name|
+      coll = @db.collection(coll_name)
+      coll.drop
+      coll.insert("test" => "insert")
+      coll.insert("test" => "update")
+      instance_variable_set("@#{coll_name}", coll)
+    end
+
+    @unique.create_index("test", :unique => true)
+  end
+
+  def test_safe_update
+    threads = []
+    300.times do |i|
+      threads << Thread.new do
+        if i % 2 == 0
+          assert_raise Mongo::OperationFailure do
+            @unique.update({"test" => "insert"}, {"$set" => {"test" => "update"}})
+          end
+        else
+          @duplicate.update({"test" => "insert"}, {"$set" => {"test" => "update"}})
+          @duplicate.update({"test" => "update"}, {"$set" => {"test" => "insert"}})
+        end
+      end
+    end
+
+    threads.each {|thread| thread.join}
+  end
+
+  def test_safe_insert
+    threads = []
+    300.times do |i|
+      threads << Thread.new do
+        if i % 2 == 0
+          assert_raise Mongo::OperationFailure do
+            @unique.insert({"test" => "insert"})
+          end
+        else
+          @duplicate.insert({"test" => "insert"})
+        end
+      end
+    end
+
+    threads.each {|thread| thread.join}
+  end
+
+  def test_concurrent_find
+    n_threads = 50
+
+    1000.times do |i|
+      @coll.insert({ "x" => "a" })
+    end
+
+    threads = []
+    n_threads.times do |i|
+      threads << Thread.new do
+        sum = 0
+        @coll.find.to_a.size
+      end
+    end
+
+    thread_values = threads.map(&:value)
+    assert thread_values.all?{|v| v == 1000}
+    assert_equal thread_values.size, n_threads
+  end
+
+  def test_threading
+    @coll.drop
+    @coll = @db.collection('thread-test-collection')
+
+    docs = []
+    1000.times {|i| docs << {:x => i}}
+    @coll.insert(docs)
+
+    threads = []
+
+    10.times do |i|
+      threads[i] = Thread.new do
+        sum = 0
+        @coll.find().each do |document|
+          sum += document["x"]
+        end
+        assert_equal 499500, sum
+      end
+    end
+
+    10.times do |i|
+      threads[i].join
+    end
+  end
+
+end