ruff

strawberry/subscriptions/protocols/graphql_transport_ws/handlers.py

@@ -43,7 +43,6 @@
     from strawberry.subscriptions.protocols.graphql_transport_ws.types import (
         GraphQLTransportMessage,
     )
-
 
 
 class BaseGraphQLTransportWSHandler(ABC):
@@ -294,9 +293,9 @@ async def single_result() -> AsyncIterator[ExecutionResult]:
         self.operations[message.id] = operation
 
     async def operation_task(self, operation: Operation) -> None:
-        """
-        The operation task's top level method. Cleans-up and de-registers the operation
-        once it is done.
+        """The operation task's top level method.
+
+        Cleans-up and de-registers the operation once it is done.
         """
         task = asyncio.current_task()
         assert task is not None  # for type checkers

tests/websockets/test_graphql_transport_ws.py

@@ -1172,12 +1172,11 @@ async def test_task_error_handler(ws: WebSocketClient):
     """
     # can't use a simple Event here, because the handler may run
     # on a different thread
-    wakeup = False
+    wakeup = asyncio.Event()
 
     # a replacement method which causes an error in th eTask
     async def op(*args: Any, **kwargs: Any):
-        nonlocal wakeup
-        wakeup = True
+        wakeup.set()
         raise ZeroDivisionError("test")
 
     with patch.object(BaseGraphQLTransportWSHandler, "task_logger") as logger:
@@ -1193,8 +1192,7 @@ async def op(*args: Any, **kwargs: Any):
             )
 
             # wait for the error to be logged
-            while not wakeup:
-                await asyncio.sleep(0.01)
+            await wakeup.wait()
             # and another little bit, for the thread to finish
             await asyncio.sleep(0.01)
             assert logger.exception.called