require './test/test_helper' class TestThreading < Test::Unit::TestCase include Mongo @@con = standard_connection(:pool_size => 10, :pool_timeout => 30) @@db = @@con[MONGO_TEST_DB] @@coll = @@db.collection('thread-test-collection') def set_up_safe_data @@db.drop_collection('duplicate') @@db.drop_collection('unique') @duplicate = @@db.collection('duplicate') @unique = @@db.collection('unique') @duplicate.insert("test" => "insert") @duplicate.insert("test" => "update") @unique.insert("test" => "insert") @unique.insert("test" => "update") @unique.create_index("test", :unique => true) end def test_safe_update times = [] set_up_safe_data threads = [] 100.times do |i| threads[i] = Thread.new do 100.times do if i % 2 == 0 assert_raise Mongo::OperationFailure do t1 = Time.now @unique.update({"test" => "insert"}, {"$set" => {"test" => "update"}}, :safe => true) times << Time.now - t1 end else t1 = Time.now @duplicate.update({"test" => "insert"}, {"$set" => {"test" => "update"}}, :safe => true) times << Time.now - t1 end end end end 100.times do |i| threads[i].join end end def test_safe_insert set_up_safe_data threads = [] 100.times do |i| threads[i] = Thread.new do if i % 2 == 0 assert_raise Mongo::OperationFailure do @unique.insert({"test" => "insert"}, :safe => true) end else @duplicate.insert({"test" => "insert"}, :safe => true) end end end 100.times do |i| threads[i].join end end def test_threading @@coll.drop @@coll = @@db.collection('thread-test-collection') 1000.times do |i| @@coll.insert("x" => i) end 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