feat: added close to supported methods for cursors

michaelkryukov · Feb 26, 2024 · e11fd2a · e11fd2a
1 parent c50c6e1
commit e11fd2a
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/mongomock_motor/__init__.py b/mongomock_motor/__init__.py
@@ -91,6 +91,12 @@ def wrapper(self, *args, **kwargs):
         'where',
     ],
 )
+@with_async_methods(
+    '__cursor',
+    [
+        'close',
+    ],
+)
 class AsyncCursor:
     def __init__(self, cursor):
         self.__cursor = cursor
@@ -178,6 +184,12 @@ async def to_list(self, *args, **kwargs):
         'update_one',
     ],
 )
+@with_async_methods(
+    '__cursor',
+    [
+        'close',
+    ],
+)
 class AsyncMongoMockCollection:
     def __init__(self, database, collection):
         self.database = database