110 lines
3.0 KiB
Ruby
110 lines
3.0 KiB
Ruby
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 = stub()
|
|
end
|
|
|
|
should "use default value on query" do
|
|
@cursor = @col.find({:a => 1})
|
|
sock = mock()
|
|
sock.expects(:close)
|
|
@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 = mock()
|
|
sock.expects(:close)
|
|
@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
|
|
# 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
|