mongo 1.3.0 → 1.12.5

data/test/tools/mongo_config_test.rb

@@ -0,0 +1,160 @@
+# Copyright (C) 2009-2013 MongoDB, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'test_helper'
+
+class MongoConfig < Test::Unit::TestCase
+
+  def startup
+    @sys_proc = nil
+  end
+
+  def shutdown
+    @sys_proc.stop if @sys_proc && @sys_proc.running?
+  end
+
+  test "config defaults" do
+    [ Mongo::Config::DEFAULT_BASE_OPTS,
+      Mongo::Config::DEFAULT_REPLICA_SET,
+      Mongo::Config::DEFAULT_SHARDED_SIMPLE,
+      Mongo::Config::DEFAULT_SHARDED_REPLICA
+    ].each do |params|
+      config = Mongo::Config.cluster(params)
+      assert(config.size > 0)
+    end
+  end
+
+  test "get available port" do
+    assert_not_nil(Mongo::Config.get_available_port)
+  end
+
+  test "SysProc start" do
+    cmd = "true"
+    @sys_proc = Mongo::Config::SysProc.new(cmd)
+    assert_equal(cmd, @sys_proc.cmd)
+    assert_nil(@sys_proc.pid)
+    start_and_assert_running?(@sys_proc)
+  end
+
+  test "SysProc wait" do
+    @sys_proc = Mongo::Config::SysProc.new("true")
+    start_and_assert_running?(@sys_proc)
+    assert(@sys_proc.running?)
+    @sys_proc.wait
+    assert(!@sys_proc.running?)
+  end
+
+  test "SysProc kill" do
+    @sys_proc = Mongo::Config::SysProc.new("true")
+    start_and_assert_running?(@sys_proc)
+    @sys_proc.kill
+    @sys_proc.wait
+    assert(!@sys_proc.running?)
+  end
+
+  test "SysProc stop" do
+    @sys_proc = Mongo::Config::SysProc.new("true")
+    start_and_assert_running?(@sys_proc)
+    @sys_proc.stop
+    assert(!@sys_proc.running?)
+  end
+
+  test "SysProc zombie respawn" do
+    @sys_proc = Mongo::Config::SysProc.new("true")
+    start_and_assert_running?(@sys_proc)
+    prev_pid = @sys_proc.pid
+    @sys_proc.kill
+    # don't wait, leaving a zombie
+    assert(@sys_proc.running?)
+    start_and_assert_running?(@sys_proc)
+    assert(prev_pid && @sys_proc.pid && prev_pid != @sys_proc.pid, 'SysProc#start should spawn a new process after a zombie')
+    @sys_proc.stop
+    assert(!@sys_proc.running?)
+  end
+
+  test "Server" do
+    server = Mongo::Config::Server.new('a cmd', 'host', 1234)
+    assert_equal('a cmd', server.cmd)
+    assert_equal('host', server.host)
+    assert_equal(1234, server.port)
+  end
+
+  test "DbServer" do
+    config = Mongo::Config::DEFAULT_BASE_OPTS
+    server = Mongo::Config::DbServer.new(config)
+    assert_equal(config, server.config)
+    assert_equal("mongod --dbpath data --logpath data/log", server.cmd)
+    assert_equal(config[:host], server.host)
+    assert_equal(config[:port], server.port)
+  end
+
+  def cluster_test(opts)
+    #debug 1, opts.inspect
+    config =  Mongo::Config.cluster(opts)
+    #debug 1, config.inspect
+    manager = Mongo::Config::ClusterManager.new(config)
+    assert_equal(config, manager.config)
+    manager.start
+    yield manager
+    manager.stop
+    manager.servers.each{|s| assert(!s.running?)}
+    manager.clobber
+  end
+
+  test "cluster manager base" do
+    cluster_test(Mongo::Config::DEFAULT_BASE_OPTS) do |manager|
+
+    end
+  end
+
+  test "cluster manager replica set" do
+    cluster_test(Mongo::Config::DEFAULT_REPLICA_SET) do |manager|
+      servers = manager.servers
+      servers.each do |server|
+        assert_not_nil(Mongo::MongoClient.new(server.host, server.port))
+        assert_match(/oplogSize/, server.cmd, '--oplogSize option should be specified')
+        assert_match(/smallfiles/, server.cmd, '--smallfiles option should be specified')
+        assert_no_match(/nojournal/, server.cmd, '--nojournal option should not be specified')
+        assert_match(/noprealloc/, server.cmd, '--noprealloc option should be specified')
+      end
+    end
+  end
+
+  test "cluster manager sharded simple" do
+    cluster_test(Mongo::Config::DEFAULT_SHARDED_SIMPLE) do |manager|
+      servers = manager.shards + manager.configs
+      servers.each do |server|
+        assert_not_nil(Mongo::MongoClient.new(server.host, server.port))
+        assert_match(/oplogSize/, server.cmd, '--oplogSize option should be specified')
+        assert_match(/smallfiles/, server.cmd, '--smallfiles option should be specified')
+        assert_no_match(/nojournal/, server.cmd, '--nojournal option should not be specified')
+        assert_match(/noprealloc/, server.cmd, '--noprealloc option should be specified')
+      end
+    end
+  end
+
+  test "cluster manager sharded replica" do
+    #cluster_test(Mongo::Config::DEFAULT_SHARDED_REPLICA) # not yet supported by ClusterManager
+  end
+
+  private
+
+  def start_and_assert_running?(sys_proc)
+    assert_not_nil(sys_proc.start(0))
+    assert_not_nil(sys_proc.pid)
+    assert(sys_proc.running?)
+  end
+
+end
+

data/test/unit/client_test.rb

@@ -0,0 +1,381 @@
+# Copyright (C) 2009-2013 MongoDB, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'test_helper'
+include Mongo
+
+class ClientUnitTest < Test::Unit::TestCase
+  context "Mongo::MongoClient initialization " do
+    context "given a single node" do
+      setup do
+        @client = MongoClient.new('localhost', 27017, :connect => false)
+        TCPSocket.stubs(:new).returns(new_mock_socket)
+
+        admin_db = new_mock_db
+        admin_db.expects(:command).returns({'ok' => 1, 'ismaster' => 1})
+        @client.expects(:[]).with('admin').returns(admin_db)
+        @client.connect
+      end
+
+      should "gle writes by default" do
+        assert_equal 1, @client.write_concern[:w]
+      end
+
+      should "set localhost and port to master" do
+        assert_equal 'localhost', @client.primary_pool.host
+        assert_equal 27017, @client.primary_pool.port
+      end
+
+      should "set connection pool to 1" do
+        assert_equal 1, @client.primary_pool.size
+      end
+
+      should "set op timeout to default" do
+        assert_equal Mongo::MongoClient::DEFAULT_OP_TIMEOUT, @client.op_timeout
+      end
+
+      should "default slave_ok to false" do
+        assert !@client.slave_ok?
+      end
+
+      should "not raise error if no host or port is supplied" do
+        assert_nothing_raised do
+          MongoClient.new(:w => 1, :connect => false)
+        end
+        assert_nothing_raised do
+          MongoClient.new('localhost', :w => 1, :connect=> false)
+        end
+      end
+
+      should "warn if invalid options are specified" do
+        client = MongoClient.allocate
+        opts = {:connect => false}
+
+        MongoReplicaSetClient::REPL_SET_OPTS.each do |opt|
+          client.expects(:warn).with("#{opt} is not a valid option for #{client.class}")
+          opts[opt] = true
+        end
+
+        args = ['localhost', 27017, opts]
+        client.send(:initialize, *args)
+      end
+
+      context 'specifying nil op timeout explicitly' do
+        setup do
+          @client = MongoClient.new('localhost', 27017, :connect => false, :op_timeout => nil)
+        end
+
+        should 'set op timeout to nil' do
+          assert_equal nil, @client.op_timeout
+        end
+      end
+
+      context 'specifying a different op timeout than default' do
+        setup do
+          @client = MongoClient.new('localhost', 27017, :connect => false, :op_timeout => 50)
+        end
+
+        should 'set op timeout to the specified value' do
+          assert_equal 50, @client.op_timeout
+        end
+      end
+
+      context "given a replica set" do
+
+        should "warn if invalid options are specified" do
+          client = MongoReplicaSetClient.allocate
+          opts = {:connect => false}
+
+          MongoClient::CLIENT_ONLY_OPTS.each do |opt|
+            client.expects(:warn).with("#{opt} is not a valid option for #{client.class}")
+            opts[opt] = true
+          end
+
+          args = [['localhost:27017'], opts]
+          client.send(:initialize, *args)
+        end
+
+        should "throw error if superflous arguments are specified" do
+          assert_raise MongoArgumentError do
+            MongoReplicaSetClient.new(['localhost:27017'], ['localhost:27018'], {:connect => false})
+          end
+        end
+      end
+    end
+
+    context "initializing with a unix socket" do
+      setup do
+        @client = MongoClient.new('/tmp/mongod.sock', :connect => false)
+        UNIXSocket.stubs(:new).returns(new_mock_unix_socket)
+      end
+      should "parse a unix socket" do
+        assert_equal "/tmp/mongod.sock", @client.host_port.first
+      end
+    end
+
+    context "initializing with a unix socket in uri" do
+      setup do
+          @client = MongoClient.from_uri("mongodb:///tmp/mongod.sock", :connect => false)
+          UNIXSocket.stubs(:new).returns(new_mock_unix_socket)
+      end
+      should "parse a unix socket" do
+          assert_equal "/tmp/mongod.sock", @client.host_port.first
+      end
+    end
+
+    context "initializing with a mongodb uri" do
+      should "parse a simple uri" do
+        @client = MongoClient.from_uri("mongodb://localhost", :connect => false)
+        assert_equal ['localhost', 27017], @client.host_port
+      end
+
+      should "set auth source" do
+        @client = MongoClient.from_uri("mongodb://user:pass@localhost?authSource=foo", :connect => false)
+        assert_equal 'foo', @client.auths.first[:source]
+      end
+
+      should "set auth mechanism" do
+        @client = MongoClient.from_uri("mongodb://user@localhost?authMechanism=MONGODB-X509", :connect => false)
+        assert_equal 'MONGODB-X509', @client.auths.first[:mechanism]
+
+        assert_raise MongoArgumentError do
+          MongoClient.from_uri("mongodb://user@localhost?authMechanism=INVALID", :connect => false)
+        end
+      end
+
+      should "allow a complex host names" do
+        host_name = "foo.bar-12345.org"
+        @client = MongoClient.from_uri("mongodb://#{host_name}", :connect => false)
+        assert_equal [host_name, 27017], @client.host_port
+      end
+
+      should "allow db without username and password" do
+        host_name = "foo.bar-12345.org"
+        @client = MongoClient.from_uri("mongodb://#{host_name}/foo", :connect => false)
+        assert_equal [host_name, 27017], @client.host_port
+      end
+
+      should "set write concern options on connection" do
+        host_name = "localhost"
+        opts = "w=2&wtimeoutMS=1000&fsync=true&journal=true"
+        @client = MongoClient.from_uri("mongodb://#{host_name}/foo?#{opts}", :connect => false)
+        assert_equal({:w => 2, :wtimeout => 1000, :fsync => true, :j => true}, @client.write_concern)
+      end
+
+      should "set timeout options on connection" do
+        host_name = "localhost"
+        opts = "connectTimeoutMS=1000&socketTimeoutMS=5000"
+        @client = MongoClient.from_uri("mongodb://#{host_name}/foo?#{opts}", :connect => false)
+        assert_equal 1, @client.connect_timeout
+        assert_equal 5, @client.op_timeout
+      end
+
+      should "parse a uri with a hyphen & underscore in the username or password" do
+        @client = MongoClient.from_uri("mongodb://hyphen-user_name:p-s_s@localhost:27017/db", :connect => false)
+        assert_equal ['localhost', 27017], @client.host_port
+
+        auth_hash = {
+          :db_name   => 'db',
+          :extra=>{},
+          :username  => 'hyphen-user_name',
+          :password  => 'p-s_s',
+          :source    => 'db',
+          :mechanism => nil
+        }
+        assert_equal auth_hash, @client.auths.first
+      end
+
+      should "attempt to connect" do
+        TCPSocket.stubs(:new).returns(new_mock_socket)
+        @client = MongoClient.from_uri("mongodb://localhost", :connect => false)
+
+        admin_db = new_mock_db
+        admin_db.expects(:command).returns({'ok' => 1, 'ismaster' => 1})
+        @client.expects(:[]).with('admin').returns(admin_db)
+        @client.connect
+      end
+
+      should "raise an error on invalid uris" do
+        assert_raise MongoArgumentError do
+          MongoClient.from_uri("mongo://localhost", :connect => false)
+        end
+
+        assert_raise MongoArgumentError do
+          MongoClient.from_uri("mongodb://localhost:abc", :connect => false)
+        end
+      end
+
+      should "require password if using legacy auth and username present" do
+        assert MongoClient.from_uri("mongodb://kyle:jones@localhost/db", :connect => false)
+
+        assert_raise MongoArgumentError do
+          MongoClient.from_uri("mongodb://kyle:@localhost", :connect => false)
+        end
+
+        assert_raise MongoArgumentError do
+          MongoClient.from_uri("mongodb://kyle@localhost", :connect => false)
+        end
+      end
+    end
+
+    context "initializing with ENV['MONGODB_URI']" do
+      should "parse a simple uri" do
+        uri = "mongodb://localhost?connect=false"
+        with_preserved_env_uri(uri) do
+          @client = MongoClient.new
+          assert_equal ['localhost', 27017], @client.host_port
+        end
+      end
+
+      should "set auth source" do
+        uri = "mongodb://user:pass@localhost?authSource=foo&connect=false"
+        with_preserved_env_uri(uri) do
+          @client = MongoClient.new
+          assert_equal 'foo', @client.auths.first[:source]
+        end
+      end
+
+      should "set auth mechanism" do
+        uri = "mongodb://user@localhost?authMechanism=MONGODB-X509&connect=false"
+        with_preserved_env_uri(uri) do
+          @client = MongoClient.new
+          assert_equal 'MONGODB-X509', @client.auths.first[:mechanism]
+
+          ENV['MONGODB_URI'] = "mongodb://user@localhost?authMechanism=INVALID&connect=false"
+          assert_raise MongoArgumentError do
+            MongoClient.new
+          end
+        end
+      end
+
+      should "allow a complex host names" do
+        host_name = "foo.bar-12345.org"
+        uri = "mongodb://#{host_name}?connect=false"
+        with_preserved_env_uri(uri) do
+          @client = MongoClient.new
+          assert_equal [host_name, 27017], @client.host_port
+        end
+      end
+
+      should "allow db without username and password" do
+        host_name = "foo.bar-12345.org"
+        uri = "mongodb://#{host_name}/foo?connect=false"
+        with_preserved_env_uri(uri) do
+          @client = MongoClient.new
+          assert_equal [host_name, 27017], @client.host_port
+        end
+      end
+
+      should "set write concern options on connection" do
+        host_name = "localhost"
+        opts = "w=2&wtimeoutMS=1000&fsync=true&journal=true&connect=false"
+        uri = "mongodb://#{host_name}/foo?#{opts}"
+        with_preserved_env_uri(uri) do
+          @client = MongoClient.new
+          assert_equal({:w => 2, :wtimeout => 1000, :fsync => true, :j => true}, @client.write_concern)
+        end
+      end
+
+      should "set timeout options on connection" do
+        host_name = "localhost"
+        opts = "connectTimeoutMS=1000&socketTimeoutMS=5000&connect=false"
+        uri = "mongodb://#{host_name}/foo?#{opts}"
+        with_preserved_env_uri(uri) do
+          @client = MongoClient.new
+          assert_equal 1, @client.connect_timeout
+          assert_equal 5, @client.op_timeout
+        end
+      end
+
+      should "parse a uri with a hyphen & underscore in the username or password" do
+        uri = "mongodb://hyphen-user_name:p-s_s@localhost:27017/db?connect=false"
+        with_preserved_env_uri(uri) do
+          @client = MongoClient.new
+          assert_equal ['localhost', 27017], @client.host_port
+
+          auth_hash = {
+            :db_name   => 'db',
+            :extra=>{},
+            :username  => 'hyphen-user_name',
+            :password  => 'p-s_s',
+            :source    => 'db',
+            :mechanism => nil
+          }
+          assert_equal auth_hash, @client.auths.first
+        end
+      end
+
+      should "attempt to connect" do
+        TCPSocket.stubs(:new).returns(new_mock_socket)
+        uri = "mongodb://localhost?connect=false"
+        with_preserved_env_uri(uri) do
+          @client = MongoClient.new
+
+          admin_db = new_mock_db
+          admin_db.expects(:command).returns({'ok' => 1, 'ismaster' => 1})
+          @client.expects(:[]).with('admin').returns(admin_db)
+          @client.connect
+        end
+      end
+
+      should "raise an error on invalid uris" do
+        uri = "mongo://localhost"
+        with_preserved_env_uri(uri) do
+          assert_raise MongoArgumentError do
+            MongoClient.new
+          end
+
+          ENV['MONGODB_URI'] = "mongodb://localhost:abc?connect=false"
+          assert_raise MongoArgumentError do
+            MongoClient.new
+          end
+        end
+      end
+
+      should "require password if using legacy auth and username present" do
+        uri = "mongodb://kyle:jones@localhost?connect=false"
+        with_preserved_env_uri(uri) do
+          assert MongoClient.new
+
+          ENV['MONGODB_URI'] = "mongodb://kyle:@localhost?connect=false"
+          assert_raise MongoArgumentError do
+            MongoClient.new
+          end
+
+          ENV['MONGODB_URI'] = "mongodb://kyle@localhost?connect=false"
+          assert_raise MongoArgumentError do
+            MongoClient.new
+          end
+        end
+      end
+
+      should "require password if using PLAIN auth and username present" do
+        uri = "mongodb://kyle:jones@localhost?connect=false&authMechanism=PLAIN"
+        with_preserved_env_uri(uri) do
+          assert MongoClient.new
+
+          ENV['MONGODB_URI'] = "mongodb://kyle:@localhost?connect=false&authMechanism=PLAIN"
+          assert_raise MongoArgumentError do
+            MongoClient.new
+          end
+
+          ENV['MONGODB_URI'] = "mongodb://kyle@localhost?connect=false&authMechanism=PLAIN"
+          assert_raise MongoArgumentError do
+            MongoClient.new
+          end
+        end
+      end
+    end
+  end
+end