126 lines
3.2 KiB
Ruby
126 lines
3.2 KiB
Ruby
require './test/test_helper'
|
|
|
|
class SafeTest < Test::Unit::TestCase
|
|
|
|
context "Safe mode on connection: " do
|
|
setup do
|
|
@safe_value = {:w => 7}
|
|
@con = Mongo::Connection.new('localhost', 27017, :safe => @safe_value, :connect => false)
|
|
end
|
|
|
|
should "propogate to DB" do
|
|
db = @con['foo']
|
|
assert_equal @safe_value, db.safe
|
|
|
|
|
|
db = @con.db('foo')
|
|
assert_equal @safe_value, db.safe
|
|
|
|
db = DB.new('foo', @con)
|
|
assert_equal @safe_value, db.safe
|
|
end
|
|
|
|
should "allow db override" do
|
|
db = DB.new('foo', @con, :safe => false)
|
|
assert_equal false, db.safe
|
|
|
|
db = @con.db('foo', :safe => false)
|
|
assert_equal false, db.safe
|
|
end
|
|
|
|
context "on DB: " do
|
|
setup do
|
|
@db = @con['foo']
|
|
end
|
|
|
|
should "propogate to collection" do
|
|
col = @db.collection('bar')
|
|
assert_equal @safe_value, col.safe
|
|
|
|
col = @db['bar']
|
|
assert_equal @safe_value, col.safe
|
|
|
|
col = Collection.new('bar', @db)
|
|
assert_equal @safe_value, col.safe
|
|
end
|
|
|
|
should "allow override on collection" do
|
|
col = @db.collection('bar', :safe => false)
|
|
assert_equal false, col.safe
|
|
|
|
col = Collection.new('bar', @db, :safe => false)
|
|
assert_equal false, col.safe
|
|
end
|
|
end
|
|
|
|
context "on operations supporting safe mode" do
|
|
setup do
|
|
@col = @con['foo']['bar']
|
|
end
|
|
|
|
should "use default value on insert" do
|
|
@con.expects(:send_message_with_safe_check).with do |op, msg, log, n, safe|
|
|
safe == @safe_value
|
|
end
|
|
|
|
@col.insert({:a => 1})
|
|
end
|
|
|
|
should "allow override alternate value on insert" do
|
|
@con.expects(:send_message_with_safe_check).with do |op, msg, log, n, safe|
|
|
safe == {:w => 100}
|
|
end
|
|
|
|
@col.insert({:a => 1}, :safe => {:w => 100})
|
|
end
|
|
|
|
should "allow override to disable on insert" do
|
|
@con.expects(:send_message)
|
|
@col.insert({:a => 1}, :safe => false)
|
|
end
|
|
|
|
should "use default value on update" do
|
|
@con.expects(:send_message_with_safe_check).with do |op, msg, log, n, safe|
|
|
safe == @safe_value
|
|
end
|
|
|
|
@col.update({:a => 1}, {:a => 2})
|
|
end
|
|
|
|
should "allow override alternate value on update" do
|
|
@con.expects(:send_message_with_safe_check).with do |op, msg, log, n, safe|
|
|
safe == {:w => 100}
|
|
end
|
|
|
|
@col.update({:a => 1}, {:a => 2}, :safe => {:w => 100})
|
|
end
|
|
|
|
should "allow override to disable on update" do
|
|
@con.expects(:send_message)
|
|
@col.update({:a => 1}, {:a => 2}, :safe => false)
|
|
end
|
|
|
|
should "use default value on remove" do
|
|
@con.expects(:send_message_with_safe_check).with do |op, msg, log, n, safe|
|
|
safe == @safe_value
|
|
end
|
|
|
|
@col.remove
|
|
end
|
|
|
|
should "allow override alternate value on remove" do
|
|
@con.expects(:send_message_with_safe_check).with do |op, msg, log, n, safe|
|
|
safe == {:w => 100}
|
|
end
|
|
|
|
@col.remove({}, :safe => {:w => 100})
|
|
end
|
|
|
|
should "allow override to disable on remove" do
|
|
@con.expects(:send_message)
|
|
@col.remove({}, :safe => false)
|
|
end
|
|
end
|
|
end
|
|
end
|