use cursorInfo command instead of intr collection

Mike Dirolf · Mike Dirolf · commit 33e563147e4d · 2009-02-23T18:08:52.000-05:00
diff --git a/test/test_cursor.py b/test/test_cursor.py
@@ -282,48 +282,47 @@ def test_where(self):
     def test_kill_cursors(self):
         db = self.db
         db.drop_collection("test")
-        intr = self.db.connection().intr
 
-        client_cursors = intr.cursors.find_one()["clientCursors_size"]
-        by_location = intr.cursors.find_one()["byLocation_size"]
+        client_cursors = db._command({"cursorInfo": 1})["clientCursors_size"]
+        by_location = db._command({"cursorInfo": 1})["byLocation_size"]
 
         for i in range(10000):
             db.test.insert({"i": i})
 
-        self.assertEqual(client_cursors, intr.cursors.find_one()["clientCursors_size"])
-        self.assertEqual(by_location, intr.cursors.find_one()["byLocation_size"])
+        self.assertEqual(client_cursors, db._command({"cursorInfo": 1})["clientCursors_size"])
+        self.assertEqual(by_location, db._command({"cursorInfo": 1})["byLocation_size"])
 
         for _ in range(10):
             db.test.find_one()
 
-        self.assertEqual(client_cursors, intr.cursors.find_one()["clientCursors_size"])
-        self.assertEqual(by_location, intr.cursors.find_one()["byLocation_size"])
+        self.assertEqual(client_cursors, db._command({"cursorInfo": 1})["clientCursors_size"])
+        self.assertEqual(by_location, db._command({"cursorInfo": 1})["byLocation_size"])
 
         for _ in range(10):
             for x in db.test.find():
                 break
 
-        self.assertEqual(client_cursors, intr.cursors.find_one()["clientCursors_size"])
-        self.assertEqual(by_location, intr.cursors.find_one()["byLocation_size"])
+        self.assertEqual(client_cursors, db._command({"cursorInfo": 1})["clientCursors_size"])
+        self.assertEqual(by_location, db._command({"cursorInfo": 1})["byLocation_size"])
 
         a = db.test.find()
         for x in a:
             break
 
-        self.assertNotEqual(client_cursors, intr.cursors.find_one()["clientCursors_size"])
-        self.assertNotEqual(by_location, intr.cursors.find_one()["byLocation_size"])
+        self.assertNotEqual(client_cursors, db._command({"cursorInfo": 1})["clientCursors_size"])
+        self.assertNotEqual(by_location, db._command({"cursorInfo": 1})["byLocation_size"])
 
         del a
 
-        self.assertEqual(client_cursors, intr.cursors.find_one()["clientCursors_size"])
-        self.assertEqual(by_location, intr.cursors.find_one()["byLocation_size"])
+        self.assertEqual(client_cursors, db._command({"cursorInfo": 1})["clientCursors_size"])
+        self.assertEqual(by_location, db._command({"cursorInfo": 1})["byLocation_size"])
 
         a = db.test.find().limit(10)
         for x in a:
             break
 
-        self.assertEqual(client_cursors, intr.cursors.find_one()["clientCursors_size"])
-        self.assertEqual(by_location, intr.cursors.find_one()["byLocation_size"])
+        self.assertEqual(client_cursors, db._command({"cursorInfo": 1})["clientCursors_size"])
+        self.assertEqual(by_location, db._command({"cursorInfo": 1})["byLocation_size"])
 
 if __name__ == "__main__":
     unittest.main()
