Call close before raising (Async)StopIteration

Author: NoahStapp

pymongo/asynchronous/cursor.py

@@ -1263,10 +1263,14 @@ async def next(self) -> _DocumentType:
             self._exhaust_checked = True
             await self._supports_exhaust()
         if self._empty:
+            if self._cursor_type == CursorType.NON_TAILABLE:
+                await self.close()
             raise StopAsyncIteration
         if len(self._data) or await self._refresh():
             return self._data.popleft()
         else:
+            if self._cursor_type == CursorType.NON_TAILABLE:
+                await self.close()
             raise StopAsyncIteration
 
     async def _next_batch(self, result: list, total: Optional[int] = None) -> bool:  # type: ignore[type-arg]

pymongo/synchronous/cursor.py

@@ -1261,10 +1261,14 @@ def next(self) -> _DocumentType:
             self._exhaust_checked = True
             self._supports_exhaust()
         if self._empty:
+            if self._cursor_type == CursorType.NON_TAILABLE:
+                self.close()
             raise StopIteration
         if len(self._data) or self._refresh():
             return self._data.popleft()
         else:
+            if self._cursor_type == CursorType.NON_TAILABLE:
+                self.close()
             raise StopIteration
 
     def _next_batch(self, result: list, total: Optional[int] = None) -> bool:  # type: ignore[type-arg]