96 lines
2.2 KiB
Ruby
96 lines
2.2 KiB
Ruby
require File.expand_path("../test_helper", __FILE__)
|
|
|
|
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, :safe => true)
|
|
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
|