fix: migrate to new waitForEventInfo handling

playwright/_impl/_connection.py

@@ -99,31 +99,6 @@ def __init__(
         if self._parent:
             self._parent._objects[guid] = self
 
-    def _wait_for_event_info_before(self, wait_id: str, api_name: str) -> None:
-        self._connection._send_message_to_server(
-            self._guid,
-            "waitForEventInfo",
-            {
-                "info": {
-                    "apiName": api_name,
-                    "waitId": wait_id,
-                    "phase": "before",
-                }
-            },
-        )
-
-    def _wait_for_event_info_after(
-        self, wait_id: str, exception: Exception = None
-    ) -> None:
-        info = {"waitId": wait_id, "phase": "after"}
-        if exception:
-            info["error"] = str(exception)
-        self._connection._send_message_to_server(
-            self._guid,
-            "waitForEventInfo",
-            {"info": info},
-        )
-
     def _dispose(self) -> None:
         # Clean up from parent and connection.
         if self._parent:

playwright/_impl/_frame.py

@@ -122,7 +122,7 @@ async def goto(
     def _setup_navigation_wait_helper(
         self, wait_name: str, timeout: float = None
     ) -> WaitHelper:
-        wait_helper = WaitHelper(self, f"frame.{wait_name}")
+        wait_helper = WaitHelper(self._page, f"frame.{wait_name}")
         wait_helper.reject_on_event(
             self._page, "close", Error("Navigation failed because page was closed!")
         )

playwright/_impl/_wait_helper.py

@@ -24,14 +24,41 @@
 
 
 class WaitHelper:
-    def __init__(self, channel_owner: ChannelOwner, api_name: str) -> None:
+    def __init__(self, channel_owner: ChannelOwner, event: str) -> None:
         self._result: asyncio.Future = asyncio.Future()
         self._wait_id = uuid.uuid4().hex
         self._loop = channel_owner._loop
         self._pending_tasks: List[Task] = []
-        self._channel_owner = channel_owner
+        self._channel = channel_owner._channel
         self._registered_listeners: List[Tuple[EventEmitter, str, Callable]] = []
-        channel_owner._wait_for_event_info_before(self._wait_id, api_name)
+        self._wait_for_event_info_before(self._wait_id, event)
+
+    def _wait_for_event_info_before(self, wait_id: str, event: str) -> None:
+        self._channel.send_no_reply(
+            "waitForEventInfo",
+            {
+                "info": {
+                    "waitId": wait_id,
+                    "phase": "before",
+                    "event": event,
+                }
+            },
+        )
+
+    def _wait_for_event_info_after(self, wait_id: str, error: Exception = None) -> None:
+        try:
+            self._channel.send_no_reply(
+                "waitForEventInfo",
+                {
+                    "info": {
+                        "waitId": wait_id,
+                        "phase": "after",
+                        "error": str(error) if error else None,
+                    }
+                },
+            )
+        except Exception:
+            pass
 
     def reject_on_event(
         self,
@@ -68,13 +95,13 @@ def _fulfill(self, result: Any) -> None:
         self._cleanup()
         if not self._result.done():
             self._result.set_result(result)
-        self._channel_owner._wait_for_event_info_after(self._wait_id)
+        self._wait_for_event_info_after(self._wait_id)
 
     def _reject(self, exception: Exception) -> None:
         self._cleanup()
         if not self._result.done():
             self._result.set_exception(exception)
-        self._channel_owner._wait_for_event_info_after(self._wait_id, exception)
+        self._wait_for_event_info_after(self._wait_id, exception)
 
     def wait_for_event(
         self,

tests/assets/networkidle.html

@@ -0,0 +1 @@
+<script src='networkidle.js'></script>

tests/assets/networkidle.js

@@ -0,0 +1,12 @@
+async function main() {
+  window.ws = new WebSocket('ws://localhost:' + window.location.port + '/ws');
+  window.ws.addEventListener('message', message => {});
+
+  fetch('fetch-request-a.js');
+  window.top.fetchSecond = () => {
+    // Do not return the promise here.
+    fetch('fetch-request-b.js');
+  };
+}
+
+main();

tests/async/test_navigation.py

@@ -19,7 +19,8 @@
 
 import pytest
 
-from playwright.async_api import Error, Request, TimeoutError
+from playwright.async_api import Error, Page, Request, TimeoutError
+from tests.server import Server
 
 
 async def test_goto_should_work(page, server):
@@ -676,6 +677,14 @@ def handler(request: Any):
     await page.wait_for_load_state("domcontentloaded")
 
 
+async def test_wait_for_load_state_networkidle(page: Page, server: Server):
+    wait_for_network_idle_future = asyncio.create_task(
+        page.wait_for_load_state("networkidle")
+    )
+    await page.goto(server.PREFIX + "/networkidle.html")
+    await wait_for_network_idle_future
+
+
 async def test_wait_for_load_state_should_work_with_pages_that_have_loaded_before_being_connected_to(
     page, context, server
 ):