mongo-ruby-driver/test/replica_sets/refresh_with_threads_test.rb

65 lines
1.6 KiB
Ruby

$:.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
require './test/replica_sets/rs_test_helper'
require 'benchmark'
class ReplicaSetRefreshWithThreadsTest < Test::Unit::TestCase
def setup
ensure_rs
@conn = nil
end
def teardown
@rs.restart_killed_nodes
@conn.close if @conn
end
def test_read_write_load_with_added_nodes
seeds = build_seeds(3)
args = {
:refresh_interval => 5,
:refresh_mode => :sync,
:read => :secondary
}
@conn = ReplSetConnection.new(seeds, args)
@duplicate = @conn[MONGO_TEST_DB]['duplicate']
@unique = @conn[MONGO_TEST_DB]['unique']
@duplicate.insert("test" => "insert")
@duplicate.insert("test" => "update")
@unique.insert("test" => "insert")
@unique.insert("test" => "update")
@unique.create_index("test", :unique => true)
threads = []
10.times do
threads << Thread.new do
1000.times do |i|
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
end
# MongoDB < 2.0 will disconnect clients on rs.reconfig()
if @rs.version.first < 2
assert_raise Mongo::ConnectionFailure do
@rs.add_node
threads.each {|t| t.join }
end
else
@rs.add_node
threads.each {|t| t.join }
end
@conn['admin'].command({:ismaster => 1})
assert_equal 3, @conn.secondary_pools.length
assert_equal 3, @conn.secondaries.length
end
end