Low-level API: endpoints should always clear the receiver buffer

src/easynetwork/lowlevel/api_async/endpoints/stream.py

@@ -94,8 +94,10 @@ async def aclose(self) -> None:
         """
         Closes the endpoint.
         """
-        await self.__transport.aclose()
-        self.__receiver.clear()
+        try:
+            await self.__transport.aclose()
+        finally:
+            self.__receiver.clear()
 
     async def recv_packet(self) -> _T_ReceivedPacket:
         """
@@ -269,8 +271,10 @@ async def aclose(self) -> None:
         Closes the endpoint.
         """
         with self.__send_guard:
-            await self.__transport.aclose()
-            self.__receiver.clear()
+            try:
+                await self.__transport.aclose()
+            finally:
+                self.__receiver.clear()
 
     async def send_packet(self, packet: _T_SentPacket) -> None:
         """

src/easynetwork/lowlevel/api_async/servers/stream.py

@@ -220,6 +220,8 @@ async def __client_coroutine(
                 case _:  # pragma: no cover
                     assert_never(self.__protocol)
 
+            # NOTE: It is safe to clear the consumer before the transport here.
+            #       There is no task reading the transport at this point.
             task_exit_stack.callback(consumer.clear)
 
             request_handler_generator = client_connected_cb(

src/easynetwork/lowlevel/api_sync/endpoints/stream.py

@@ -92,8 +92,10 @@ def close(self) -> None:
         """
         Closes the endpoint.
         """
-        self.__transport.close()
-        self.__receiver.clear()
+        try:
+            self.__transport.close()
+        finally:
+            self.__receiver.clear()
 
     def recv_packet(self, *, timeout: float | None = None) -> _T_ReceivedPacket:
         """
@@ -270,8 +272,10 @@ def close(self) -> None:
         """
         Closes the endpoint.
         """
-        self.__transport.close()
-        self.__receiver.clear()
+        try:
+            self.__transport.close()
+        finally:
+            self.__receiver.clear()
 
     def send_packet(self, packet: _T_SentPacket, *, timeout: float | None = None) -> None:
         """