PYTHON-4533 - Convert test/test_cursor.py to async (mongodb#1731)

Author: NoahStapp

pymongo/asynchronous/command_cursor.py

@@ -424,4 +424,4 @@ def _unpack_response( # type: ignore[override]
  return raw_response # type: ignore[return-value]
 
  def __getitem__(self, index: int) -> NoReturn:
- raise InvalidOperation("Cannot call __getitem__ on RawBatchCursor")
+ raise InvalidOperation("Cannot call __getitem__ on AsyncRawBatchCommandCursor")

pymongo/asynchronous/cursor.py

@@ -242,6 +242,7 @@ def __init__(
  self._exhaust_checked = True
  self._supports_exhaust() # type: ignore[unused-coroutine]
  else:
+ self._exhaust = cursor_type == CursorType.EXHAUST
  self._exhaust_checked = False
 
  async def _supports_exhaust(self) -> None:
@@ -1318,4 +1319,4 @@ async def explain(self) -> _DocumentType:
  return await clone.explain()
 
  def __getitem__(self, index: Any) -> NoReturn:
- raise InvalidOperation("Cannot call __getitem__ on RawBatchCursor")
+ raise InvalidOperation("Cannot call __getitem__ on AsyncRawBatchCursor")

pymongo/synchronous/command_cursor.py

@@ -424,4 +424,4 @@ def _unpack_response( # type: ignore[override]
  return raw_response # type: ignore[return-value]
 
  def __getitem__(self, index: int) -> NoReturn:
- raise InvalidOperation("Cannot call __getitem__ on RawBatchCursor")
+ raise InvalidOperation("Cannot call __getitem__ on RawBatchCommandCursor")

pymongo/synchronous/cursor.py

@@ -242,6 +242,7 @@ def __init__(
  self._exhaust_checked = True
  self._supports_exhaust() # type: ignore[unused-coroutine]
  else:
+ self._exhaust = cursor_type == CursorType.EXHAUST
  self._exhaust_checked = False
 
  def _supports_exhaust(self) -> None: