From 3a3c740457c6845e57c5949ec339c83cc2772f96 Mon Sep 17 00:00:00 2001
From: Emily Stolfo <emily@mongodb.com>
Date: Tue, 9 Dec 2014 12:50:31 +0100
Subject: [PATCH] Remove use of cursorInfo in tests for versions > 2.6

---
 test/functional/cursor_test.rb | 34 +++++++++++++++++-----------------
 1 file changed, 17 insertions(+), 17 deletions(-)

diff --git a/test/functional/cursor_test.rb b/test/functional/cursor_test.rb
index 17dc15a5d8..ecbeb3f09f 100644
--- a/test/functional/cursor_test.rb
+++ b/test/functional/cursor_test.rb
@@ -468,21 +468,19 @@ def test_close_after_query_sent
   def test_kill_cursors
     @coll.drop
 
-    client_cursors = @db.command("cursorInfo" => 1)["clientCursors_size"]
+    client_cursors = cursor_count(@db)
 
     10000.times do |i|
       @coll.insert("i" => i)
     end
 
-    assert_equal(client_cursors,
-                 @db.command("cursorInfo" => 1)["clientCursors_size"])
+    assert_equal(client_cursors, cursor_count(@db))
 
     10.times do |i|
       @coll.find_one()
     end
 
-    assert_equal(client_cursors,
-                 @db.command("cursorInfo" => 1)["clientCursors_size"])
+    assert_equal(client_cursors, cursor_count(@db))
 
     10.times do |i|
       a = @coll.find()
@@ -490,38 +488,32 @@ def test_kill_cursors
       a.close()
     end
 
-    assert_equal(client_cursors,
-                 @db.command("cursorInfo" => 1)["clientCursors_size"])
+    assert_equal(client_cursors, cursor_count(@db))
 
     a = @coll.find()
     a.next_document
 
-    assert_not_equal(client_cursors,
-                     @db.command("cursorInfo" => 1)["clientCursors_size"])
+    assert_not_equal(client_cursors, cursor_count(@db))
 
     a.close()
 
-    assert_equal(client_cursors,
-                 @db.command("cursorInfo" => 1)["clientCursors_size"])
+    assert_equal(client_cursors, cursor_count(@db))
 
     a = @coll.find({}, :limit => 10).next_document
 
-    assert_equal(client_cursors,
-                 @db.command("cursorInfo" => 1)["clientCursors_size"])
+    assert_equal(client_cursors, cursor_count(@db))
 
     @coll.find() do |cursor|
       cursor.next_document
     end
 
-    assert_equal(client_cursors,
-                 @db.command("cursorInfo" => 1)["clientCursors_size"])
+    assert_equal(client_cursors, cursor_count(@db))
 
     @coll.find() { |cursor|
       cursor.next_document
     }
 
-    assert_equal(client_cursors,
-                 @db.command("cursorInfo" => 1)["clientCursors_size"])
+    assert_equal(client_cursors, cursor_count(@db))
   end
 
   def test_count_with_fields
@@ -680,4 +672,12 @@ def test_instance_transformation_with_each
       assert_instance_of(klass, instance)
     end
   end
+
+  def cursor_count(db)
+    if @version > '2.6.0'
+      db.command("serverStatus" => 1)["metrics"]["cursor"]["open"]["total"]
+    else
+      db.command("cursorInfo" => 1)["clientCursors_size"]
+    end
+  end
 end