113 lines
3.2 KiB
Ruby
113 lines
3.2 KiB
Ruby
$:.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
|
|
require './test/replica_sets/rs_test_helper'
|
|
|
|
# NOTE: This test expects a replica set of three nodes to be running on RS.host,
|
|
# on ports TEST_PORT, RS.ports[1], and TEST + 2.
|
|
class ConnectTest < Test::Unit::TestCase
|
|
include Mongo
|
|
|
|
def setup
|
|
RS.restart_killed_nodes
|
|
end
|
|
|
|
def teardown
|
|
RS.restart_killed_nodes
|
|
end
|
|
|
|
def test_connect_with_deprecated_multi
|
|
@conn = Connection.multi([[RS.host, RS.ports[0]], [RS.host, RS.ports[1]]], :name => RS.name)
|
|
assert @conn.is_a?(ReplSetConnection)
|
|
assert @conn.connected?
|
|
end
|
|
|
|
def test_connect_bad_name
|
|
assert_raise_error(ReplicaSetConnectionError, "-wrong") do
|
|
ReplSetConnection.new([RS.host, RS.ports[0]], [RS.host, RS.ports[1]],
|
|
[RS.host, RS.ports[2]], :rs_name => RS.name + "-wrong")
|
|
end
|
|
end
|
|
|
|
def test_connect_timeout
|
|
passed = false
|
|
timeout = 3
|
|
begin
|
|
t0 = Time.now
|
|
ReplSetConnection.new(['192.169.169.1', 27017], :connect_timeout => timeout)
|
|
rescue OperationTimeout
|
|
passed = true
|
|
t1 = Time.now
|
|
end
|
|
|
|
assert passed
|
|
assert t1 - t0 < timeout + 1
|
|
end
|
|
|
|
def test_connect
|
|
@conn = ReplSetConnection.new([RS.host, RS.ports[1]], [RS.host, RS.ports[0]],
|
|
[RS.host, RS.ports[2]], :name => RS.name)
|
|
assert @conn.connected?
|
|
assert @conn.read_primary?
|
|
assert @conn.primary?
|
|
|
|
assert_equal RS.primary, @conn.primary
|
|
assert_equal RS.secondaries.sort, @conn.secondaries.sort
|
|
assert_equal RS.arbiters.sort, @conn.arbiters.sort
|
|
|
|
@conn = ReplSetConnection.new([RS.host, RS.ports[1]], [RS.host, RS.ports[0]],
|
|
:name => RS.name)
|
|
assert @conn.connected?
|
|
end
|
|
|
|
def test_host_port_accessors
|
|
@conn = ReplSetConnection.new([RS.host, RS.ports[0]], [RS.host, RS.ports[1]],
|
|
[RS.host, RS.ports[2]], :name => RS.name)
|
|
|
|
assert_equal @conn.host, RS.primary[0]
|
|
assert_equal @conn.port, RS.primary[1]
|
|
end
|
|
|
|
def test_connect_with_primary_node_killed
|
|
node = RS.kill_primary
|
|
|
|
# Becuase we're killing the primary and trying to connect right away,
|
|
# this is going to fail right away.
|
|
assert_raise_error(ConnectionFailure, "Failed to connect to primary node") do
|
|
@conn = ReplSetConnection.new([RS.host, RS.ports[0]], [RS.host, RS.ports[1]],
|
|
[RS.host, RS.ports[2]])
|
|
end
|
|
|
|
# This allows the secondary to come up as a primary
|
|
rescue_connection_failure do
|
|
@conn = ReplSetConnection.new([RS.host, RS.ports[0]], [RS.host, RS.ports[1]],
|
|
[RS.host, RS.ports[2]])
|
|
end
|
|
end
|
|
|
|
def test_connect_with_secondary_node_killed
|
|
node = RS.kill_secondary
|
|
|
|
@conn = ReplSetConnection.new([RS.host, RS.ports[0]], [RS.host, RS.ports[1]],
|
|
[RS.host, RS.ports[2]])
|
|
assert @conn.connected?
|
|
end
|
|
|
|
def test_connect_with_third_node_killed
|
|
RS.kill(RS.get_node_from_port(RS.ports[2]))
|
|
|
|
@conn = ReplSetConnection.new([RS.host, RS.ports[0]], [RS.host, RS.ports[1]],
|
|
[RS.host, RS.ports[2]])
|
|
assert @conn.connected?
|
|
end
|
|
|
|
def test_connect_with_primary_stepped_down
|
|
RS.step_down_primary
|
|
|
|
rescue_connection_failure do
|
|
@conn = ReplSetConnection.new([RS.host, RS.ports[0]], [RS.host, RS.ports[1]],
|
|
[RS.host, RS.ports[2]])
|
|
end
|
|
assert @conn.connected?
|
|
end
|
|
|
|
end
|