mongo-ruby-driver/test/safe_test.rb

69 lines
1.7 KiB
Ruby

require File.expand_path("../test_helper", __FILE__)
include Mongo
class SafeTest < Test::Unit::TestCase
context "Safe mode propogation: " do
setup do
@con = standard_connection(:safe => {:w => 1})
@db = @con[MONGO_TEST_DB]
@col = @db['test-safe']
@col.create_index([[:a, 1]], :unique => true)
@col.remove
end
should "propogate safe option on insert" do
@col.insert({:a => 1})
assert_raise_error(OperationFailure, "duplicate key") do
@col.insert({:a => 1})
end
end
should "allow safe override on insert" do
@col.insert({:a => 1})
@col.insert({:a => 1}, :safe => false)
end
should "propogate safe option on update" do
@col.insert({:a => 1})
@col.insert({:a => 2})
assert_raise_error(OperationFailure, "duplicate key") do
@col.update({:a => 2}, {:a => 1})
end
end
should "allow safe override on update" do
@col.insert({:a => 1})
@col.insert({:a => 2})
@col.update({:a => 2}, {:a => 1}, :safe => false)
end
end
context "Safe error objects" do
setup do
@con = standard_connection
@db = @con[MONGO_TEST_DB]
@col = @db['test']
@col.remove
@col.insert({:a => 1})
@col.insert({:a => 1})
@col.insert({:a => 1})
end
should "return object on update" do
response = @col.update({:a => 1}, {"$set" => {:a => 2}},
:multi => true, :safe => true)
assert response['updatedExisting']
assert_equal 3, response['n']
end
should "return object on remove" do
response = @col.remove({}, :safe => true)
assert_equal 3, response['n']
end
end
end