[wdspec] Add tests for "userContexts" argument for "browsingContext.setViewport" command.

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

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

Author: lutien

Diff for: webdriver/tests/bidi/browsing_context/set_viewport/device_pixel_ratio.py

@@ -22,6 +22,15 @@ async def test_device_pixel_ratio_only(bidi_session, inline, new_tab, device_pix
     assert await get_device_pixel_ratio(bidi_session, new_tab) == device_pixel_ratio
     assert await get_viewport_dimensions(bidi_session, new_tab) == viewport
 
+    url = inline("<div>foo</div>", domain="alt")
+    await bidi_session.browsing_context.navigate(
+        context=new_tab["context"], url=url, wait="complete"
+    )
+
+    # Make sure that overrides are applied after cross-origin navigation.
+    assert await get_device_pixel_ratio(bidi_session, new_tab) == device_pixel_ratio
+    assert await get_viewport_dimensions(bidi_session, new_tab) == viewport
+
 
 @pytest.mark.asyncio
 @pytest.mark.parametrize("device_pixel_ratio", [0.5, 2])

Diff for: webdriver/tests/bidi/browsing_context/set_viewport/invalid.py

@@ -6,7 +6,7 @@
 pytestmark = pytest.mark.asyncio
 
 
-@pytest.mark.parametrize("value", [None, False, 42, {}, []])
+@pytest.mark.parametrize("value", [False, 42, {}, []])
 async def test_params_context_invalid_type(bidi_session, value):
     with pytest.raises(error.InvalidArgumentException):
         await bidi_session.browsing_context.set_viewport(context=value, viewport={
@@ -89,3 +89,72 @@ async def test_params_devicePixelRatio_invalid_value(bidi_session, new_tab, devi
             device_pixel_ratio=device_pixel_ratio,
             viewport=None
         )
+
+
+@pytest.mark.parametrize("value", [True, "foo", 42, {}])
+async def test_params_user_contexts_invalid_type(bidi_session, value):
+    with pytest.raises(error.InvalidArgumentException):
+        await bidi_session.browsing_context.set_viewport(
+            user_contexts=value,
+            viewport={
+                "width": 100,
+                "height": 200,
+            }
+        )
+
+
+async def test_params_user_contexts_empty_list(bidi_session):
+    with pytest.raises(error.InvalidArgumentException):
+        await bidi_session.browsing_context.set_viewport(
+            user_contexts=[],
+            viewport={
+                "width": 100,
+                "height": 200,
+            }
+        )
+
+
+@pytest.mark.parametrize("value", [None, False, 42, {}, []])
+async def test_params_user_contexts_entry_invalid_type(bidi_session, value):
+    with pytest.raises(error.InvalidArgumentException):
+        await bidi_session.browsing_context.set_viewport(
+            user_contexts=[value],
+            viewport={
+                "width": 100,
+                "height": 200,
+            }
+        )
+
+
+@pytest.mark.parametrize("value", ["", "somestring"])
+async def test_params_user_contexts_entry_invalid_value(bidi_session, value):
+    with pytest.raises(error.NoSuchUserContextException):
+        await bidi_session.browsing_context.set_viewport(
+            user_contexts=[value],
+            viewport={
+                "width": 100,
+                "height": 200,
+            }
+        )
+
+
+async def test_params_both_user_contexts_and_context(bidi_session, top_context):
+    with pytest.raises(error.InvalidArgumentException):
+        await bidi_session.browsing_context.set_viewport(
+            context=top_context["context"],
+            user_contexts=["default"],
+            viewport={
+                "width": 100,
+                "height": 200,
+            }
+        )
+
+
+async def test_params_no_user_contexts_and_context(bidi_session):
+    with pytest.raises(error.InvalidArgumentException):
+        await bidi_session.browsing_context.set_viewport(
+            viewport={
+                "width": 100,
+                "height": 200,
+            }
+        )