mongo 1.3.0 → 1.12.5

data/test/connection_test.rb

@@ -1,309 +0,0 @@
-require './test/test_helper'
-require 'logger'
-require 'stringio'
-require 'thread'
-
-class TestConnection < Test::Unit::TestCase
-
-  include Mongo
-  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_host_port_accessors
-    assert_equal @conn.host, TEST_HOST
-    assert_equal @conn.port, TEST_PORT
-  end
-
-  def test_server_info
-    server_info = @conn.server_info
-    assert server_info.keys.include?("version")
-    assert Mongo::Support.ok?(server_info)
-  end
-
-  def test_ping
-    ping = @conn.ping
-    assert ping['ok']
-  end
-
-  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
-  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
-    assert_not_nil info[MONGO_TEST_DB]
-    assert info[MONGO_TEST_DB] > 0
-
-    @conn.drop_database(MONGO_TEST_DB)
-  end
-
-  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
-    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')
-  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
-    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
-
-  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
-
-  def test_drop_database
-    db = @conn.db('ruby-mongo-will-be-deleted')
-    coll = db.collection('temp')
-    coll.remove
-    coll.insert(:name => 'temp')
-    assert_equal 1, coll.count()
-    assert @conn.database_names.include?('ruby-mongo-will-be-deleted')
-
-    @conn.drop_database('ruby-mongo-will-be-deleted')
-    assert !@conn.database_names.include?('ruby-mongo-will-be-deleted')
-  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
-
-  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_match(/unlock/, @conn.unlock!['info'])
-    unlocked = false
-    counter  = 0
-    while counter < 5
-      if @conn['admin']['$cmd.sys.inprog'].find_one['fsyncLock'].nil?
-        unlocked = true
-        break
-      else
-        sleep(1)
-        counter += 1
-      end
-    end
-    assert !@conn.locked?
-    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("@pids", {dropped_socket => Process.pid})
-    conn.primary_pool.instance_variable_set("@sockets", [dropped_socket])
-
-    assert !conn.active?
-  end
-
-  context "Saved authentications" do
-    setup do
-      @conn = standard_connection
-      @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

data/test/cursor_fail_test.rb

@@ -1,75 +0,0 @@
-require './test/test_helper'
-require 'logger'
-
-class CursorFailTest < Test::Unit::TestCase
-
-  include Mongo
-
-  @@connection = standard_connection
-  @@db   = @@connection.db(MONGO_TEST_DB)
-  @@coll = @@db.collection('test')
-  @@version = @@connection.server_version
-
-  def setup
-    @@coll.remove
-    @@coll.insert('a' => 1)     # collection not created until it's used
-    @@coll_full_name = "#{MONGO_TEST_DB}.test"
-  end
-
-  def test_refill_via_get_more
-    assert_equal 1, @@coll.count
-    1000.times { |i|
-      assert_equal 1 + i, @@coll.count
-      @@coll.insert('a' => i)
-    }
-
-    assert_equal 1001, @@coll.count
-    count = 0
-    @@coll.find.each { |obj|
-      count += obj['a']
-    }
-    assert_equal 1001, @@coll.count
-
-    # do the same thing again for debugging
-    assert_equal 1001, @@coll.count
-    count2 = 0
-    @@coll.find.each { |obj|
-      count2 += obj['a']
-    }
-    assert_equal 1001, @@coll.count
-
-    assert_equal count, count2
-    assert_equal 499501, count
-  end
-
-  def test_refill_via_get_more_alt_coll
-    coll = @@db.collection('test-alt-coll')
-    coll.remove
-    coll.insert('a' => 1)     # collection not created until it's used
-    assert_equal 1, coll.count
-
-    1000.times { |i|
-      assert_equal 1 + i, coll.count
-      coll.insert('a' => i)
-    }
-
-    assert_equal 1001, coll.count
-    count = 0
-    coll.find.each { |obj|
-      count += obj['a']
-    }
-    assert_equal 1001, coll.count
-
-    # do the same thing again for debugging
-    assert_equal 1001, coll.count
-    count2 = 0
-    coll.find.each { |obj|
-      count2 += obj['a']
-    }
-    assert_equal 1001, coll.count
-
-    assert_equal count, count2
-    assert_equal 499501, count
-  end
-
-end

data/test/cursor_message_test.rb

@@ -1,43 +0,0 @@
-require './test/test_helper'
-require 'logger'
-
-class CursorMessageTest < Test::Unit::TestCase
-
-  include Mongo
-
-  @@connection = standard_connection
-  @@db   = @@connection.db(MONGO_TEST_DB)
-  @@coll = @@db.collection('test')
-  @@version = @@connection.server_version
-
-  def setup
-    @@coll.remove
-    @@coll.insert('a' => 1)     # collection not created until it's used
-    @@coll_full_name = "#{MONGO_TEST_DB}.test"
-  end
-
-  def test_valid_batch_sizes
-    assert_raise ArgumentError do
-      @@coll.find({}, :batch_size => 1, :limit => 5)
-    end
-
-    assert_raise ArgumentError do
-      @@coll.find({}, :batch_size => -1, :limit => 5)
-    end
-
-    assert @@coll.find({}, :batch_size => 0, :limit => 5)
-  end
-
-  def test_batch_size
-    @@coll.remove
-    200.times do |n|
-      @@coll.insert({:a => n})
-    end
-
-    list = @@coll.find({}, :batch_size => 2, :limit => 6).to_a
-    assert_equal 6, list.length
-
-    list = @@coll.find({}, :batch_size => 100, :limit => 101).to_a
-    assert_equal 101, list.length
-  end
-end