mongo 1.12.5 → 2.0.0.beta

data/test/unit/cursor_test.rb

@@ -1,307 +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 CursorUnitTest < Test::Unit::TestCase
-  class Mongo::Cursor
-    public :construct_query_spec
-  end
-
-  context "Cursor options" do
-    setup do
-      @logger     = mock()
-      @logger.stubs(:debug)
-      @connection = stub(:class => MongoClient, :logger => @logger,
-        :slave_ok? => false, :read => :primary, :log_duration => false,
-        :tag_sets => [], :acceptable_latency => 10)
-      @db         = stub(:name => "testing", :slave_ok? => false,
-        :connection => @connection, :read => :primary,
-        :tag_sets => [], :acceptable_latency => 10)
-      @collection = stub(:db => @db, :name => "items", :read => :primary,
-        :tag_sets => [], :acceptable_latency => 10)
-      @cursor     = Cursor.new(@collection)
-    end
-
-    should "set timeout" do
-      assert @cursor.timeout
-      assert @cursor.query_options_hash[:timeout]
-    end
-
-    should "set selector" do
-      assert_equal({}, @cursor.selector)
-
-      @cursor = Cursor.new(@collection, :selector => {:name => "Jones"})
-      assert_equal({:name => "Jones"}, @cursor.selector)
-      assert_equal({:name => "Jones"}, @cursor.query_options_hash[:selector])
-    end
-
-    should "set fields" do
-      assert_nil @cursor.fields
-
-      @cursor = Cursor.new(@collection, :fields => [:name, :date])
-      assert_equal({:name => 1, :date => 1}, @cursor.fields)
-      assert_equal({:name => 1, :date => 1}, @cursor.query_options_hash[:fields])
-    end
-
-    should "allow $meta projection operator" do
-      assert_nil @cursor.fields
-
-      @cursor = Cursor.new(@collection, :fields => [{ :score => { :$meta => 'textScore' } }])
-      assert_equal({ :score  => { :$meta => 'textScore' } }, @cursor.fields)
-      assert_equal({ :score  => { :$meta => 'textScore' } }, @cursor.query_options_hash[:fields])
-
-      @cursor = Cursor.new(@collection, :fields => [:name, { :score => { :$meta => 'textScore' } }])
-      assert_equal({ :name => 1, :score => { :$meta => 'textScore' } }, @cursor.fields)
-      assert_equal({ :name => 1, :score => { :$meta => 'textScore' } }, @cursor.query_options_hash[:fields])
-    end
-
-    should "set mix fields 0 and 1" do
-      assert_nil @cursor.fields
-
-      @cursor = Cursor.new(@collection, :fields => {:name => 1, :date => 0})
-      assert_equal({:name => 1, :date => 0}, @cursor.fields)
-      assert_equal({:name => 1, :date => 0}, @cursor.query_options_hash[:fields])
-    end
-
-    should "set limit" do
-      assert_equal 0, @cursor.limit
-
-      @cursor = Cursor.new(@collection, :limit => 10)
-      assert_equal 10, @cursor.limit
-      assert_equal 10, @cursor.query_options_hash[:limit]
-    end
-
-
-    should "set skip" do
-      assert_equal 0, @cursor.skip
-
-      @cursor = Cursor.new(@collection, :skip => 5)
-      assert_equal 5, @cursor.skip
-      assert_equal 5, @cursor.query_options_hash[:skip]
-    end
-
-    should "set sort order" do
-      assert_nil @cursor.order
-
-      @cursor = Cursor.new(@collection, :order => "last_name")
-      assert_equal "last_name", @cursor.order
-      assert_equal "last_name", @cursor.query_options_hash[:order]
-    end
-
-    should "set hint" do
-      assert_nil @cursor.hint
-
-      @cursor = Cursor.new(@collection, :hint => "name")
-      assert_equal "name", @cursor.hint
-      assert_equal "name", @cursor.query_options_hash[:hint]
-    end
-
-    should "set comment" do
-      assert_nil @cursor.comment
-
-      @cursor = Cursor.new(@collection, :comment => "comment")
-      assert_equal "comment", @cursor.comment
-      assert_equal "comment", @cursor.query_options_hash[:comment]
-    end
-
-    should "cache full collection name" do
-      assert_equal "testing.items", @cursor.full_collection_name
-    end
-
-    should "raise error when batch_size is 1" do
-      e = assert_raise ArgumentError do
-        @cursor.batch_size(1)
-        end
-        assert_equal "Invalid value for batch_size 1; must be 0 or > 1.", e.message
-    end
-
-    should "use the limit for batch size when it's smaller than the specified batch_size" do
-      @cursor.limit(99)
-      @cursor.batch_size(100)
-      assert_equal 99, @cursor.batch_size
-      end
-
-    should "use the specified batch_size" do
-      @cursor.batch_size(100)
-      assert_equal 100, @cursor.batch_size
-    end
-
-    context 'when an alternate namespace is specified' do
-
-      should 'use the alternate namespace' do
-        cursor = Cursor.new(@collection, :ns => 'other_db.other_collection')
-        assert_equal('other_db.other_collection', cursor.full_collection_name)
-      end
-    end
-
-    context "connected to mongos" do
-      setup do
-        @connection.stubs(:mongos?).returns(true)
-        @tag_sets = [{:dc => "ny"}]
-      end
-
-      should "set $readPreference" do
-        # secondary
-        cursor = Cursor.new(@collection, { :read => :secondary })
-
-        spec = cursor.construct_query_spec
-        assert spec.has_key?('$readPreference')
-        assert_equal 'secondary', spec['$readPreference'][:mode]
-        assert !spec['$readPreference'].has_key?(:tags)
-
-        # secondary preferred with tags
-        cursor = Cursor.new(@collection, { :read => :secondary_preferred, :tag_sets => @tag_sets })
-
-        spec = cursor.construct_query_spec
-        assert spec.has_key?('$readPreference')
-        assert_equal 'secondaryPreferred', spec['$readPreference'][:mode]
-        assert_equal @tag_sets, spec['$readPreference'][:tags]
-
-        # primary preferred
-        cursor = Cursor.new(@collection, { :read => :primary_preferred })
-
-        spec = cursor.construct_query_spec
-        assert spec.has_key?('$readPreference')
-        assert_equal 'primaryPreferred', spec['$readPreference'][:mode]
-        assert !spec['$readPreference'].has_key?(:tags)
-
-        # primary preferred with tags
-        cursor = Cursor.new(@collection, { :read => :primary_preferred, :tag_sets => @tag_sets })
-
-        spec = cursor.construct_query_spec
-        assert spec.has_key?('$readPreference')
-        assert_equal 'primaryPreferred', spec['$readPreference'][:mode]
-        assert_equal @tag_sets, spec['$readPreference'][:tags]
-
-        # nearest
-        cursor = Cursor.new(@collection, { :read => :nearest })
-
-        spec = cursor.construct_query_spec
-        assert spec.has_key?('$readPreference')
-        assert_equal 'nearest', spec['$readPreference'][:mode]
-        assert !spec['$readPreference'].has_key?(:tags)
-
-        # nearest with tags
-        cursor = Cursor.new(@collection, { :read => :nearest, :tag_sets => @tag_sets })
-
-        spec = cursor.construct_query_spec
-        assert spec.has_key?('$readPreference')
-        assert_equal 'nearest', spec['$readPreference'][:mode]
-        assert_equal @tag_sets, spec['$readPreference'][:tags]
-      end
-
-      should "not set $readPreference" do
-        # for primary
-        cursor = Cursor.new(@collection, { :read => :primary, :tag_sets => @tag_sets })
-        assert !cursor.construct_query_spec.has_key?('$readPreference')
-
-        # for secondary_preferred with no tags
-        cursor = Cursor.new(@collection, { :read => :secondary_preferred })
-        assert !cursor.construct_query_spec.has_key?('$readPreference')
-
-        cursor = Cursor.new(@collection, { :read => :secondary_preferred, :tag_sets => [] })
-        assert !cursor.construct_query_spec.has_key?('$readPreference')
-
-        cursor = Cursor.new(@collection, { :read => :secondary_preferred, :tag_sets => nil })
-        assert !cursor.construct_query_spec.has_key?('$readPreference')
-      end
-    end
-
-    context "not connected to mongos" do
-      setup do
-        @connection.stubs(:mongos?).returns(false)
-      end
-
-      should "not set $readPreference" do
-        cursor = Cursor.new(@collection, { :read => :primary })
-        assert !cursor.construct_query_spec.has_key?('$readPreference')
-
-        cursor = Cursor.new(@collection, { :read => :primary_preferred })
-        assert !cursor.construct_query_spec.has_key?('$readPreference')
-
-        cursor = Cursor.new(@collection, { :read => :secondary })
-        assert !cursor.construct_query_spec.has_key?('$readPreference')
-
-        cursor = Cursor.new(@collection, { :read => :secondary_preferred })
-        assert !cursor.construct_query_spec.has_key?('$readPreference')
-
-        cursor = Cursor.new(@collection, { :read => :nearest })
-        assert !cursor.construct_query_spec.has_key?('$readPreference')
-
-        cursor = Cursor.new(@collection, { :read => :secondary , :tag_sets => @tag_sets})
-        assert !cursor.construct_query_spec.has_key?('$readPreference')
-      end
-    end
-  end
-
-  context "Query fields" do
-    setup do
-      @logger     = mock()
-      @logger.stubs(:debug)
-      @connection = stub(:class => MongoClient, :logger => @logger, :slave_ok? => false,
-        :log_duration => false, :tag_sets =>{}, :acceptable_latency => 10)
-      @db = stub(:slave_ok? => true, :name => "testing", :connection => @connection,
-        :tag_sets => {}, :acceptable_latency => 10)
-      @collection = stub(:db => @db, :name => "items", :read => :primary,
-        :tag_sets => {}, :acceptable_latency => 10)
-    end
-
-    should "when an array should return a hash with each key" do
-      @cursor = Cursor.new(@collection, :fields => [:name, :age])
-      result  = @cursor.fields
-      assert_equal result.keys.sort{|a,b| a.to_s <=> b.to_s}, [:age, :name].sort{|a,b| a.to_s <=> b.to_s}
-      assert result.values.all? {|v| v == 1}
-    end
-
-    should "when a string, return a hash with just the key" do
-      @cursor = Cursor.new(@collection, :fields => "name")
-      result  = @cursor.fields
-      assert_equal result.keys.sort, ["name"]
-      assert result.values.all? {|v| v == 1}
-    end
-
-    should "return nil when neither hash nor string nor symbol" do
-      @cursor = Cursor.new(@collection, :fields => 1234567)
-      assert_nil @cursor.fields
-    end
-  end
-
-  context "counts" do
-    setup do
-      @logger     = mock()
-      @logger.stubs(:debug)
-      @connection = stub(:class => Connection, :logger => @logger,
-        :slave_ok? => false, :read => :primary, :log_duration => false,
-        :tag_sets => {}, :acceptable_latency => 10)
-      @db         = stub(:name => "testing", :slave_ok? => false,
-        :connection => @connection, :read => :primary,
-        :tag_sets => {}, :acceptable_latency => 10)
-      @collection = stub(:db => @db, :name => "items", :read => :primary,
-        :tag_sets => {}, :acceptable_latency => 10)
-      @cursor     = Cursor.new(@collection)
-    end
-
-    should "pass the comment parameter" do
-      query = {:field => 7}
-      @db.expects(:command).with({ 'count' => "items",
-                                   'query' => query,
-                                   'fields' => nil},
-                                 { :read => :primary,
-                                   :comment => "my comment"}).
-        returns({'ok' => 1, 'n' => 1})
-      assert_equal(1, Cursor.new(@collection, :selector => query, :comment => 'my comment').count())
-    end
-  end
-end

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