github-migration/mongo-ruby-driver
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

RUBY-222 update tests for new Collection#group API

Browse Source
This commit is contained in:
Kyle Banker 2011-01-06 10:32:50 -05:00
parent bcb37e62c7
commit ba88f239de
1 changed files with 17 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
test/db_api_test.rb
View File

@ -509,33 +509,37 @@ class DBAPITest < Test::Unit::TestCase
@@db.drop_collection("test") @@db.drop_collection("test")
test = @@db.collection("test") test = @@db.collection("test")
assert_equal [], test.group([], {}, {"count" => 0}, "function (obj, prev) { prev.count++; }") assert_equal [], test.group(:initial => {"count" => 0}, :reduce => "function (obj, prev) { prev.count++; }")
assert_equal [], test.group([], {}, {"count" => 0}, "function (obj, prev) { prev.count++; }") assert_equal [], test.group(:initial => {"count" => 0}, :reduce => "function (obj, prev) { prev.count++; }")
test.insert("a" => 2) test.insert("a" => 2)
test.insert("b" => 5) test.insert("b" => 5)
test.insert("a" => 1) test.insert("a" => 1)
assert_equal 3, test.group([], {}, {"count" => 0}, "function (obj, prev) { prev.count++; }")[0]["count"] assert_equal 3, test.group(:initial => {"count" => 0},
assert_equal 3, test.group([], {}, {"count" => 0}, "function (obj, prev) { prev.count++; }")[0]["count"] :reduce => "function (obj, prev) { prev.count++; }")[0]["count"]
assert_equal 1, test.group([], {"a" => {"$gt" => 1}}, {"count" => 0}, "function (obj, prev) { prev.count++; }")[0]["count"] assert_equal 3, test.group(:initial => {"count" => 0},
assert_equal 1, test.group([], {"a" => {"$gt" => 1}}, {"count" => 0}, "function (obj, prev) { prev.count++; }")[0]["count"] :reduce => "function (obj, prev) { prev.count++; }")[0]["count"]
assert_equal 1, test.group(:cond => {"a" => {"$gt" => 1}},
:initial => {"count" => 0}, :reduce => "function (obj, prev) { prev.count++; }")[0]["count"]
assert_equal 1, test.group(:cond => {"a" => {"$gt" => 1}},
:initial => {"count" => 0}, :reduce => "function (obj, prev) { prev.count++; }")[0]["count"]
finalize = "function (obj) { obj.f = obj.count - 1; }" finalize = "function (obj) { obj.f = obj.count - 1; }"
assert_equal 2, test.group([], {}, {"count" => 0}, "function (obj, prev) { prev.count++; }", finalize)[0]["f"] assert_equal 2, test.group(:initial => {"count" => 0},
:reduce => "function (obj, prev) { prev.count++; }", :finalize => finalize)[0]["f"]
test.insert("a" => 2, "b" => 3) test.insert("a" => 2, "b" => 3)
expected = [{"a" => 2, "count" => 2}, expected = [{"a" => 2, "count" => 2},
{"a" => nil, "count" => 1}, {"a" => nil, "count" => 1},
{"a" => 1, "count" => 1}] {"a" => 1, "count" => 1}]
assert_equal expected, test.group(["a"], {}, {"count" => 0}, "function (obj, prev) { prev.count++; }") assert_equal expected, test.group(:key => ["a"], :initial => {"count" => 0},
assert_equal expected, test.group(["a"], {}, {"count" => 0}, "function (obj, prev) { prev.count++; }", true) :reduce => "function (obj, prev) { prev.count++; }")
assert_equal expected, test.group(:key => :a, :initial => {"count" => 0},
:reduce => "function (obj, prev) { prev.count++; }")
assert_raise OperationFailure do assert_raise OperationFailure do
test.group([], {}, {}, "5 ++ 5") test.group(:initial => {}, :reduce => "5 ++ 5")
end
assert_raise OperationFailure do
test.group([], {}, {}, "5 ++ 5", true)
end end
end end