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

require File.expand_path("../../test_helper", __FILE__)

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