75 lines
2.2 KiB
Ruby
75 lines
2.2 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
|
|
@mongo.drop_database('ruby-mongo-info-test')
|
|
@mongo.db('ruby-mongo-info-test').collection('info-test').insert('a' => 1)
|
|
|
|
info = @mongo.database_info
|
|
assert_not_nil info
|
|
assert_kind_of Hash, info
|
|
assert_not_nil info['ruby-mongo-info-test']
|
|
assert info['ruby-mongo-info-test'] > 0
|
|
|
|
@mongo.drop_database('ruby-mongo-info-test')
|
|
end
|
|
|
|
def test_database_names
|
|
@mongo.drop_database('ruby-mongo-info-test')
|
|
@mongo.db('ruby-mongo-info-test').collection('info-test').insert('a' => 1)
|
|
|
|
names = @mongo.database_names
|
|
assert_not_nil names
|
|
assert_kind_of Array, names
|
|
assert names.length >= 1
|
|
assert names.include?('ruby-mongo-info-test')
|
|
|
|
@mongo.drop_database('ruby-mongo-info-test')
|
|
end
|
|
|
|
def test_drop_database
|
|
db = @mongo.db('ruby-mongo-will-be-deleted')
|
|
coll = db.collection('temp')
|
|
coll.clear
|
|
coll.insert(:name => 'temp')
|
|
assert_equal 1, coll.count()
|
|
assert @mongo.database_names.include?('ruby-mongo-will-be-deleted')
|
|
|
|
@mongo.drop_database('ruby-mongo-will-be-deleted')
|
|
assert !@mongo.database_names.include?('ruby-mongo-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
|