mozilla-central/testing/web-platform/tests/webdriver/tests/bidi/browsing_context/get_tree/frames.py

Source code

Revision control

Copy as Markdown

Other Tools

Test Info: Warnings

import pytest
from tests.support.sync import AsyncPoll
from .. import assert_browsing_context
pytestmark = pytest.mark.asyncio
async def test_multiple_frames(
bidi_session,
top_context,
test_page,
test_page2,
test_page_multiple_frames,
):
await bidi_session.browsing_context.navigate(
context=top_context["context"], url=test_page_multiple_frames, wait="complete"
)
# First retrieve all browsing contexts of the first tab
top_level_context_id = top_context["context"]
all_contexts = await bidi_session.browsing_context.get_tree(root=top_level_context_id)
assert len(all_contexts) == 1
root_info = all_contexts[0]
assert_browsing_context(
root_info,
top_level_context_id,
children=2,
parent=None,
url=test_page_multiple_frames,
)
child1_info = root_info["children"][0]
assert_browsing_context(
child1_info,
context=None,
children=0,
is_root=False,
parent=None,
url=test_page,
)
assert child1_info["context"] != root_info["context"]
child2_info = root_info["children"][1]
assert_browsing_context(
child2_info,
context=None,
children=0,
is_root=False,
parent=None,
url=test_page2,
)
assert child2_info["context"] != root_info["context"]
assert child2_info["context"] != child1_info["context"]
async def test_cross_origin(
bidi_session,
top_context,
test_page_cross_origin,
test_page_cross_origin_frame,
):
await bidi_session.browsing_context.navigate(
context=top_context["context"], url=test_page_cross_origin_frame, wait="complete"
)
# First retrieve all browsing contexts of the first tab
top_level_context_id = top_context["context"]
all_contexts = await bidi_session.browsing_context.get_tree(root=top_level_context_id)
assert len(all_contexts) == 1
root_info = all_contexts[0]
assert_browsing_context(
root_info,
top_level_context_id,
children=1,
parent=None,
url=test_page_cross_origin_frame,
)
child1_info = root_info["children"][0]
assert_browsing_context(
child1_info,
context=None,
children=0,
is_root=False,
parent=None,
url=test_page_cross_origin,
)
assert child1_info["context"] != root_info["context"]
@pytest.mark.parametrize("user_context", ["default", "new"])
@pytest.mark.parametrize("domain", ["", "alt"], ids=["same_origin", "cross_origin"])
async def test_user_context(
bidi_session,
create_user_context,
subscribe_events,
wait_for_event,
inline,
user_context,
domain,
):
await subscribe_events(["browsingContext.load"])
user_context_id = (
await create_user_context() if user_context == "new" else user_context
)
iframe_url_1 = inline("<div>foo</div>", domain=domain)
iframe_url_2 = inline("<div>bar</div>", domain=domain)
page_url = inline(
f"<iframe src='{iframe_url_1}'></iframe><iframe src='{iframe_url_2}'></iframe>"
)
context = await bidi_session.browsing_context.create(
type_hint="tab", user_context=user_context_id
)
# Record all load events.
events = []
async def on_event(method, data):
events.append(data)
remove_listener = bidi_session.add_event_listener("browsingContext.load", on_event)
await bidi_session.browsing_context.navigate(
context=context["context"], url=page_url, wait="complete"
)
# Wait until all iframes have been loaded.
wait = AsyncPoll(bidi_session, timeout=2)
await wait.until(lambda _: len(events) >= 3)
top_level_context_id = context["context"]
all_contexts = await bidi_session.browsing_context.get_tree(
root=top_level_context_id
)
assert len(all_contexts) == 1
root_info = all_contexts[0]
assert_browsing_context(
root_info,
top_level_context_id,
children=2,
parent=None,
url=page_url,
user_context=user_context_id,
)
# The contexts can be returned in any order, find the info matching iframe_url_1
child1_info = next(
filter(lambda x: x["url"] == iframe_url_1, root_info["children"]), None
)
assert child1_info is not None
assert_browsing_context(
child1_info,
context=None,
children=0,
is_root=False,
parent=None,
url=iframe_url_1,
user_context=user_context_id,
)
assert child1_info["context"] != root_info["context"]
child2_info = next(
filter(lambda x: x["url"] == iframe_url_2, root_info["children"]), None
)
assert child2_info is not None
assert_browsing_context(
child2_info,
context=None,
children=0,
is_root=False,
parent=None,
url=iframe_url_2,
user_context=user_context_id,
)
assert child2_info["context"] != root_info["context"]
assert child2_info["context"] != child1_info["context"]