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

require 'test/test_helper'

class CollectionTest < Test::Unit::TestCase

  class MockDB < DB
    def connect_to_master
      true
    end
  end

  context "Basic operations: " do 
    setup do 
      @logger = mock()
    end

    should "send update message" do 
      @db = MockDB.new("testing", ['localhost', 27017], :logger => @logger)
      @coll = @db.collection('books')
      @db.expects(:send_message_with_operation).with do |op, msg, log| 
        op == 2001 && log.include?("db.books.update")
      end
      @coll.update({}, {:title => 'Moby Dick'})
    end

    should "send insert message" do 
      @db = MockDB.new("testing", ['localhost', 27017], :logger => @logger)
      @coll = @db.collection('books')
      @db.expects(:send_message_with_operation).with do |op, msg, log| 
        op == 2002 && log.include?("db.books.insert")
      end
      @coll.insert({:title => 'Moby Dick'})
    end

    should "send safe update message" do
      @db = MockDB.new("testing", ['localhost', 27017], :logger => @logger)
      @coll = @db.collection('books')
      @db.expects(:send_message_with_safe_check).with do |op, msg, log|
        op == 2001 && log.include?("db.books.update")
      end
      @coll.update({}, {:title => 'Moby Dick'}, :safe => true)
    end

    should "send safe insert message" do
      @db = MockDB.new("testing", ['localhost', 27017], :logger => @logger)
      @coll = @db.collection('books')
      @db.expects(:send_message_with_safe_check).with do |op, msg, log|
        op == 2001 && log.include?("db.books.update")
      end
      @coll.update({}, {:title => 'Moby Dick'}, :safe => true)
    end
  end
end