diff --git a/ming/mim.py b/ming/mim.py index 1a4ec16..98e631f 100644 --- 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 index 418d4ab..e43953d 100644 --- 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