AsyncSocket: shutdown() was not checking current running loop (#172)

pyproject.toml

@@ -202,7 +202,7 @@ enable_error_code = ["truthy-bool", "ignore-without-code", "unused-awaitable"]
 
 [tool.pytest.ini_options]
 asyncio_mode = "strict"  # Avoid some unwanted behaviour
-addopts = "--strict-markers"
+addopts = "--strict-markers -p 'no:anyio'"  # hatch CLI dependencies installs anyio
 minversion = "7.1.2"
 testpaths = ["tests"]
 norecursedirs = ["scripts"]

src/easynetwork/lowlevel/asyncio/socket.py

@@ -160,14 +160,17 @@ async def recvfrom(self, bufsize: int, /) -> tuple[bytes, _socket._RetAddress]:
             return await self.__loop.sock_recvfrom(socket, bufsize)
 
     async def shutdown(self, how: int, /) -> None:
-        socket: _socket.socket = self.__check_not_closed()
+        # Checks if we are within the bound loop
+        TaskUtils.check_current_asyncio_task(self.__loop)
+
         if how in {_socket.SHUT_RDWR, _socket.SHUT_WR}:
             while (waiter := self.__waiters.get("send")) is not None:
                 try:
                     await asyncio.shield(waiter)
                 finally:
-                    waiter = None  # Breack cyclic reference with raised exception
+                    waiter = None  # Break cyclic reference with raised exception
 
+        socket: _socket.socket = self.__check_not_closed()
         socket.shutdown(how)
         await asyncio.sleep(0)
 
@@ -177,9 +180,9 @@ def __conflict_detection(self, task_id: _SocketTaskId, *, abort_errno: int = _er
             raise _utils.error_from_errno(_errno.EBUSY)
 
         # Checks if we are within the bound loop
-        TaskUtils.current_asyncio_task(self.__loop)  # type: ignore[unused-awaitable]
+        TaskUtils.check_current_asyncio_task(self.__loop)
 
-        with CancelScope() as scope, contextlib.ExitStack() as stack:
+        with contextlib.ExitStack() as stack, CancelScope() as scope:
             self.__scopes.add(scope)
             stack.callback(self.__scopes.discard, scope)
 

src/easynetwork/lowlevel/asyncio/tasks.py

@@ -310,6 +310,10 @@ def __cancel_task_unless_done(task: asyncio.Task[Any], cancel_msg: str | None) -
 
 @final
 class TaskUtils:
+    @staticmethod
+    def check_current_asyncio_task(loop: asyncio.AbstractEventLoop | None = None) -> None:
+        _ = TaskUtils.current_asyncio_task(loop=loop)
+
     @staticmethod
     def current_asyncio_task(loop: asyncio.AbstractEventLoop | None = None) -> asyncio.Task[Any]:
         t: asyncio.Task[Any] | None = asyncio.current_task(loop=loop)