[wdspec] Add tests for "emulation.setScreenOrientationOverride" command in iframes.

Differential Revision: https://phabricator.services.mozilla.com/D262789

bugzilla-url: https://bugzilla.mozilla.org/show_bug.cgi?id=1974167
gecko-commit: 573eeceeda30046a018b52c57e7404e76e95fe49
gecko-reviewers: webdriver-reviewers, jdescottes

Author: lutien

webdriver/tests/bidi/emulation/set_screen_orientation_override/conftest.py

@@ -8,11 +8,12 @@
 
 @pytest_asyncio.fixture
 async def get_screen_orientation(bidi_session):
-    async def get_screen_orientation(context):
+    async def get_screen_orientation(context, top_context=None):
         # Activation is required, as orientation is only available on an active
         # context.
-        await bidi_session.browsing_context.activate(context=context["context"])
-
+        await bidi_session.browsing_context.activate(
+            context=top_context["context"] if top_context else context["context"]
+        )
         result = await bidi_session.script.evaluate(
             expression="""({
                     angle: screen.orientation.angle,

webdriver/tests/bidi/emulation/set_screen_orientation_override/contexts.py

@@ -67,3 +67,55 @@ async def test_multiple_contexts(
     assert await get_screen_orientation(new_tab) == default_screen_orientation
     assert await get_screen_orientation(
         top_context) == default_screen_orientation
+
+
+@pytest.mark.parametrize("domain", ["", "alt"], ids=["same_origin", "cross_origin"])
+async def test_iframe(
+    bidi_session,
+    new_tab,
+    get_screen_orientation,
+    some_bidi_screen_orientation,
+    some_web_screen_orientation,
+    another_bidi_screen_orientation,
+    another_web_screen_orientation,
+    inline,
+    domain,
+):
+    await bidi_session.emulation.set_screen_orientation_override(
+        contexts=[new_tab["context"]],
+        screen_orientation=some_bidi_screen_orientation,
+    )
+
+    # Assert screen orientation in the required context.
+    assert await get_screen_orientation(new_tab) == some_web_screen_orientation
+
+    iframe_url = inline("<div id='in-iframe'>foo</div>", domain=domain)
+    page_url = inline(f"<iframe src='{iframe_url}'></iframe>")
+
+    # Load the page with iframes.
+    await bidi_session.browsing_context.navigate(
+        context=new_tab["context"],
+        url=page_url,
+        wait="complete",
+    )
+
+    contexts = await bidi_session.browsing_context.get_tree(root=new_tab["context"])
+    iframe = contexts[0]["children"][0]
+
+    # Assert locale is emulated in the iframe context.
+    assert (
+        await get_screen_orientation(iframe, top_context=new_tab)
+        == some_web_screen_orientation
+    )
+
+    # Set another screen orientation override.
+    await bidi_session.emulation.set_screen_orientation_override(
+        contexts=[new_tab["context"]],
+        screen_orientation=another_bidi_screen_orientation,
+    )
+
+    # Assert screen orientation is emulated in the iframe context.
+    assert (
+        await get_screen_orientation(iframe, top_context=new_tab)
+        == another_web_screen_orientation
+    )

webdriver/tests/bidi/emulation/set_screen_orientation_override/screen_orientation.py

@@ -1,6 +1,7 @@
 import pytest
 
 from . import get_angle
+from ... import remote_mapping_to_dict
 
 pytestmark = pytest.mark.asyncio
 
@@ -32,3 +33,76 @@ async def test_screen_orientation(bidi_session, top_context,
     # Assert screen orientation is the default.
     assert await get_screen_orientation(
         top_context) == default_screen_orientation
+
+
+async def test_screen_orientation_change_event(
+    bidi_session,
+    new_tab,
+    some_bidi_screen_orientation,
+    some_web_screen_orientation,
+    another_bidi_screen_orientation,
+    another_web_screen_orientation,
+    default_screen_orientation,
+    subscribe_events,
+    add_preload_script,
+    wait_for_event,
+    wait_for_future_safe,
+    inline,
+):
+    await subscribe_events(["script.message"])
+    await add_preload_script(
+        function_declaration="""(channel) => {
+            window.screen.orientation.addEventListener(
+                "change",
+                (e)=>channel({type: e.target.type, angle: e.target.angle})
+            )
+        }""",
+        arguments=[{"type": "channel", "value": {"channel": "change_event"}}],
+    )
+
+    await bidi_session.browsing_context.navigate(
+        context=new_tab["context"], url=inline("<div>foo</div>"), wait="complete"
+    )
+
+    on_script_message = wait_for_event("script.message")
+
+    # Set screen orientation override.
+    await bidi_session.emulation.set_screen_orientation_override(
+        contexts=[new_tab["context"]], screen_orientation=some_bidi_screen_orientation
+    )
+
+    event_data = await wait_for_future_safe(on_script_message)
+
+    assert (
+        remote_mapping_to_dict(event_data["data"]["value"])
+        == some_web_screen_orientation
+    )
+
+    on_script_message = wait_for_event("script.message")
+
+    # Set another screen orientation override.
+    await bidi_session.emulation.set_screen_orientation_override(
+        contexts=[new_tab["context"]],
+        screen_orientation=another_bidi_screen_orientation,
+    )
+
+    event_data = await wait_for_future_safe(on_script_message)
+
+    assert (
+        remote_mapping_to_dict(event_data["data"]["value"])
+        == another_web_screen_orientation
+    )
+
+    on_script_message = wait_for_event("script.message")
+
+    # Reset screen orientation override.
+    await bidi_session.emulation.set_screen_orientation_override(
+        contexts=[new_tab["context"]], screen_orientation=None
+    )
+
+    event_data = await wait_for_future_safe(on_script_message)
+
+    assert (
+        remote_mapping_to_dict(event_data["data"]["value"])
+        == default_screen_orientation
+    )