mongo 1.3.0 → 1.12.5

data/test/functional/collection_writer_test.rb

@@ -0,0 +1,83 @@
+# 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'
+
+module Mongo
+  class Collection
+    public :batch_write
+  end
+  class CollectionWriter
+    public :sort_by_first_sym, :ordered_group_by_first
+  end
+end
+
+class CollectionWriterTest < Test::Unit::TestCase
+
+  DATABASE_NAME = 'ruby_test_collection_writer'
+  COLLECTION_NAME = 'test'
+
+  def default_setup
+    @client = MongoClient.from_uri(TEST_URI)
+    @db = @client[DATABASE_NAME]
+    @collection = @db[COLLECTION_NAME]
+    @collection.drop
+  end
+
+  context "Bulk API Execute" do
+    setup do
+      default_setup
+    end
+
+    should "sort_by_first_sym for grouping unordered ops" do
+      pairs = [
+          [:insert, {:n => 0}],
+          [:update, {:n => 1}], [:update, {:n => 2}],
+          [:delete, {:n => 3}],
+          [:insert, {:n => 5}], [:insert, {:n => 6}], [:insert, {:n => 7}],
+          [:update, {:n => 8}],
+          [:delete, {:n => 9}], [:delete, {:n => 10}]
+      ]
+      result = @collection.command_writer.sort_by_first_sym(pairs)
+      expected = [
+          :delete, :delete, :delete,
+          :insert, :insert, :insert, :insert,
+          :update, :update, :update
+      ]
+      assert_equal expected, result.collect{|first, rest| first}
+    end
+
+    should "calculate ordered_group_by_first" do
+      pairs = [
+          [:insert, {:n => 0}],
+          [:update, {:n => 1}], [:update, {:n => 2}],
+          [:delete, {:n => 3}],
+          [:insert, {:n => 5}], [:insert, {:n => 6}], [:insert, {:n => 7}],
+          [:update, {:n => 8}],
+          [:delete, {:n => 9}], [:delete, {:n => 10}]
+      ]
+      result = @collection.command_writer.ordered_group_by_first(pairs)
+      expected = [
+          [:insert, [{:n => 0}]],
+          [:update, [{:n => 1}, {:n => 2}]],
+          [:delete, [{:n => 3}]],
+          [:insert, [{:n => 5}, {:n => 6}, {:n => 7}]],
+          [:update, [{:n => 8}]],
+          [:delete, [{:n => 9}, {:n => 10}]]
+      ]
+      assert_equal expected, result
+    end
+
+  end
+end

data/test/{conversions_test.rb → functional/conversions_test.rb}

@@ -1,6 +1,18 @@
-require './test/test_helper'
-require 'mongo/exceptions'
-require 'mongo/util/conversions'
+# 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 ConversionsTest < Test::Unit::TestCase
   include Mongo::Conversions
@@ -15,6 +27,34 @@ class ConversionsTest < Test::Unit::TestCase
     assert_equal({ "field1" => 1, "field2" => -1 }, params)
   end
 
+  def test_array_as_sort_parameters_with_array_of_key_and_hash
+    params = array_as_sort_parameters(["score", {"$meta" => "textScore"}])
+    assert_equal({"score" => {"$meta" => "textScore"}}, params)
+  end
+
+  def test_array_as_sort_parameters_with_array_of_key_and_hashes
+    params = array_as_sort_parameters([["field1", :asc],["score", {"$meta" => "textScore"}]])
+    assert_equal({"field1" => 1, "score" => {"$meta" => "textScore"}}, params)
+  end
+
+  def test_hash_as_sort_parameters_with_string
+    sort = BSON::OrderedHash["field", "asc"]
+    params = hash_as_sort_parameters(sort)
+    assert_equal({"field" => 1}, params)
+  end
+
+  def test_hash_as_sort_parameters_with_hash
+    sort = BSON::OrderedHash["score", {"$meta" => "textScore"}]
+    params = hash_as_sort_parameters(sort)
+    assert_equal({"score" => {"$meta" => "textScore"}}, params)
+  end
+
+  def test_hash_as_sort_parameters_with_hash_and_string
+    sort = BSON::OrderedHash["score", {"$meta" => "textScore"}, "field", "asc"]
+    params = hash_as_sort_parameters(sort)
+    assert_equal({ "score" => {"$meta" => "textScore"}, "field" => 1 }, params)
+  end
+
   def test_string_as_sort_parameters_with_string
     params = string_as_sort_parameters("field")
     assert_equal({ "field" => 1 }, params)
@@ -110,6 +150,10 @@ class ConversionsTest < Test::Unit::TestCase
     assert_equal(-1, sort_value(:DESC))
   end
 
+  def test_sort_value_when_value_is_hash
+    assert_equal({"$meta" => "textScore"}, sort_value("$meta" => "textScore"))
+  end
+
   def test_sort_value_when_value_is_invalid
     assert_raise Mongo::InvalidSortValueError do
       sort_value(2)

data/test/functional/cursor_fail_test.rb

@@ -0,0 +1,57 @@
+# 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'
+require 'logger'
+
+class CursorFailTest < Test::Unit::TestCase
+
+  include Mongo
+
+  def setup
+    @connection = standard_connection
+    @db = @connection[TEST_DB]
+  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

data/test/functional/cursor_message_test.rb

@@ -0,0 +1,56 @@
+# 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'
+require 'logger'
+
+class CursorMessageTest < Test::Unit::TestCase
+
+  include Mongo
+
+  def setup
+    @connection = standard_connection
+    @db         = @connection.db(TEST_DB)
+    @coll       = @db.collection('test')
+    @version    = @connection.server_version
+    @coll.remove
+    @coll.insert('a' => 1)     # collection not created until it's used
+    @coll_full_name = "#{TEST_DB}.test"
+  end
+
+  def test_valid_batch_sizes
+    assert_raise ArgumentError do
+      @coll.find({}, :batch_size => 1, :limit => 5)
+    end
+
+    assert_raise ArgumentError do
+      @coll.find({}, :batch_size => -1, :limit => 5)
+    end
+
+    assert @coll.find({}, :batch_size => 0, :limit => 5)
+  end
+
+  def test_batch_size
+    @coll.remove
+    200.times do |n|
+      @coll.insert({:a => n})
+    end
+
+    list = @coll.find({}, :batch_size => 2, :limit => 6).to_a
+    assert_equal 6, list.length
+
+    list = @coll.find({}, :batch_size => 100, :limit => 101).to_a
+    assert_equal 101, list.length
+  end
+end