jmongo 1.0.3 → 1.1.0

data/test/connection_test.rb

@@ -0,0 +1,325 @@
+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.close
+  end
+
+  def test_connection_failure
+    assert_raise Mongo::ConnectionFailure do
+      Mongo::Connection.new('localhost', 27347)
+    end
+  end
+
+  def test_connection_timeout
+    passed = false
+    begin
+      t0 = Time.now
+      Mongo::Connection.new('192.169.169.1', 27017, :connect_timeout => 3)
+    rescue OperationTimeout
+      passed = true
+      t1 = Time.now
+    end
+
+    assert passed
+    assert t1 - t0 < 4
+  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
+    conn = Connection.multi([['foo', 27017], ['bar', 27018]], :connect => false)
+    nodes = conn.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(:connect => false)
+
+    admin_db = Object.new
+    admin_db.expects(:command).returns({'ok' => 1, 'ismaster' => 1, 'maxBsonObjectSize' => 15_000_000})
+    conn.expects(:[]).with('admin').returns(admin_db)
+    conn.connect
+    assert_equal 15_000_000, conn.max_bson_size
+
+    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
+    doc = {'n' => 'a' * (conn.max_bson_size)}
+    assert_raise InvalidDocument do
+      assert BSON::BSON_CODER.serialize(doc, false, true, @conn.max_bson_size)
+    end
+  end
+
+  def test_max_bson_size_with_no_reported_max_size
+    conn = standard_connection(:connect => false)
+
+    admin_db = Object.new
+    admin_db.expects(:command).returns({'ok' => 1, 'ismaster' => 1})
+    conn.expects(:[]).with('admin').returns(admin_db)
+
+    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
+
+    teardown do
+      @conn.clear_auths
+    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.stubs(:new).returns(fake_socket)
+
+      @con.primary_pool.checkout_new_socket
+      assert_equal [], @con.primary_pool.close
+    end
+  end
+end

data/test/conversions_test.rb

@@ -0,0 +1,121 @@
+__END__
+
+require './test/test_helper'
+require 'mongo/exceptions'
+require 'mongo/util/conversions'
+
+class ConversionsTest < Test::Unit::TestCase
+  include Mongo::Conversions
+
+  def test_array_as_sort_parameters_with_array_of_key_and_value
+    params = array_as_sort_parameters(["field1", "asc"])
+    assert_equal({"field1" => 1}, params)
+  end
+
+  def test_array_as_sort_parameters_with_array_of_string_and_values
+    params = array_as_sort_parameters([["field1", :asc], ["field2", :desc]])
+    assert_equal({ "field1" => 1, "field2" => -1 }, params)
+  end
+
+  def test_string_as_sort_parameters_with_string
+    params = string_as_sort_parameters("field")
+    assert_equal({ "field" => 1 }, params)
+  end
+
+  def test_string_as_sort_parameters_with_empty_string
+    params = string_as_sort_parameters("")
+    assert_equal({}, params)
+  end
+
+  def test_symbol_as_sort_parameters
+    params = string_as_sort_parameters(:field)
+    assert_equal({ "field" => 1 }, params)
+  end
+
+  def test_sort_value_when_value_is_one
+    assert_equal 1, sort_value(1)
+  end
+
+  def test_sort_value_when_value_is_one_as_a_string
+    assert_equal 1, sort_value("1")
+  end
+
+  def test_sort_value_when_value_is_negative_one
+    assert_equal(-1, sort_value(-1))
+  end
+
+  def test_sort_value_when_value_is_negative_one_as_a_string
+    assert_equal(-1, sort_value("-1"))
+  end
+
+  def test_sort_value_when_value_is_ascending
+    assert_equal 1, sort_value("ascending")
+  end
+
+  def test_sort_value_when_value_is_asc
+    assert_equal 1, sort_value("asc")
+  end
+
+  def test_sort_value_when_value_is_uppercase_ascending
+    assert_equal 1, sort_value("ASCENDING")
+  end
+
+  def test_sort_value_when_value_is_uppercase_asc
+    assert_equal 1, sort_value("ASC")
+  end
+
+  def test_sort_value_when_value_is_symbol_ascending
+    assert_equal 1, sort_value(:ascending)
+  end
+
+  def test_sort_value_when_value_is_symbol_asc
+    assert_equal 1, sort_value(:asc)
+  end
+
+  def test_sort_value_when_value_is_symbol_uppercase_ascending
+    assert_equal 1, sort_value(:ASCENDING)
+  end
+
+  def test_sort_value_when_value_is_symbol_uppercase_asc
+    assert_equal 1, sort_value(:ASC)
+  end
+
+  def test_sort_value_when_value_is_descending
+    assert_equal(-1, sort_value("descending"))
+  end
+
+  def test_sort_value_when_value_is_desc
+    assert_equal(-1, sort_value("desc"))
+  end
+
+  def test_sort_value_when_value_is_uppercase_descending
+    assert_equal(-1, sort_value("DESCENDING"))
+  end
+
+  def test_sort_value_when_value_is_uppercase_desc
+    assert_equal(-1, sort_value("DESC"))
+  end
+
+  def test_sort_value_when_value_is_symbol_descending
+    assert_equal(-1, sort_value(:descending))
+  end
+
+  def test_sort_value_when_value_is_symbol_desc
+    assert_equal(-1, sort_value(:desc))
+  end
+
+  def test_sort_value_when_value_is_uppercase_symbol_descending
+    assert_equal(-1, sort_value(:DESCENDING))
+  end
+
+  def test_sort_value_when_value_is_uppercase_symbol_desc
+    assert_equal(-1, sort_value(:DESC))
+  end
+
+  def test_sort_value_when_value_is_invalid
+    assert_raise Mongo::InvalidSortValueError do
+      sort_value(2)
+    end
+  end
+
+end

data/test/cursor_fail_test.rb

@@ -0,0 +1,75 @@
+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