minor: updates for replica set tests, mock socket stubs

This commit is contained in:
Tyler Brock 2012-03-06 17:06:42 -05:00
parent 274ce690e7
commit 68f19039d2
2 changed files with 3 additions and 2 deletions

View File

@ -50,7 +50,7 @@ class ReadPreferenceTest < Test::Unit::TestCase
@coll.save({:a => 20}, :safe => {:w => 2}) @coll.save({:a => 20}, :safe => {:w => 2})
# Test that reads are going to secondary on ReplSetConnection # Test that reads are going to secondary on ReplSetConnection
@secondary = Connection.new(@rs.host, @conn.read_pool.port, :slave_ok => true) @secondary = Connection.new(@rs.host, @conn.secondary_pool.port, :slave_ok => true)
queries_before = @secondary['admin'].command({:serverStatus => 1})['opcounters']['query'] queries_before = @secondary['admin'].command({:serverStatus => 1})['opcounters']['query']
@coll.find_one @coll.find_one
queries_after = @secondary['admin'].command({:serverStatus => 1})['opcounters']['query'] queries_after = @secondary['admin'].command({:serverStatus => 1})['opcounters']['query']
@ -60,7 +60,7 @@ class ReadPreferenceTest < Test::Unit::TestCase
@conn.refresh @conn.refresh
# Test that reads are only allowed from secondaries # Test that reads are only allowed from secondaries
assert_raise ConnectionFailure.new("Could not connect to a secondary for reading.") do assert_raise ConnectionFailure.new("Could not checkout a socket.") do
@coll.find_one @coll.find_one
end end

View File

@ -84,6 +84,7 @@ class Test::Unit::TestCase
socket.stubs(:setsockopt).with(Socket::IPPROTO_TCP, Socket::TCP_NODELAY, 1) socket.stubs(:setsockopt).with(Socket::IPPROTO_TCP, Socket::TCP_NODELAY, 1)
socket.stubs(:close) socket.stubs(:close)
socket.stubs(:closed?) socket.stubs(:closed?)
socket.stubs(:pool)
socket socket
end end