mongo 1.12.5 → 2.0.0

data/test/unit/db_test.rb

@@ -1,136 +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'
-
-def insert_message(db, documents)
-  documents = [documents] unless documents.is_a?(Array)
-  message = ByteBuffer.new
-  message.put_int(0)
-  Mongo::BSON_CODER.serialize_cstr(message, "#{db.name}.test")
-  documents.each { |doc| message.put_array(Mongo::BSON_CODER.new.serialize(doc, true).to_a) }
-  message = db.add_message_headers(Mongo::Constants::OP_INSERT, message)
-end
-
-class DBUnitTest < Test::Unit::TestCase
-  context "DBTest: " do
-    context "DB commands" do
-      setup do
-        @client = stub()
-        @client.stubs(:write_concern).returns({})
-        @client.stubs(:read).returns(:primary)
-        @client.stubs(:tag_sets)
-        @client.stubs(:acceptable_latency)
-        @client.stubs(:add_auth).returns({})
-        @db = DB.new("testing", @client)
-        @db.stubs(:safe)
-        @db.stubs(:read)
-        @db.stubs(:tag_sets)
-        @db.stubs(:acceptable_latency)
-        @collection = mock()
-        @db.stubs(:system_command_collection).returns(@collection)
-      end
-
-      should "raise an error if given a hash with more than one key" do
-        if RUBY_VERSION < '1.9'
-          assert_raise MongoArgumentError do
-            @db.command(:buildinfo => 1, :somekey => 1)
-          end
-        end
-      end
-
-      should "raise an error if the selector is omitted" do
-        assert_raise MongoArgumentError do
-          @db.command({}, :check_response => true)
-        end
-      end
-
-      should "not include named nil opts in selector" do
-        @cursor = mock(:next_document => {"ok" => 1})
-        Cursor.expects(:new).with(@collection, :limit => -1,
-          :selector => {:ping => 1}, :socket => nil).returns(@cursor)
-        command = {:ping => 1}
-        @db.command(command, :socket => nil)
-      end
-
-      should "create the proper cursor" do
-        @cursor = mock(:next_document => {"ok" => 1})
-        Cursor.expects(:new).with(@collection,
-          :limit => -1, :selector => {:buildinfo => 1}).returns(@cursor)
-        command = {:buildinfo => 1}
-        @db.command(command, :check_response => true)
-      end
-
-      should "raise an error when the command fails" do
-        @cursor = mock(:next_document => {"ok" => 0})
-        Cursor.expects(:new).with(@collection,
-          :limit => -1, :selector => {:buildinfo => 1}).returns(@cursor)
-        assert_raise OperationFailure do
-          command = {:buildinfo => 1}
-          @db.command(command, :check_response => true)
-        end
-      end
-
-      should "pass on the comment" do
-        @cursor = mock(:next_document => {"ok" => 0})
-        Cursor.expects(:new).with(@collection,
-          :limit => -1, :selector => {:buildinfo => 1},
-          :comment => "my comment").returns(@cursor)
-        assert_raise OperationFailure do
-          command = {:buildinfo => 1}
-          @db.command(command, :check_response => true, :comment => 'my comment')
-        end
-      end
-
-      should "raise an error if collection creation fails" do
-        @db.expects(:command).returns({'ok' => 0})
-        assert_raise Mongo::MongoDBError do
-          @db.create_collection("foo")
-        end
-      end
-
-      should "raise an error if getlasterror fails" do
-        @db.expects(:command).returns({})
-        assert_raise Mongo::MongoDBError do
-          @db.get_last_error
-        end
-      end
-
-      should "raise an error if drop_index fails" do
-        @db.expects(:command).returns({})
-        assert_raise Mongo::MongoDBError do
-          @db.drop_index("foo", "bar")
-        end
-      end
-
-      should "raise an error if set_profiling_level fails" do
-        @db.expects(:command).returns({})
-        assert_raise Mongo::MongoDBError do
-          @db.profiling_level = :slow_only
-        end
-      end
-
-      should "warn when save_auth is not nil" do
-        assert @db.expects(:warn).with(regexp_matches(/\[DEPRECATED\] Disabling the 'save_auth' option/))
-        @db.authenticate('foo', 'bar', false)
-      end
-
-      should "allow extra authentication options" do
-        extra_opts = { :service_name => 'example', :canonicalize_host_name => true }
-        assert @client.expects(:add_auth).with(@db.name, 'emily', nil, nil, 'GSSAPI', extra_opts)
-        @db.authenticate('emily', nil, nil, nil, 'GSSAPI', extra_opts)
-      end
-    end
-  end
-end

data/test/unit/grid_test.rb

@@ -1,76 +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 GridUnitTest < Test::Unit::TestCase
-
-  context "GridFS: " do
-    setup do
-      @client   = stub()
-      @client.stubs(:write_concern).returns({})
-      @client.stubs(:read).returns(:primary)
-      @client.stubs(:tag_sets)
-      @client.stubs(:acceptable_latency)
-      @db     = DB.new("testing", @client)
-      @files  = mock()
-      @chunks = mock()
-
-      @db.stubs(:[]).with('fs.files').returns(@files)
-      @db.stubs(:[]).with('fs.chunks').returns(@chunks)
-      @db.stubs(:safe)
-      @db.stubs(:read).returns(:primary)
-    end
-
-    context "Grid classes with standard connections" do
-      setup do
-        @chunks.expects(:ensure_index)
-      end
-
-      should "create indexes for Grid" do
-        Grid.new(@db)
-      end
-
-      should "create indexes for GridFileSystem" do
-        @files.expects(:ensure_index)
-        GridFileSystem.new(@db)
-      end
-    end
-
-    context "Grid classes with slave connection" do
-      setup do
-        @chunks.stubs(:ensure_index).raises(Mongo::ConnectionFailure)
-        @files.stubs(:ensure_index).raises(Mongo::ConnectionFailure)
-      end
-
-      should "not create indexes for Grid" do
-        grid = Grid.new(@db)
-        data = "hello world!"
-        assert_raise Mongo::ConnectionFailure do
-          grid.put(data)
-        end
-      end
-
-      should "not create indexes for GridFileSystem" do
-        gridfs = GridFileSystem.new(@db)
-        data = "hello world!"
-        assert_raise Mongo::ConnectionFailure do
-          gridfs.open('image.jpg', 'w') do |f|
-            f.write data
-          end
-        end
-      end
-    end
-  end
-end

data/test/unit/mongo_sharded_client_test.rb

@@ -1,48 +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 MongoShardedClientUnitTest < Test::Unit::TestCase
-  include Mongo
-
-  def test_initialize_with_single_mongos_uri
-    uri = "mongodb://localhost:27017"
-    with_preserved_env_uri(uri) do
-      client = MongoShardedClient.new(:connect => false)
-      assert_equal [[ "localhost", 27017 ]], client.seeds
-    end
-  end
-
-  def test_initialize_with_multiple_mongos_uris
-    uri = "mongodb://localhost:27017,localhost:27018"
-    with_preserved_env_uri(uri) do
-      client = MongoShardedClient.new(:connect => false)
-      assert_equal [[ "localhost", 27017 ], [ "localhost", 27018 ]], client.seeds
-    end
-  end
-
-  def test_from_uri_with_string
-    client = MongoShardedClient.from_uri("mongodb://localhost:27017,localhost:27018", :connect => false)
-    assert_equal [[ "localhost", 27017 ], [ "localhost", 27018 ]], client.seeds
-  end
-
-  def test_from_uri_with_env_variable
-    uri = "mongodb://localhost:27017,localhost:27018"
-    with_preserved_env_uri(uri) do
-      client = MongoShardedClient.from_uri(nil, :connect => false)
-      assert_equal [[ "localhost", 27017 ], [ "localhost", 27018 ]], client.seeds
-    end
-  end
-end

data/test/unit/node_test.rb

@@ -1,93 +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 NodeUnitTest < Test::Unit::TestCase
-
-  def setup
-    @client = stub()
-    manager = mock('pool_manager')
-    manager.stubs(:update_max_sizes)
-    @client.stubs(:local_manager).returns(manager)
-  end
-
-  should "refuse to connect to node without 'hosts' key" do
-    tcp = mock()
-    node = Node.new(@client, ['localhost', 27017])
-    tcp.stubs(:new).returns(new_mock_socket)
-    @client.stubs(:socket_class).returns(tcp)
-
-    admin_db = new_mock_db
-    admin_db.stubs(:command).returns({'ok' => 1, 'ismaster' => 1})
-    @client.stubs(:[]).with('admin').returns(admin_db)
-    @client.stubs(:op_timeout).returns(nil)
-    @client.stubs(:connect_timeout).returns(nil)
-    @client.expects(:log)
-    @client.expects(:mongos?).returns(false)
-    @client.stubs(:socket_opts)
-
-    assert node.connect
-    node.config
-  end
-
-  should "load a node from an array" do
-    node = Node.new(@client, ['power.level.com', 9001])
-    assert_equal 'power.level.com', node.host
-    assert_equal 9001, node.port
-    assert_equal 'power.level.com:9001', node.address
-  end
-
-  should "should default the port for an array" do
-    node = Node.new(@client, ['power.level.com'])
-    assert_equal 'power.level.com', node.host
-    assert_equal MongoClient::DEFAULT_PORT, node.port
-    assert_equal "power.level.com:#{MongoClient::DEFAULT_PORT}", node.address
-  end
-
-  should "load a node from a string" do
-    node = Node.new(@client, 'localhost:1234')
-    assert_equal 'localhost', node.host
-    assert_equal 1234, node.port
-    assert_equal 'localhost:1234', node.address
-  end
-
-  should "should default the port for a string" do
-    node = Node.new(@client, '192.168.0.1')
-    assert_equal '192.168.0.1', node.host
-    assert_equal MongoClient::DEFAULT_PORT, node.port
-    assert_equal "192.168.0.1:#{MongoClient::DEFAULT_PORT}", node.address
-  end
-
-  should "two nodes with the same address should be equal" do
-    assert_equal Node.new(@client, '192.168.0.1'),
-      Node.new(@client, ['192.168.0.1', MongoClient::DEFAULT_PORT])
-  end
-
-  should "two nodes with the same address should have the same hash" do
-    assert_equal Node.new(@client, '192.168.0.1').hash,
-      Node.new(@client, ['192.168.0.1', MongoClient::DEFAULT_PORT]).hash
-  end
-
-  should "two nodes with different addresses should not be equal" do
-    assert_not_equal Node.new(@client, '192.168.0.2'),
-      Node.new(@client, ['192.168.0.1', MongoClient::DEFAULT_PORT])
-  end
-
-  should "two nodes with the same address should have the same hash negate" do
-    assert_not_equal Node.new(@client, '192.168.0.1').hash,
-      Node.new(@client, '1239.33.4.2393:29949').hash
-  end
-
-end

data/test/unit/pool_manager_test.rb

@@ -1,111 +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'
-include Mongo
-
-class PoolManagerUnitTest < Test::Unit::TestCase
-
-  context "Initialization: " do
-
-    setup do
-      TCPSocket.stubs(:new).returns(new_mock_socket)
-      @db = new_mock_db
-
-      @client = stub("MongoClient")
-      @client.stubs(:connect_timeout).returns(5)
-      @client.stubs(:op_timeout).returns(5)
-      @client.stubs(:pool_size).returns(2)
-      @client.stubs(:pool_timeout).returns(100)
-      @client.stubs(:seeds).returns(['localhost:30000'])
-      @client.stubs(:socket_class).returns(TCPSocket)
-      @client.stubs(:mongos?).returns(false)
-      @client.stubs(:[]).returns(@db)
-      @client.stubs(:socket_opts)
-
-      @client.stubs(:replica_set_name).returns(nil)
-      @client.stubs(:log)
-      @arbiters = ['localhost:27020']
-      @hosts = [
-        'localhost:27017',
-        'localhost:27018',
-        'localhost:27019',
-        'localhost:27020'
-      ]
-
-      @ismaster = {
-        'hosts' => @hosts,
-        'arbiters' => @arbiters,
-        'maxBsonObjectSize' => 1024,
-        'maxMessageSizeBytes' => 1024 * 2.5,
-        'maxWireVersion' => 1,
-        'minWireVersion' => 0
-      }
-    end
-
-    should "populate pools correctly" do
-
-      @db.stubs(:command).returns(
-        # First call to get a socket.
-        @ismaster.merge({'ismaster' => true}),
-
-        # Subsequent calls to configure pools.
-        @ismaster.merge({'ismaster' => true}),
-        @ismaster.merge({'secondary' => true, 'maxBsonObjectSize' => 500}),
-        @ismaster.merge({'secondary' => true, 'maxMessageSizeBytes' => 700}),
-        @ismaster.merge({'arbiterOnly' => true})
-      )
-
-      seeds = [['localhost', 27017]]
-      manager = Mongo::PoolManager.new(@client, seeds)
-      @client.stubs(:local_manager).returns(manager)
-      manager.connect
-
-      assert_equal ['localhost', 27017], manager.primary
-      assert_equal 27017, manager.primary_pool.port
-      assert_equal 2, manager.secondaries.length
-      assert_equal [27018, 27019], manager.secondary_pools.map(&:port).sort
-      assert_equal [['localhost', 27020]], manager.arbiters
-      assert_equal 500, manager.max_bson_size
-      assert_equal 700, manager.max_message_size
-    end
-
-    should "populate pools with single unqueryable seed" do
-
-      @db.stubs(:command).returns(
-        # First call to recovering node
-        @ismaster.merge({'ismaster' => false, 'secondary' => false}),
-
-        # Subsequent calls to configure pools.
-        @ismaster.merge({'ismaster' => false, 'secondary' => false}),
-        @ismaster.merge({'ismaster' => true}),
-        @ismaster.merge({'secondary' => true}),
-        @ismaster.merge({'arbiterOnly' => true})
-      )
-
-      seeds = [['localhost', 27017]]
-      manager = PoolManager.new(@client, seeds)
-      @client.stubs(:local_manager).returns(manager)
-      manager.connect
-
-      assert_equal ['localhost', 27018], manager.primary
-      assert_equal 27018, manager.primary_pool.port
-      assert_equal 1, manager.secondaries.length
-      assert_equal 27019, manager.secondary_pools[0].port
-      assert_equal [['localhost', 27020]], manager.arbiters
-    end
-
-  end
-
-end