From ec8f1f6ad83f97f0a9d57a6534018cf32ebfe452 Mon Sep 17 00:00:00 2001 From: Dave Brondsema Date: Thu, 16 Mar 2023 12:51:35 -0400 Subject: [PATCH] add is_mongos @property, so that mim won't error using an exhaust cursor --- ming/mim.py | 4 ++++ ming/tests/test_mim.py | 3 ++- 2 files changed, 6 insertions(+), 1 deletion(-) 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