minor: updates for replica set tests, mock socket stubs

test/replica_sets/read_preference_test.rb

@@ -50,7 +50,7 @@ class ReadPreferenceTest < Test::Unit::TestCase
     @coll.save({:a => 20}, :safe => {:w => 2})
 
     # 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']
     @coll.find_one
     queries_after = @secondary['admin'].command({:serverStatus => 1})['opcounters']['query']
@@ -60,7 +60,7 @@ class ReadPreferenceTest < Test::Unit::TestCase
     @conn.refresh
     
     # 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
     end
         

test/test_helper.rb

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