test: add network interception continue request invalidation tests

create create_blocked_request helper

Author: Thiago Perrotta

tools/webdriver/webdriver/bidi/modules/network.py

@@ -41,7 +41,9 @@ def __init__(self, pattern: str):
 class Network(BidiModule):
     @command
     def add_intercept(
-        self, phases: List[str], url_patterns: Optional[List[URLPattern]] = None
+            self,
+            phases: List[str],
+            url_patterns: Optional[List[URLPattern]] = None
     ) -> Mapping[str, Any]:
         params: MutableMapping[str, Any] = {
             "phases": phases,
@@ -57,6 +59,27 @@ def _add_intercept(self, result: Mapping[str, Any]) -> Any:
         assert result["intercept"] is not None
         return result["intercept"]
 
+    @command
+    def continue_request(self,
+                         request: str,
+                         method: Optional[str] = None,
+                         url: Optional[str] = None) -> Mapping[str, Any]:
+        params: MutableMapping[str, Any] = {
+            "request": request,
+        }
+
+        if method is not None:
+            params["method"] = method
+
+        if url is not None:
+            params["url"] = url
+
+        # TODO: Add body.
+        # TODO: Add cookies.
+        # TODO: Add headers.
+
+        return params
+
     @command
     def fail_request(self, request: str) -> Mapping[str, Any]:
         params: MutableMapping[str, Any] = {"request": request}

webdriver/tests/bidi/network/continue_request/__init__.py

@@ -0,0 +1,101 @@
+import asyncio
+import pytest
+import webdriver.bidi.error as error
+from webdriver.bidi.modules.script import ScriptEvaluateResultException
+
+pytestmark = pytest.mark.asyncio
+
+PAGE_EMPTY_TEXT = "/webdriver/tests/bidi/network/support/empty.txt"
+
+
+@pytest.mark.parametrize("value", [None, False, 42, {}, []])
+async def test_params_request_invalid_type(bidi_session, value):
+    with pytest.raises(error.InvalidArgumentException):
+        await bidi_session.network.continue_request(request=value)
+
+
+@pytest.mark.parametrize("value", ["", "foo"])
+async def test_params_request_invalid_value(bidi_session, value):
+    with pytest.raises(error.NoSuchRequestException):
+        await bidi_session.network.continue_request(request=value)
+
+
+async def test_params_request_no_such_request(bidi_session, setup_network_test,
+                                              wait_for_event, fetch, url):
+    await setup_network_test(events=[
+        "network.responseCompleted",
+    ])
+    on_response_completed = wait_for_event("network.responseCompleted")
+
+    text_url = url(PAGE_EMPTY_TEXT)
+    await fetch(text_url)
+
+    response_completed_event = await on_response_completed
+    request = response_completed_event["request"]["request"]
+
+    with pytest.raises(error.NoSuchRequestException):
+        await bidi_session.network.continue_request(request=request)
+
+
+@pytest.mark.parametrize("value", [False, 42, {}, []])
+async def test_params_method_invalid_type(bidi_session, value,
+                                          setup_network_test, url, fetch,
+                                          wait_for_event, add_intercept):
+    request = await create_blocked_request(setup_network_test, url,
+                                           add_intercept, fetch,
+                                           wait_for_event)
+
+    with pytest.raises(error.InvalidArgumentException):
+        await bidi_session.network.continue_request(request=request,
+                                                    method=value)
+
+
+@pytest.mark.parametrize("value", [False, 42, {}, []])
+async def test_params_url_invalid_type(bidi_session, value, setup_network_test,
+                                       url, fetch, wait_for_event,
+                                       add_intercept):
+    request = await create_blocked_request(setup_network_test, url,
+                                           add_intercept, fetch,
+                                           wait_for_event)
+
+    with pytest.raises(error.InvalidArgumentException):
+        await bidi_session.network.continue_request(request=request, url=value)
+
+
+@pytest.mark.parametrize("protocol", ["http", "https"])
+@pytest.mark.parametrize("value", [":invalid", "#invalid"])
+async def test_params_url_invalid_value(bidi_session, protocol, value,
+                                        setup_network_test, url, fetch,
+                                        wait_for_event, add_intercept):
+    request = await create_blocked_request(setup_network_test, url,
+                                           add_intercept, fetch,
+                                           wait_for_event)
+
+    with pytest.raises(error.InvalidArgumentException):
+        await bidi_session.network.continue_request(
+            request=request, url=f"{protocol}://{value}")
+
+
+# TODO: Add body.
+# TODO: Add cookies.
+# TODO: Add headers.
+
+
+async def create_blocked_request(setup_network_test, url, add_intercept, fetch,
+                                 wait_for_event):
+    await setup_network_test(events=["network.beforeRequestSent"])
+
+    text_url = url(PAGE_EMPTY_TEXT)
+    await add_intercept(
+        phases=["beforeRequestSent"],
+        url_patterns=[{
+            "type": "string",
+            "pattern": text_url,
+        }],
+    )
+
+    asyncio.ensure_future(fetch(text_url))
+    event = await wait_for_event("network.beforeRequestSent")
+    request = event["request"]["request"]
+
+    return request