40 lines
1.0 KiB
Ruby
40 lines
1.0 KiB
Ruby
$:.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
|
|
require './test/test_helper'
|
|
require './test/tools/repl_set_manager'
|
|
|
|
class Test::Unit::TestCase
|
|
# Ensure replica set is available as an instance variable and that
|
|
# a new set is spun up for each TestCase class
|
|
def ensure_rs
|
|
unless defined?(@@current_class) and @@current_class == self.class
|
|
@@current_class = self.class
|
|
@@rs = ReplSetManager.new
|
|
@@rs.start_set
|
|
end
|
|
@rs = @@rs
|
|
end
|
|
|
|
# Generic code for rescuing connection failures and retrying operations.
|
|
# This could be combined with some timeout functionality.
|
|
def rescue_connection_failure(max_retries=30)
|
|
retries = 0
|
|
begin
|
|
yield
|
|
rescue Mongo::ConnectionFailure => ex
|
|
#puts "Rescue attempt #{retries}: from #{ex}"
|
|
retries += 1
|
|
raise ex if retries > max_retries
|
|
sleep(2)
|
|
retry
|
|
end
|
|
end
|
|
|
|
def build_seeds(num_hosts)
|
|
seeds = []
|
|
num_hosts.times do |n|
|
|
seeds << "#{@rs.host}:#{@rs.ports[n]}"
|
|
end
|
|
seeds
|
|
end
|
|
end
|