mongo 1.12.5 → 2.0.0

data/test/replica_set/basic_test.rb

@@ -1,189 +0,0 @@
-# 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 ReplicaSetBasicTest < Test::Unit::TestCase
-
-  def setup
-    ensure_cluster(:rs)
-  end
-
-  def test_connect
-    client = MongoReplicaSetClient.new(@rs.repl_set_seeds, :name => @rs.repl_set_name, :op_timeout => TEST_OP_TIMEOUT)
-    assert client.connected?
-    assert_equal @rs.primary_name, client.primary.join(':')
-    assert_equal @rs.secondary_names.sort, client.secondaries.collect{|s| s.join(':')}.sort
-    assert_equal @rs.arbiter_names.sort, client.arbiters.collect{|s| s.join(':')}.sort
-    client.close
-
-    silently do
-      client = MongoReplicaSetClient.new(@rs.repl_set_seeds_old, :name => @rs.repl_set_name)
-    end
-
-    assert client.connected?
-    client.close
-  end
-
-  def test_safe_option
-    client = MongoReplicaSetClient.new(@rs.repl_set_seeds, :name => @rs.repl_set_name)
-    assert client.connected?
-    assert client.write_concern[:w] > 0
-    client.close
-    client = MongoReplicaSetClient.new(@rs.repl_set_seeds, :name => @rs.repl_set_name, :w => 0)
-    assert client.connected?
-    assert client.write_concern[:w] < 1
-    client.close
-    client = MongoReplicaSetClient.new(@rs.repl_set_seeds, :name => @rs.repl_set_name, :w => 2)
-    assert client.connected?
-    assert client.write_concern[:w] > 0
-    client.close
-  end
-
-  def test_multiple_concurrent_replica_set_connection
-    client1 = MongoReplicaSetClient.new(@rs.repl_set_seeds, :name => @rs.repl_set_name)
-    client2 = MongoReplicaSetClient.new(@rs.repl_set_seeds, :name => @rs.repl_set_name)
-    assert client1.connected?
-    assert client2.connected?
-    assert client1.manager != client2.manager
-    assert client1.local_manager != client2.local_manager
-    client1.close
-    client2.close
-  end
-
-  def test_cache_original_seed_nodes
-    host = @rs.servers.first.host
-    seeds = @rs.repl_set_seeds << "#{host}:19356"
-    client = MongoReplicaSetClient.new(seeds, :name => @rs.repl_set_name)
-    assert client.connected?
-    assert client.seeds.include?([host, 19356]), "Original seed nodes not cached!"
-    assert_equal [host, 19356], client.seeds.last, "Original seed nodes not cached!"
-    client.close
-  end
-
-  def test_accessors
-    client = MongoReplicaSetClient.from_uri(@uri)
-    assert_equal @rs.primary_name, [client.host, client.port].join(':')
-    assert_equal client.host, client.primary_pool.host
-    assert_equal client.port, client.primary_pool.port
-    assert_equal 2, client.secondaries.length
-    assert_equal 2, client.secondary_pools.length
-    assert_equal @rs.repl_set_name, client.replica_set_name
-    assert client.secondary_pools.include?(client.read_pool({:mode => :secondary}))
-    assert_equal 90, client.refresh_interval
-    assert_equal client.refresh_mode, false
-    client.close
-  end
-
-  def test_write_commands_and_operations
-    @client = MongoReplicaSetClient.from_uri(@uri)
-    @coll = @client[TEST_DB]['test-write-commands-and-operations']
-    with_write_commands_and_operations(@client) do
-      @coll.remove
-      @coll.insert({:foo => "bar"})
-      assert_equal(1, @coll.count)
-    end
-  end
-
-  def test_wnote_does_not_raise_exception_with_err_nil
-    @client = MongoReplicaSetClient.from_uri(@uri)
-    if @client.server_version < '2.5.5'
-      @coll = @client[TEST_DB]['test-wnote']
-      begin
-        result = @coll.remove({:foo => 1}, :w => 2)
-      rescue => ex
-        assert(false, "should not raise an exception for a wnote response field from a remove that does not match any documents")
-      end
-      assert_nil result["err"]
-      assert_true result.has_key?("wnote")
-    end
-  end
-
-  def test_sockets_used_by_forked_process
-    @client = MongoReplicaSetClient.from_uri(@uri)
-    primary_node = @client.manager.primary_pool.node
-    primary_socket = primary_node.socket
-    primary_socket.instance_variable_set(:@pid, primary_socket.pid + 1)
-    primary_node.set_config
-    assert primary_socket != primary_node.socket
-  end
-
-  context "Socket pools" do
-    context "checking out writers" do
-      setup do
-        @client = MongoReplicaSetClient.from_uri(@uri)
-        @coll = @client[TEST_DB]['test-connection-exceptions']
-      end
-
-      should "close the connection on send_message for major exceptions" do
-        with_write_operations(@client) do # explicit even if w 0 maps to write operations
-          @client.expects(:checkout_writer).raises(SystemStackError)
-          @client.expects(:close)
-          begin
-            @coll.insert({:foo => "bar"}, :w => 0)
-          rescue SystemStackError
-          end
-        end
-      end
-
-      should "close the connection on send_write_command for major exceptions" do
-        with_write_commands(@client) do
-          @client.expects(:checkout_reader).raises(SystemStackError)
-          @client.expects(:close)
-          begin
-            @coll.insert({:foo => "bar"})
-          rescue SystemStackError
-          end
-        end
-      end
-
-      should "close the connection on send_message_with_gle for major exceptions" do
-        with_write_operations(@client) do
-          @client.expects(:checkout_writer).raises(SystemStackError)
-          @client.expects(:close)
-          begin
-            @coll.insert({:foo => "bar"})
-          rescue SystemStackError
-          end
-        end
-      end
-
-      should "close the connection on receive_message for major exceptions" do
-        @client.expects(:checkout_reader).raises(SystemStackError)
-        @client.expects(:close)
-        begin
-          @coll.find({}, :read => :primary).next
-        rescue SystemStackError
-        end
-      end
-    end
-
-    context "checking out readers" do
-      setup do
-        @client = MongoReplicaSetClient.from_uri(@uri)
-        @coll = @client[TEST_DB]['test-connection-exceptions']
-      end
-
-      should "close the connection on receive_message for major exceptions" do
-        @client.expects(:checkout_reader).raises(SystemStackError)
-        @client.expects(:close)
-        begin
-          @coll.find({}, :read => :secondary).next
-        rescue SystemStackError
-        end
-      end
-    end
-  end
-
-end

data/test/replica_set/client_test.rb

@@ -1,393 +0,0 @@
-# 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 ReplicaSetClientTest < Test::Unit::TestCase
-
-  def setup
-    ensure_cluster(:rs)
-    @client = nil
-  end
-
-  def teardown
-    @client.close if @client
-  end
-
-  def test_reconnection
-    @client = MongoReplicaSetClient.from_uri(@uri, :op_timeout => TEST_OP_TIMEOUT)
-    assert @client.connected?
-
-    manager = @client.local_manager
-
-    @client.close
-    assert !@client.connected?
-    assert !@client.local_manager
-
-    @client.connect
-    assert @client.connected?
-    assert_equal @client.local_manager, manager
-  end
-
-  def test_reconnect_method
-    @client = MongoReplicaSetClient.from_uri(@uri, :op_timeout => TEST_OP_TIMEOUT)
-    assert @client.connected?
-
-    manager = @client.local_manager
-
-    @client.close
-    assert !@client.connected?
-    assert !@client.local_manager
-
-    @client.reconnect
-    assert @client.connected?
-    assert_equal @client.local_manager, manager
-  end
-
-  # TODO: test connect timeout.
-
-  def test_connect_with_deprecated_multi
-    silently do
-      # guaranteed to have one data-holding member
-      @client = MongoClient.multi(@rs.repl_set_seeds_old, :name => @rs.repl_set_name)
-    end
-    assert !@client.nil?
-    assert @client.connected?
-  end
-
-  def test_connect_bad_name
-    assert_raise_error(ReplicaSetConnectionError, "-wrong") do
-      @client = MongoReplicaSetClient.new(@rs.repl_set_seeds, :name => @rs.repl_set_name + "-wrong")
-    end
-  end
-
-  def test_connect_with_first_secondary_node_terminated
-    @rs.secondaries.first.stop
-
-    rescue_connection_failure do
-      @client = MongoReplicaSetClient.new @rs.repl_set_seeds
-    end
-    assert @client.connected?
-  end
-
-  def test_connect_with_last_secondary_node_terminated
-    @rs.secondaries.last.stop
-
-    rescue_connection_failure do
-      @client = MongoReplicaSetClient.new @rs.repl_set_seeds
-    end
-    assert @client.connected?
-  end
-
-  def test_connect_with_primary_stepped_down
-    @client = MongoReplicaSetClient.from_uri(@uri)
-    @client[TEST_DB]['bar'].save({:a => 1}, {:w => 3})
-    assert @client[TEST_DB]['bar'].find_one
-
-    primary = Mongo::MongoClient.new(*@client.primary)
-    authenticate_client(primary)
-    assert_raise Mongo::ConnectionFailure do
-      perform_step_down(primary)
-    end
-    assert @client.connected?
-
-    rescue_connection_failure do
-      @client[TEST_DB]['bar'].find_one
-    end
-    @client[TEST_DB]['bar'].find_one
-  end
-
-  def test_connect_with_primary_killed
-    @client = MongoReplicaSetClient.from_uri(@uri)
-    assert @client.connected?
-    @client[TEST_DB]['bar'].save({:a => 1}, {:w => 3})
-    assert @client[TEST_DB]['bar'].find_one
-
-    @rs.primary.kill(Signal.list['KILL'])
-
-    sleep(3)
-
-    rescue_connection_failure do
-      @client[TEST_DB]['bar'].find_one
-    end
-    @client[TEST_DB]['bar'].find_one
-  end
-
-  def test_save_with_primary_stepped_down
-    @client = MongoReplicaSetClient.from_uri(@uri)
-    assert @client.connected?
-
-    primary = Mongo::MongoClient.new(*@client.primary)
-    authenticate_client(primary)
-    assert_raise Mongo::ConnectionFailure do
-      perform_step_down(primary)
-    end
-
-    rescue_connection_failure do
-      @client[TEST_DB]['bar'].save({:a => 1}, {:w => 2})
-    end
-    @client[TEST_DB]['bar'].find_one
-  end
-
-  # def test_connect_with_first_node_removed
-  #   @client = MongoReplicaSetClient.from_uri(@uri)
-  #   @client[TEST_DB]['bar'].save({:a => 1}, {:w => 3})
-
-  #   # Make sure everyone's views of optimes are caught up
-  #   loop do
-  #     break if @rs.repl_set_get_status.all? do |status|
-  #       members = status['members']
-  #       primary_optime = members.find{|m| m['state'] == 1}['optime'].seconds
-  #       members.any?{|m| m['state'] == 2 && primary_optime - m['optime'].seconds < 5}
-  #     end
-  #     sleep 1
-  #   end
-
-  #   old_primary = [@client.primary_pool.host, @client.primary_pool.port]
-  #   old_primary_conn = Mongo::MongoClient.new(*old_primary)
-
-  #   assert_raise Mongo::ConnectionFailure do
-  #     perform_step_down(old_primary_conn)
-  #   end
-
-  #   # Wait for new primary
-  #   rescue_connection_failure do
-  #     sleep 1 until @rs.primary
-  #   end
-
-  #   new_primary = [@rs.primary.host, @rs.primary.port]
-  #   new_primary_conn = Mongo::MongoClient.new(*new_primary)
-
-  #   assert new_primary != old_primary
-
-  #   config = nil
-
-  #   # Remove old primary from replset
-  #   rescue_connection_failure do
-  #     config = @client['local']['system.replset'].find_one
-  #   end
-
-  #   old_member = config['members'].select {|m| m['host'] == old_primary.join(':')}.first
-  #   config['members'].reject! {|m| m['host'] == old_primary.join(':')}
-  #   config['version'] += 1
-
-  #   begin
-  #     new_primary_conn['admin'].command({'replSetReconfig' => config})
-  #   rescue Mongo::ConnectionFailure
-  #   end
-
-  #   # Wait for the dust to settle
-  #   rescue_connection_failure do
-  #     assert @client[TEST_DB]['bar'].find_one
-  #   end
-
-  #   begin
-  #     # Make sure a new connection skips the old primary
-  #     @new_conn = MongoReplicaSetClient.new @rs.repl_set_seeds
-  #     @new_conn.connect
-  #     new_nodes = @new_conn.secondaries + [@new_conn.primary]
-  #     assert !new_nodes.include?(old_primary)
-  #   ensure
-  #     # Add the old primary back
-  #     config['members'] << old_member
-  #     config['version'] += 1
-
-  #     begin
-  #       new_primary_conn['admin'].command({'replSetReconfig' => config})
-  #     rescue Mongo::ConnectionFailure
-  #     end
-  #   end
-  # end
-
-  def test_connect_with_hung_first_node
-    hung_node = nil
-    begin
-      hung_node = IO.popen('nc -lk 127.0.0.1 29999 >/dev/null 2>&1')
-
-      Timeout.timeout(3) do
-        @client = MongoReplicaSetClient.new(['localhost:29999'] + @rs.repl_set_seeds,
-          :connect_timeout => 2)
-        @client.connect
-      end
-      assert ['localhost:29999'] != @client.primary
-      assert !@client.secondaries.include?('localhost:29999')
-    ensure
-      begin
-        Process.kill("KILL", hung_node.pid) if hung_node
-      rescue
-        # the process ended, was killed already, or the system doesn't support nc
-      end
-    end
-  end
-
-  def test_read_pref_primary_with_tags
-    parser = Mongo::URIParser.new("mongodb://#{@rs.replicas[0].host_port},#{@rs.replicas[1].host_port}" +
-                                    "?replicaset=#{@rs.repl_set_name}&readPreference=primary&" +
-                                    "readPreferenceTags=dc:ny,rack:1")
-    assert_raise_error Mongo::MongoArgumentError do
-      parser.connection.read_pool
-    end
-  end
-
-  def test_read_pref_with_tags
-    parser = Mongo::URIParser.new("mongodb://#{@rs.replicas[0].host_port},#{@rs.replicas[1].host_port}" +
-                                    "?replicaset=#{@rs.repl_set_name}&" +
-                                    "readPreferenceTags=dc:ny,rack:1")
-    assert_raise_error Mongo::MongoArgumentError do
-      parser.connection.read_pool
-    end
-  end
-
-  def test_connect_with_connection_string
-    @client = MongoClient.from_uri("mongodb://#{@rs.replicas[0].host_port},#{@rs.replicas[1].host_port}?replicaset=#{@rs.repl_set_name}")
-    assert !@client.nil?
-    assert @client.connected?
-  end
-
-  def test_connect_with_connection_string_in_env_var
-    uri = "mongodb://#{@rs.replicas[0].host_port},#{@rs.replicas[1].host_port}?replicaset=#{@rs.repl_set_name}"
-    with_preserved_env_uri(uri) do
-      @client = MongoReplicaSetClient.new
-      assert !@client.nil?
-      assert_equal 2, @client.seeds.length
-      assert_equal @rs.replicas[0].host, @client.seeds[0][0]
-      assert_equal @rs.replicas[1].host, @client.seeds[1][0]
-      assert_equal @rs.replicas[0].port, @client.seeds[0][1]
-      assert_equal @rs.replicas[1].port, @client.seeds[1][1]
-      assert_equal @rs.repl_set_name, @client.replica_set_name
-      assert @client.connected?
-    end
-  end
-
-  def test_connect_with_connection_string_in_implicit_mongodb_uri
-    uri = "mongodb://#{@rs.replicas[0].host_port},#{@rs.replicas[1].host_port}?replicaset=#{@rs.repl_set_name}"
-    with_preserved_env_uri(uri) do
-      @client = MongoClient.from_uri
-      assert !@client.nil?
-      assert_equal 2, @client.seeds.length
-      assert_equal @rs.replicas[0].host, @client.seeds[0][0]
-      assert_equal @rs.replicas[1].host, @client.seeds[1][0]
-      assert_equal @rs.replicas[0].port, @client.seeds[0][1]
-      assert_equal @rs.replicas[1].port, @client.seeds[1][1]
-      assert_equal @rs.repl_set_name, @client.replica_set_name
-      assert @client.connected?
-    end
-  end
-
-  def test_connect_with_new_seed_format
-    @client = MongoReplicaSetClient.new @rs.repl_set_seeds
-    assert @client.connected?
-  end
-
-  def test_connect_with_old_seed_format
-    silently do
-      @client = MongoReplicaSetClient.new(@rs.repl_set_seeds_old)
-    end
-    assert @client.connected?
-  end
-
-  def test_connect_with_full_connection_string
-    @client = MongoClient.from_uri("mongodb://#{@rs.replicas[0].host_port},#{@rs.replicas[1].host_port}?replicaset=#{@rs.repl_set_name};w=2;fsync=true;slaveok=true")
-    assert !@client.nil?
-    assert @client.connected?
-    assert_equal 2, @client.write_concern[:w]
-    assert @client.write_concern[:fsync]
-    assert @client.read_pool
-  end
-
-  def test_connect_with_full_connection_string_in_env_var
-    uri = "mongodb://#{@rs.replicas[0].host_port},#{@rs.replicas[1].host_port}?replicaset=#{@rs.repl_set_name};w=2;fsync=true;slaveok=true"
-    with_preserved_env_uri(uri) do
-      @client = MongoReplicaSetClient.new
-      assert !@client.nil?
-      assert @client.connected?
-      assert_equal 2, @client.write_concern[:w]
-      assert @client.write_concern[:fsync]
-      assert @client.read_pool
-    end
-  end
-
-  def test_connect_options_override_env_var
-    uri = "mongodb://#{@rs.replicas[0].host_port},#{@rs.replicas[1].host_port}?replicaset=#{@rs.repl_set_name};w=2;fsync=true;slaveok=true"
-    with_preserved_env_uri(uri) do
-      @client = MongoReplicaSetClient.new({:w => 0})
-      assert !@client.nil?
-      assert @client.connected?
-      assert_equal 0, @client.write_concern[:w]
-    end
-  end
-
-  def test_ipv6
-    @client = MongoReplicaSetClient.new(@rs.repl_set_seeds)
-    with_ipv6_enabled(@client) do
-      assert MongoReplicaSetClient.new(["[::1]:#{@rs.replicas[0].port}"])
-    end
-  end
-
-  def test_ipv6_with_uri
-    @client = MongoReplicaSetClient.new(@rs.repl_set_seeds)
-    with_ipv6_enabled(@client) do
-      uri = "mongodb://[::1]:#{@rs.replicas[0].port},[::1]:#{@rs.replicas[1].port}"
-      with_preserved_env_uri(uri) do
-        assert MongoReplicaSetClient.new
-      end
-    end
-  end
-
-  def test_ipv6_with_uri_opts
-    @client = MongoReplicaSetClient.new(@rs.repl_set_seeds)
-    with_ipv6_enabled(@client) do
-      uri = "mongodb://[::1]:#{@rs.replicas[0].port},[::1]:#{@rs.replicas[1].port}/?safe=true;"
-      with_preserved_env_uri(uri) do
-        assert MongoReplicaSetClient.new
-      end
-    end
-  end
-
-  def test_ipv6_with_different_formats
-    @client = MongoReplicaSetClient.new(@rs.repl_set_seeds)
-    with_ipv6_enabled(@client) do
-      uri = "mongodb://[::1]:#{@rs.replicas[0].port},localhost:#{@rs.replicas[1].port}"
-      with_preserved_env_uri(uri) do
-        assert MongoReplicaSetClient.new
-      end
-    end
-  end
-
-  def test_find_and_modify_with_secondary_read_preference
-    @client = MongoReplicaSetClient.from_uri(@uri)
-    collection = @client[TEST_DB].collection('test', :read => :secondary)
-    id = BSON::ObjectId.new
-    collection << { :a => id, :processed => false }
-
-    collection.find_and_modify(
-      :query => { 'a' => id },
-      :update => { "$set" => { :processed => true }}
-    )
-    assert_equal true, collection.find_one({ 'a' => id }, :read => :primary)['processed']
-  end
-
-  def test_op_timeout_option
-    client = MongoReplicaSetClient.new(@rs.repl_set_seeds, :connect => false,
-                                                           :op_timeout => nil)
-    assert_equal nil, client.op_timeout
-
-    client = MongoReplicaSetClient.new(@rs.repl_set_seeds, :connect => false,
-                                                           :op_timeout => 50)
-    assert_equal 50, client.op_timeout
-
-    client = MongoReplicaSetClient.new(@rs.repl_set_seeds, :connect => false)
-    assert_equal Mongo::MongoClient::DEFAULT_OP_TIMEOUT, client.op_timeout
-  end
-end