mongo 1.8.6 → 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/functional/conversions_test.rb

@@ -1,6 +1,18 @@
+# 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 'mongo/exceptions'
-require 'mongo/util/conversions'
 
 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

@@ -1,3 +1,17 @@
+# 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'
 
@@ -5,19 +19,13 @@ class CursorFailTest < Test::Unit::TestCase
 
   include Mongo
 
-  @@connection = standard_connection
-  @@db   = @@connection.db(MONGO_TEST_DB)
-  @@coll = @@db.collection('test')
-  @@version = @@connection.server_version
-
   def setup
-    @@coll.remove({})
-    @@coll.insert({'a' => 1})     # collection not created until it's used
-    @@coll_full_name = "#{MONGO_TEST_DB}.test"
+    @connection = standard_connection
+    @db = @connection[TEST_DB]
   end
 
   def test_refill_via_get_more_alt_coll
-    coll = @@db.collection('test-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

data/test/functional/cursor_message_test.rb

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