Add 'with' statement support to cursors PYTHON-39

Author: Mher Movsisyan

pymongo/cursor.py

@@ -611,3 +611,9 @@ def next(self):
  else:
  raise StopIteration
  return next
+ 
+ def __enter__(self):
+ return self
+ 
+ def __exit__(self, type, value, traceback):
+ self.__die()

test/test_cursor.py

@@ -664,6 +664,12 @@ def test_max_scan(self):
  self.assertEqual(50, len(list(self.db.test.find()
  .max_scan(90).max_scan(50))))
 
+ def test_with_statement(self):
+ c1 = self.db.test.find()
+ with self.db.test.find() as c2:
+ self.assertTrue(c2.alive)
+ self.assertTrue(c1.alive)
+ self.assertFalse(c2.alive)
 
 if __name__ == "__main__":
  unittest.main()