From e2dfca2570ccdd8d6163c3759334a4b08e79112f Mon Sep 17 00:00:00 2001
From: Max Schmitt <max@schmitt.mx>
Date: Thu, 31 Oct 2024 10:54:14 +0100
Subject: [PATCH] chore: create WebSocket reply only calls using own loop

---
 playwright/_impl/_network.py | 37 +++++++++++++++++++++++++-----------
 1 file changed, 26 insertions(+), 11 deletions(-)

diff --git a/playwright/_impl/_network.py b/playwright/_impl/_network.py
index 649b89198..53f97a46c 100644
--- a/playwright/_impl/_network.py
+++ b/playwright/_impl/_network.py
@@ -541,14 +541,16 @@ async def _race_with_page_close(self, future: Coroutine) -> None:
             await asyncio.gather(fut, return_exceptions=True)
 
 
-def _create_task_and_ignore_exception(coro: Coroutine) -> None:
+def _create_task_and_ignore_exception(
+    loop: asyncio.AbstractEventLoop, coro: Coroutine
+) -> None:
     async def _ignore_exception() -> None:
         try:
             await coro
         except Exception:
             pass
 
-    asyncio.create_task(_ignore_exception())
+    loop.create_task(_ignore_exception())
 
 
 class ServerWebSocketRoute:
@@ -572,6 +574,7 @@ def url(self) -> str:
 
     def close(self, code: int = None, reason: str = None) -> None:
         _create_task_and_ignore_exception(
+            self._ws._loop,
             self._ws._channel.send(
                 "closeServer",
                 {
@@ -579,22 +582,24 @@ def close(self, code: int = None, reason: str = None) -> None:
                     "reason": reason,
                     "wasClean": True,
                 },
-            )
+            ),
         )
 
     def send(self, message: Union[str, bytes]) -> None:
         if isinstance(message, str):
             _create_task_and_ignore_exception(
+                self._ws._loop,
                 self._ws._channel.send(
                     "sendToServer", {"message": message, "isBase64": False}
-                )
+                ),
             )
         else:
             _create_task_and_ignore_exception(
+                self._ws._loop,
                 self._ws._channel.send(
                     "sendToServer",
                     {"message": base64.b64encode(message).decode(), "isBase64": True},
-                )
+                ),
             )
 
 
@@ -628,7 +633,9 @@ def _channel_message_from_page(self, event: Dict) -> None:
                 else event["message"]
             )
         elif self._connected:
-            _create_task_and_ignore_exception(self._channel.send("sendToServer", event))
+            _create_task_and_ignore_exception(
+                self._loop, self._channel.send("sendToServer", event)
+            )
 
     def _channel_message_from_server(self, event: Dict) -> None:
         if self._on_server_message:
@@ -638,19 +645,25 @@ def _channel_message_from_server(self, event: Dict) -> None:
                 else event["message"]
             )
         else:
-            _create_task_and_ignore_exception(self._channel.send("sendToPage", event))
+            _create_task_and_ignore_exception(
+                self._loop, self._channel.send("sendToPage", event)
+            )
 
     def _channel_close_page(self, event: Dict) -> None:
         if self._on_page_close:
             self._on_page_close(event["code"], event["reason"])
         else:
-            _create_task_and_ignore_exception(self._channel.send("closeServer", event))
+            _create_task_and_ignore_exception(
+                self._loop, self._channel.send("closeServer", event)
+            )
 
     def _channel_close_server(self, event: Dict) -> None:
         if self._on_server_close:
             self._on_server_close(event["code"], event["reason"])
         else:
-            _create_task_and_ignore_exception(self._channel.send("closePage", event))
+            _create_task_and_ignore_exception(
+                self._loop, self._channel.send("closePage", event)
+            )
 
     @property
     def url(self) -> str:
@@ -674,19 +687,21 @@ def connect_to_server(self) -> "WebSocketRoute":
     def send(self, message: Union[str, bytes]) -> None:
         if isinstance(message, str):
             _create_task_and_ignore_exception(
+                self._loop,
                 self._channel.send(
                     "sendToPage", {"message": message, "isBase64": False}
-                )
+                ),
             )
         else:
             _create_task_and_ignore_exception(
+                self._loop,
                 self._channel.send(
                     "sendToPage",
                     {
                         "message": base64.b64encode(message).decode(),
                         "isBase64": True,
                     },
-                )
+                ),
             )
 
     def on_message(self, handler: Callable[[Union[str, bytes]], Any]) -> None: