[Feature Request]: Expect Page Load context manager

[dickermoshe]

Checklist before requesting

• I have searched for similar feature requests and didn't find a duplicate.
• I have checked the documentation to confirm this feature doesn't already exist.

Problem Statement

Many times a button click should trigger a page load and you need to wait for the new page to be loaded.
You can wait for this by waiting for a certain element to be visible on the next page, but many times this is not possible.

In my case it is because besides for some changed cookies, the next page is completely identical to the 1st page.

Proposed Solution

I would love to have an async context manager which only returns when a load event is fired.
This is how I'm using it in my code.

@asynccontextmanager
async def expect_page_load(self, timeout: int = 10):
    event = asyncio.Event()

    async def event_handler(_):
        event.set()

    if self.page_events_enabled is False:
        _before_page_events_enabled = False
        await self.enable_page_events()
    else:
        _before_page_events_enabled = True

    await self.on(
        PageEvents.PAGE_LOADED, event_handler, temporary=True
    )

    yield

    await asyncio.wait_for(event.wait(), timeout=timeout)

    if _before_page_events_enabled is False:
        await self.disable_page_events()

Usage would look like

async with page.expect_page_load():
    ... # Do something which  should trigger a reload

# Continues here upon reload, or throws if timed out

Alternatives Considered

No response

Additional Context

No response

Importance

Important

Contribution

Yes, I'd be willing to implement this feature