minor: switched additional tests to socket mock

This commit is contained in:
Tyler Brock 2012-03-06 22:56:39 -05:00
parent 03eb8a8c96
commit 8c3283f514
1 changed files with 3 additions and 3 deletions

View File

@ -67,12 +67,12 @@ class ReadTest < Test::Unit::TestCase
context "on read mode ops" do context "on read mode ops" do
setup do setup do
@col = @con['foo']['bar'] @col = @con['foo']['bar']
@mock_socket = stub() @mock_socket = new_mock_socket
end end
should "use default value on query" do should "use default value on query" do
@cursor = @col.find({:a => 1}) @cursor = @col.find({:a => 1})
sock = mock() sock = new_mock_socket
read_pool = stub(:checkin => true) read_pool = stub(:checkin => true)
@con.stubs(:read_pool).returns(read_pool) @con.stubs(:read_pool).returns(read_pool)
primary_pool = stub(:checkin => true) primary_pool = stub(:checkin => true)
@ -87,7 +87,7 @@ class ReadTest < Test::Unit::TestCase
should "allow override default value on query" do should "allow override default value on query" do
@cursor = @col.find({:a => 1}, :read => :primary) @cursor = @col.find({:a => 1}, :read => :primary)
sock = mock() sock = new_mock_socket
primary_pool = stub(:checkin => true) primary_pool = stub(:checkin => true)
@con.stubs(:primary_pool).returns(primary_pool) @con.stubs(:primary_pool).returns(primary_pool)
@con.expects(:checkout_writer).returns(sock) @con.expects(:checkout_writer).returns(sock)