mongo-ruby-driver/test/unit/read_test.rb

116 lines
3.4 KiB
Ruby
Raw Normal View History

require './test/test_helper'
class ReadTest < Test::Unit::TestCase
context "Read mode on standard connection: " do
setup do
@read_preference = :secondary
@con = Mongo::Connection.new('localhost', 27017, :read => @read_preference, :connect => false)
end
end
context "Read mode on replica set connection: " do
setup do
@read_preference = :secondary
@con = Mongo::ReplSetConnection.new(['localhost:27017'], :read => @read_preference, :connect => false)
end
should "store read preference on Connection" do
assert_equal @read_preference, @con.read_preference
end
should "propogate to DB" do
db = @con['foo']
assert_equal @read_preference, db.read_preference
db = @con.db('foo')
assert_equal @read_preference, db.read_preference
db = DB.new('foo', @con)
assert_equal @read_preference, db.read_preference
end
should "allow db override" do
db = DB.new('foo', @con, :read => :primary)
assert_equal :primary, db.read_preference
db = @con.db('foo', :read => :primary)
assert_equal :primary, db.read_preference
end
context "on DB: " do
setup do
@db = @con['foo']
end
should "propogate to collection" do
col = @db.collection('bar')
assert_equal @read_preference, col.read_preference
col = @db['bar']
assert_equal @read_preference, col.read_preference
col = Collection.new('bar', @db)
assert_equal @read_preference, col.read_preference
end
should "allow override on collection" do
col = @db.collection('bar', :read => :primary)
assert_equal :primary, col.read_preference
col = Collection.new('bar', @db, :read => :primary)
assert_equal :primary, col.read_preference
end
end
context "on read mode ops" do
setup do
@col = @con['foo']['bar']
@mock_socket = new_mock_socket
end
should "use default value on query" do
@cursor = @col.find({:a => 1})
sock = new_mock_socket
read_pool = stub(:checkin => true)
2011-11-04 19:12:58 +00:00
@con.stubs(:read_pool).returns(read_pool)
primary_pool = stub(:checkin => true)
2012-03-07 17:38:04 +00:00
sock.stubs(:pool).returns(primary_pool)
2011-11-04 19:12:58 +00:00
@con.stubs(:primary_pool).returns(primary_pool)
@con.expects(:checkout_reader).returns(sock)
@con.expects(:receive_message).with do |o, m, l, s, c, r|
r == nil
end.returns([[], 0, 0])
@cursor.next
end
should "allow override default value on query" do
@cursor = @col.find({:a => 1}, :read => :primary)
sock = new_mock_socket
primary_pool = stub(:checkin => true)
2012-03-07 17:38:04 +00:00
sock.stubs(:pool).returns(primary_pool)
2011-11-04 19:12:58 +00:00
@con.stubs(:primary_pool).returns(primary_pool)
@con.expects(:checkout_writer).returns(sock)
@con.expects(:receive_message).with do |o, m, l, s, c, r|
r == nil
end.returns([[], 0, 0])
@cursor.next
end
should "allow override alternate value on query" do
2011-09-06 18:58:03 +00:00
# TODO: enable this test once we enable reading from tags.
# @con.expects(:receive_message).with do |o, m, l, s, c, r|
# tags = {:dc => "ny"}
# end.returns([[], 0, 0])
assert_raise MongoArgumentError do
@col.find_one({:a => 1}, :read => {:dc => "ny"})
end
end
end
end
end