mongo-ruby-driver/test/connection_test.rb

304 lines
8.9 KiB
Ruby
Raw Normal View History

2010-09-09 19:58:51 +00:00
require './test/test_helper'
require 'logger'
require 'stringio'
require 'thread'
class TestConnection < Test::Unit::TestCase
include Mongo
2010-04-05 15:07:01 +00:00
include BSON
def setup
@conn = standard_connection
end
def teardown
@conn[MONGO_TEST_DB].get_last_error
end
def test_connection_failure
assert_raise Mongo::ConnectionFailure do
Mongo::Connection.new('localhost', 27347)
end
end
def test_server_info
server_info = @conn.server_info
2009-10-26 20:41:17 +00:00
assert server_info.keys.include?("version")
assert Mongo::Support.ok?(server_info)
end
def test_ping
ping = @conn.ping
assert ping['ok']
end
2010-06-12 14:11:29 +00:00
def test_connection_uri
con = Connection.from_uri("mongodb://#{host_port}")
assert_equal mongo_host, con.primary_pool.host
assert_equal mongo_port, con.primary_pool.port
2010-06-12 14:11:29 +00:00
end
def test_server_version
assert_match /\d\.\d+(\.\d+)?/, @conn.server_version.to_s
end
def test_invalid_database_names
assert_raise TypeError do @conn.db(4) end
assert_raise Mongo::InvalidNSName do @conn.db('') end
assert_raise Mongo::InvalidNSName do @conn.db('te$t') end
assert_raise Mongo::InvalidNSName do @conn.db('te.t') end
assert_raise Mongo::InvalidNSName do @conn.db('te\\t') end
assert_raise Mongo::InvalidNSName do @conn.db('te/t') end
assert_raise Mongo::InvalidNSName do @conn.db('te st') end
end
def test_options_passed_to_db
@pk_mock = Object.new
db = @conn.db('test', :pk => @pk_mock, :strict => true)
assert_equal @pk_mock, db.pk_factory
assert db.strict?
end
def test_database_info
@conn.drop_database(MONGO_TEST_DB)
@conn.db(MONGO_TEST_DB).collection('info-test').insert('a' => 1)
info = @conn.database_info
assert_not_nil info
assert_kind_of Hash, info
2010-04-05 19:48:35 +00:00
assert_not_nil info[MONGO_TEST_DB]
assert info[MONGO_TEST_DB] > 0
@conn.drop_database(MONGO_TEST_DB)
end
2009-11-04 16:57:03 +00:00
def test_copy_database
@conn.db('old').collection('copy-test').insert('a' => 1)
@conn.copy_database('old', 'new', host_port)
old_object = @conn.db('old').collection('copy-test').find.next_document
new_object = @conn.db('new').collection('copy-test').find.next_document
2009-11-04 16:57:03 +00:00
assert_equal old_object, new_object
@conn.drop_database('old')
@conn.drop_database('new')
end
def test_copy_database_with_auth
@conn.db('old').collection('copy-test').insert('a' => 1)
@conn.db('old').add_user('bob', 'secret')
assert_raise Mongo::OperationFailure do
@conn.copy_database('old', 'new', host_port, 'bob', 'badpassword')
end
result = @conn.copy_database('old', 'new', host_port, 'bob', 'secret')
assert Mongo::Support.ok?(result)
@conn.drop_database('old')
@conn.drop_database('new')
2009-11-04 16:57:03 +00:00
end
def test_database_names
@conn.drop_database(MONGO_TEST_DB)
@conn.db(MONGO_TEST_DB).collection('info-test').insert('a' => 1)
names = @conn.database_names
assert_not_nil names
assert_kind_of Array, names
assert names.length >= 1
2010-04-05 19:48:35 +00:00
assert names.include?(MONGO_TEST_DB)
end
def test_logging
output = StringIO.new
logger = Logger.new(output)
logger.level = Logger::DEBUG
connection = standard_connection(:logger => logger).db(MONGO_TEST_DB)
assert output.string.include?("admin['$cmd'].find")
end
2010-02-25 19:58:32 +00:00
def test_connection_logger
output = StringIO.new
logger = Logger.new(output)
logger.level = Logger::DEBUG
connection = standard_connection(:logger => logger)
assert_equal logger, connection.logger
connection.logger.debug 'testing'
assert output.string.include?('testing')
end
2009-01-20 14:21:19 +00:00
def test_drop_database
db = @conn.db('ruby-mongo-will-be-deleted')
2009-01-20 14:21:19 +00:00
coll = db.collection('temp')
coll.remove
2009-01-20 14:21:19 +00:00
coll.insert(:name => 'temp')
assert_equal 1, coll.count()
assert @conn.database_names.include?('ruby-mongo-will-be-deleted')
2009-01-20 14:21:19 +00:00
@conn.drop_database('ruby-mongo-will-be-deleted')
assert !@conn.database_names.include?('ruby-mongo-will-be-deleted')
2009-01-20 14:21:19 +00:00
end
def test_nodes
db = Connection.multi([['foo', 27017], ['bar', 27018]], :connect => false)
nodes = db.nodes
assert_equal 2, nodes.length
assert_equal ['foo', 27017], nodes[0]
assert_equal ['bar', 27018], nodes[1]
end
2010-09-28 16:15:45 +00:00
def test_fsync_lock
assert !@conn.locked?
@conn.lock!
assert @conn.locked?
assert_equal 1, @conn['admin']['$cmd.sys.inprog'].find_one['fsyncLock'], "Not fsync-locked"
assert_equal "unlock requested", @conn.unlock!['info']
2010-09-28 16:15:45 +00:00
unlocked = false
counter = 0
while counter < 5
if @conn['admin']['$cmd.sys.inprog'].find_one['fsyncLock'].nil?
2010-09-28 16:15:45 +00:00
unlocked = true
break
else
sleep(1)
counter += 1
end
end
assert !@conn.locked?
2010-09-28 16:15:45 +00:00
assert unlocked, "mongod failed to unlock"
end
def test_max_bson_size_value
conn = standard_connection
if conn.server_version > "1.7.2"
assert_equal conn['admin'].command({:ismaster => 1})['maxBsonObjectSize'], conn.max_bson_size
end
conn.connect
assert_equal BSON::BSON_CODER.max_bson_size, conn.max_bson_size
doc = {'n' => 'a' * (BSON_CODER.max_bson_size - 11)}
assert_raise InvalidDocument do
assert BSON::BSON_CODER.serialize(doc)
end
limit = 7 * 1024 * 1024
conn.stubs(:max_bson_size).returns(limit)
conn.connect
assert_equal limit, conn.max_bson_size
assert_equal limit, BSON::BSON_CODER.max_bson_size
doc = {'n' => 'a' * ((limit) - 11)}
assert_raise_error InvalidDocument, "limited to #{limit}" do
assert BSON::BSON_CODER.serialize(doc)
end
end
def test_max_bson_size_with_old_mongod
conn = standard_connection(:connect => false)
admin_db = Object.new
admin_db.expects(:command).returns({'ok' => 1, 'ismaster' => 1}).twice
conn.expects(:[]).with('admin').returns(admin_db).twice
conn.connect
assert_equal Mongo::DEFAULT_MAX_BSON_SIZE, BSON::BSON_CODER.max_bson_size
end
def test_connection_activity
conn = standard_connection
assert conn.active?
conn.primary_pool.close
assert !conn.active?
# Simulate a dropped connection.
dropped_socket = Mocha::Mock.new
dropped_socket.stubs(:read).raises(Errno::ECONNRESET)
dropped_socket.stubs(:send).raises(Errno::ECONNRESET)
dropped_socket.stub_everything
conn.primary_pool.host = 'localhost'
conn.primary_pool.port = Mongo::Connection::DEFAULT_PORT
conn.primary_pool.instance_variable_set("@sockets", [dropped_socket])
assert !conn.active?
end
2010-02-25 19:58:32 +00:00
context "Saved authentications" do
setup do
@conn = standard_connection
2010-02-25 19:58:32 +00:00
@auth = {'db_name' => 'test', 'username' => 'bob', 'password' => 'secret'}
@conn.add_auth(@auth['db_name'], @auth['username'], @auth['password'])
end
should "save the authentication" do
assert_equal @auth, @conn.auths[0]
end
should "replace the auth if given a new auth for the same db" do
auth = {'db_name' => 'test', 'username' => 'mickey', 'password' => 'm0u53'}
@conn.add_auth(auth['db_name'], auth['username'], auth['password'])
assert_equal 1, @conn.auths.length
assert_equal auth, @conn.auths[0]
end
should "remove auths by database" do
@conn.remove_auth('non-existent database')
assert_equal 1, @conn.auths.length
@conn.remove_auth('test')
assert_equal 0, @conn.auths.length
end
should "remove all auths" do
@conn.clear_auths
assert_equal 0, @conn.auths.length
end
end
context "Connection exceptions" do
setup do
@con = standard_connection(:pool_size => 10, :timeout => 10)
@coll = @con[MONGO_TEST_DB]['test-connection-exceptions']
end
should "release connection if an exception is raised on send_message" do
@con.stubs(:send_message_on_socket).raises(ConnectionFailure)
assert_equal 0, @con.primary_pool.checked_out.size
assert_raise ConnectionFailure do
@coll.insert({:test => "insert"})
end
assert_equal 0, @con.primary_pool.checked_out.size
end
should "release connection if an exception is raised on send_with_safe_check" do
@con.stubs(:receive).raises(ConnectionFailure)
assert_equal 0, @con.primary_pool.checked_out.size
assert_raise ConnectionFailure do
@coll.insert({:test => "insert"}, :safe => true)
end
assert_equal 0, @con.primary_pool.checked_out.size
end
should "release connection if an exception is raised on receive_message" do
@con.stubs(:receive).raises(ConnectionFailure)
assert_equal 0, @con.primary_pool.checked_out.size
assert_raise ConnectionFailure do
@coll.find.to_a
end
assert_equal 0, @con.primary_pool.checked_out.size
end
should "show a proper exception message if an IOError is raised while closing a socket" do
fake_socket = Mocha::Mock.new
fake_socket.stubs(:close).raises(IOError.new)
fake_socket.stub_everything
TCPSocket.expects(:new).returns(fake_socket)
@con.primary_pool.checkout_new_socket
assert_equal [], @con.primary_pool.close
end
end
end