mongo-ruby-driver/tests/test_mongo.rb

65 lines
1.8 KiB
Ruby

$LOAD_PATH[0,0] = File.join(File.dirname(__FILE__), '..', 'lib')
require 'mongo'
require 'test/unit'
# NOTE: assumes Mongo is running
class MongoTest < Test::Unit::TestCase
include XGen::Mongo::Driver
def setup
@host = ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost'
@port = ENV['MONGO_RUBY_DRIVER_PORT'] || Mongo::DEFAULT_PORT
@mongo = Mongo.new(@host, @port)
end
def test_database_info
info = @mongo.database_info
assert_not_nil info
assert_kind_of Hash, info
assert_not_nil info['admin']
assert info['admin'] > 0
end
def test_database_names
names = @mongo.database_names
assert_not_nil names
assert_kind_of Array, names
assert names.length >= 1
assert names.include?('admin')
end
def test_drop_database
db = @mongo.db('will-be-deleted')
coll = db.collection('temp')
coll.clear
coll.insert(:name => 'temp')
assert_equal 1, coll.count()
assert @mongo.database_names.include?('will-be-deleted')
@mongo.drop_database('will-be-deleted')
assert !@mongo.database_names.include?('will-be-deleted')
end
def test_pair
db = Mongo.new({:left => ['foo', 123]})
pair = db.instance_variable_get('@pair')
assert_equal 2, pair.length
assert_equal ['foo', 123], pair[0]
assert_equal ['localhost', Mongo::DEFAULT_PORT], pair[1]
db = Mongo.new({:right => 'bar'})
pair = db.instance_variable_get('@pair')
assert_equal 2, pair.length
assert_equal ['localhost', Mongo::DEFAULT_PORT], pair[0]
assert_equal ['bar', Mongo::DEFAULT_PORT], pair[1]
db = Mongo.new({:right => [123, 'foo'], :left => 'bar'})
pair = db.instance_variable_get('@pair')
assert_equal 2, pair.length
assert_equal ['bar', Mongo::DEFAULT_PORT], pair[0]
assert_equal ['foo', 123], pair[1]
end
end