mongo-ruby-driver/test/test_admin.rb

$LOAD_PATH[0,0] = File.join(File.dirname(__FILE__), '..', 'lib')
require 'mongo'
require 'test/unit'

# NOTE: assumes Mongo is running
class AdminTest < Test::Unit::TestCase

  include Mongo

  @@db = Connection.new(ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost',
                        ENV['MONGO_RUBY_DRIVER_PORT'] || Connection::DEFAULT_PORT).db('ruby-mongo-test')
  @@coll = @@db.collection('test')

  def setup
    # Insert some data to make sure the database itself exists.
    @@coll.remove
    @r1 = @@coll.insert('a' => 1) # collection not created until it's used
    @@coll_full_name = 'ruby-mongo-test.test'
    @admin = @@db.admin
  end

  def teardown
    @admin.profiling_level = :off
    @@coll.remove if @@coll
    @@db.error
  end

  def test_default_profiling_level
    assert_equal :off, @admin.profiling_level
  end

  def test_change_profiling_level
    @admin.profiling_level = :slow_only
    assert_equal :slow_only, @admin.profiling_level
    @admin.profiling_level = :off
    assert_equal :off, @admin.profiling_level
    @admin.profiling_level = :all
    assert_equal :all, @admin.profiling_level
    begin
      @admin.profiling_level = :medium
      fail "shouldn't be able to do this"
    rescue
    end
  end

  def test_profiling_info
    # Perform at least one query while profiling so we have something to see.
    @admin.profiling_level = :all
    @@coll.find()
    @admin.profiling_level = :off

    info = @admin.profiling_info
    assert_kind_of Array, info
    assert info.length >= 1
    first = info.first
    assert_kind_of String, first['info']
    assert_kind_of Time, first['ts']
    assert_kind_of Numeric, first['millis']
  end

  def test_validate_collection
    doc = @admin.validate_collection(@@coll.name)
    assert_not_nil doc
    result = doc['result']
    assert_not_nil result
    assert_match /firstExtent/, result
  end

end