webdriver/tests/interop/frames.py

import pytest
from webdriver.error import NoSuchWindowException

from tests.support.sync import AsyncPoll

pytestmark = pytest.mark.asyncio


async def test_classic_switch_to_parent_no_browsing_context(bidi_session, current_session, url):
    # With WebDriver classic it cannot be checked if the parent frame is already
    # gone before switching to it. To prevent race conditions such a check needs
    # to be done via WebDriver BiDi.
    current_session.url = url("/webdriver/tests/support/html/frames.html")

    subframe = current_session.find.css("#sub-frame", all=False)
    current_session.switch_to_frame(subframe)

    deleteframe = current_session.find.css("#delete-frame", all=False)
    current_session.switch_to_frame(deleteframe)

    button = current_session.find.css("#remove-top", all=False)
    button.click()

    # Wait until iframe is gone.
    async def is_frame_removed(_):
        contexts = await bidi_session.browsing_context.get_tree(root=current_session.window_handle)
        assert not contexts[0]["children"], "iframe that should be closed is still open"

    wait = AsyncPoll(current_session)
    await wait.until(is_frame_removed)

    with pytest.raises(NoSuchWindowException):
        current_session.switch_to_parent_frame()