126 lines
3.8 KiB
Ruby
126 lines
3.8 KiB
Ruby
require 'test/test_helper'
|
|
require 'logger'
|
|
require 'stringio'
|
|
|
|
# NOTE: assumes Mongo is running
|
|
class TestConnection < Test::Unit::TestCase
|
|
|
|
include Mongo
|
|
|
|
def setup
|
|
@host = ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost'
|
|
@port = ENV['MONGO_RUBY_DRIVER_PORT'] || Connection::DEFAULT_PORT
|
|
@mongo = Connection.new(@host, @port)
|
|
end
|
|
|
|
def teardown
|
|
@mongo.db('ruby-mongo-test').error
|
|
end
|
|
|
|
def test_server_info
|
|
server_info = @mongo.server_info
|
|
assert server_info.keys.include?("version")
|
|
assert_equal 1.0, server_info["ok"]
|
|
end
|
|
|
|
def test_server_version
|
|
assert_match /\d\.\d+(\.\d+)?/, @mongo.server_version.to_s
|
|
end
|
|
|
|
def test_invalid_database_names
|
|
assert_raise TypeError do @mongo.db(4) end
|
|
|
|
assert_raise InvalidName do @mongo.db('') end
|
|
assert_raise InvalidName do @mongo.db('te$t') end
|
|
assert_raise InvalidName do @mongo.db('te.t') end
|
|
assert_raise InvalidName do @mongo.db('te\\t') end
|
|
assert_raise InvalidName do @mongo.db('te/t') end
|
|
assert_raise InvalidName do @mongo.db('te st') end
|
|
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_copy_database
|
|
@mongo.db('old').collection('copy-test').insert('a' => 1)
|
|
@mongo.copy_database('old', 'new')
|
|
old_object = @mongo.db('old').collection('copy-test').find.next_document
|
|
new_object = @mongo.db('new').collection('copy-test').find.next_document
|
|
assert_equal old_object, new_object
|
|
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_logging
|
|
output = StringIO.new
|
|
logger = Logger.new(output)
|
|
logger.level = Logger::DEBUG
|
|
db = Connection.new(@host, @port, :logger => logger).db('ruby-mongo-test')
|
|
assert output.string.include?("admin.$cmd.find")
|
|
end
|
|
|
|
def test_connection_logger
|
|
output = StringIO.new
|
|
logger = Logger.new(output)
|
|
logger.level = Logger::DEBUG
|
|
connection = Connection.new(@host, @port, :logger => logger)
|
|
assert_equal logger, connection.logger
|
|
|
|
connection.logger.debug 'testing'
|
|
assert output.string.include?('testing')
|
|
end
|
|
|
|
def test_drop_database
|
|
db = @mongo.db('ruby-mongo-will-be-deleted')
|
|
coll = db.collection('temp')
|
|
coll.remove
|
|
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_nodes
|
|
db = Connection.new({:left => ['foo', 123]}, nil, :connect => false)
|
|
nodes = db.nodes
|
|
assert_equal 2, db.nodes.length
|
|
assert_equal ['foo', 123], nodes[0]
|
|
assert_equal ['localhost', Connection::DEFAULT_PORT], nodes[1]
|
|
|
|
db = Connection.new({:right => 'bar'}, nil, :connect => false)
|
|
nodes = db.nodes
|
|
assert_equal 2, nodes.length
|
|
assert_equal ['localhost', Connection::DEFAULT_PORT], nodes[0]
|
|
assert_equal ['bar', Connection::DEFAULT_PORT], nodes[1]
|
|
|
|
db = Connection.new({:right => ['foo', 123], :left => 'bar'}, nil, :connect => false)
|
|
nodes = db.nodes
|
|
assert_equal 2, nodes.length
|
|
assert_equal ['bar', Connection::DEFAULT_PORT], nodes[0]
|
|
assert_equal ['foo', 123], nodes[1]
|
|
end
|
|
end
|