add is_mongos @Property, so that mim won't error using an exhaust cursor

TurboGears · Mar 16, 2023 · ec8f1f6 · ec8f1f6
1 parent 51d8073
commit ec8f1f6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/ming/mim.py b/ming/mim.py
@@ -103,6 +103,10 @@ def _ensure_connected(self):
  def _is_writable(self):
  return True
 
+ @property
+ def is_mongos(self):
+ return False
+
 
 class Database(database.Database):
  def __init__(self, client, name, **__):

diff --git a/ming/tests/test_mim.py b/ming/tests/test_mim.py
@@ -7,7 +7,7 @@
 from bson.raw_bson import RawBSONDocument
 
 from ming import create_datastore, mim
-from pymongo import UpdateOne
+from pymongo import UpdateOne, CursorType
 from pymongo.errors import OperationFailure, DuplicateKeyError
 from unittest.mock import patch
 
@@ -155,6 +155,7 @@ def test_find_with_invalid_kwargs(self):
  self.assertRaises(TypeError, self.bind.db.coll.find, {'a': 2}, foo=123)
  self.assertRaises(TypeError, self.bind.db.coll.find_one, foo=123)
  self.bind.db.coll.find(allow_disk_use=True) # kwargs that pymongo knows are ok
+ self.bind.db.coll.find(cursor_type=CursorType.EXHAUST)
 
  def test_rewind(self):
  collection = self.bind.db.coll