From bdf6e5f009f60a8a7bb423a831e8aaeb427edb99 Mon Sep 17 00:00:00 2001
From: Maksim Sadym <sadym@chromium.org>
Date: Wed, 13 Nov 2024 15:13:11 +0100
Subject: [PATCH] Add testdriver_features and use in Chrome

* Add testdriver_features
* Update doc
* Enable bidi in Chrome based on `testdriver_features` containing `bidi`
* Enable `infrastructure/webdriver/bidi/subscription.html` for chrome
* Enable `infrastructure/testdriver/bidi/permissions/set_permission.https.html` for chrome
* Enable `infrastructure/testdriver/bidi/bluetooth/simulate_adapter.https.html` for Chrome
---
 .../testdriver-extension-tutorial.md          | 12 +++-
 docs/writing-tests/testdriver.md              | 14 +++++
 .../bluetooth/simulate_adapter.https.html.ini |  3 +-
 .../permissions/set_permission.https.html.ini |  3 +-
 .../webdriver/bidi/subscription.html.ini      |  3 +-
 .../webdriver/bidi/subscription.window.js.ini |  2 +
 .../bluetooth/simulate_adapter.https.html     |  2 +-
 .../permissions/set_permission.https.html     |  2 +-
 .../webdriver/bidi/subscription.html          |  4 +-
 .../webdriver/bidi/subscription.window.js     | 22 ++++++++
 lint.ignore                                   |  1 +
 resources/testdriver.js                       | 24 ++++++++
 tools/lint/lint.py                            | 28 +++++++---
 tools/manifest/item.py                        |  6 ++
 tools/manifest/sourcefile.py                  | 56 ++++++++++++++++++-
 tools/manifest/tests/test_manifest.py         | 10 +++-
 tools/manifest/tests/test_sourcefile.py       | 53 ++++++++++++++++++
 tools/wptrunner/wptrunner/browsers/chrome.py  | 19 ++++++-
 .../wptrunner/executors/executorchrome.py     | 32 ++++++++++-
 tools/wptrunner/wptrunner/wpttest.py          | 11 +++-
 20 files changed, 280 insertions(+), 27 deletions(-)
 create mode 100644 infrastructure/metadata/infrastructure/webdriver/bidi/subscription.window.js.ini
 create mode 100644 infrastructure/webdriver/bidi/subscription.window.js

diff --git a/docs/writing-tests/testdriver-extension-tutorial.md b/docs/writing-tests/testdriver-extension-tutorial.md
index ca867c392da70e..6a1da4e0077de1 100644
--- a/docs/writing-tests/testdriver-extension-tutorial.md
+++ b/docs/writing-tests/testdriver-extension-tutorial.md
@@ -90,8 +90,16 @@ We will leave this unimplemented and override it in another file. Lets do that n
 
 #### WebDriver BiDi
 
-For commands using WebDriver BiDi, add the methods to `window.test_driver.bidi`. Parameters are passed as a single object `params`.
-<!-- TODO: add an example link once a first bidi command (probably `test_driver.bidi.permissions.set_permission`) is implemented.-->
+For commands using WebDriver BiDi, add the methods to `test_driver.bidi`. Parameters are passed as a single object `params`. For example [`test_driver.bidi.permissions.set_permission`](https://github.com/web-platform-tests/wpt/blob/5ec8ba6d68f27d49a056cbf940e3bc9a8324c538/resources/testdriver.js#L183).
+
+Before calling `test_driver_internal` method, assert the `bidi` testdriver feature is enabled. 
+```javascript
+set_permission: function (params) {
+    assertBidiIsEnabled();
+    return window.test_driver_internal.bidi.permissions.set_permission(
+        params);
+}
+```
 
 ### [tools/wptrunner/wptrunner/testdriver-extra.js](https://github.com/web-platform-tests/wpt/blob/master/tools/wptrunner/wptrunner/testdriver-extra.js)
 
diff --git a/docs/writing-tests/testdriver.md b/docs/writing-tests/testdriver.md
index 462c7b85503cb9..42a862fbe1e139 100644
--- a/docs/writing-tests/testdriver.md
+++ b/docs/writing-tests/testdriver.md
@@ -32,6 +32,20 @@ document when using testdriver from a different context):
 The api in `test_driver.bidi` provides access to the
 [WebDriver BiDi](https://w3c.github.io/webdriver-bidi) protocol.
 
+### Markup ###
+
+To use WebDriver BiDi, enable the `bidi` feature in `testdriver.js` by adding
+`feature=bidi` query string parameter. Details are in [RFC 214: Add testdriver features](https://github.com/web-platform-tests/rfcs/blob/master/rfcs/testdriver-features.md).
+```html
+<script src="/resources/testdriver.js?feature=bidi"></script>
+```
+
+```javascript
+// META: script=/resources/testdriver.js?feature=bidi
+```
+
+[Example](https://github.com/web-platform-tests/wpt/blob/aae46926b1fdccd460e1c6eaaf01ca20b941fbce/infrastructure/webdriver/bidi/subscription.html#L6).
+
 ### Context ###
 
 A WebDriver BiDi "browsing context" is equivalent to an
diff --git a/infrastructure/metadata/infrastructure/testdriver/bidi/bluetooth/simulate_adapter.https.html.ini b/infrastructure/metadata/infrastructure/testdriver/bidi/bluetooth/simulate_adapter.https.html.ini
index 7c3127d167a740..a40a56bcdb9bd8 100644
--- a/infrastructure/metadata/infrastructure/testdriver/bidi/bluetooth/simulate_adapter.https.html.ini
+++ b/infrastructure/metadata/infrastructure/testdriver/bidi/bluetooth/simulate_adapter.https.html.ini
@@ -1 +1,2 @@
-disabled: https://github.com/web-platform-tests/wpt/issues/47544
+disabled :
+  if product != "chrome": @True
diff --git a/infrastructure/metadata/infrastructure/testdriver/bidi/permissions/set_permission.https.html.ini b/infrastructure/metadata/infrastructure/testdriver/bidi/permissions/set_permission.https.html.ini
index 7c3127d167a740..1a5ac47c67d942 100644
--- a/infrastructure/metadata/infrastructure/testdriver/bidi/permissions/set_permission.https.html.ini
+++ b/infrastructure/metadata/infrastructure/testdriver/bidi/permissions/set_permission.https.html.ini
@@ -1 +1,2 @@
-disabled: https://github.com/web-platform-tests/wpt/issues/47544
+disabled:
+  if product != "chrome": @True
diff --git a/infrastructure/metadata/infrastructure/webdriver/bidi/subscription.html.ini b/infrastructure/metadata/infrastructure/webdriver/bidi/subscription.html.ini
index 7c3127d167a740..1a5ac47c67d942 100644
--- a/infrastructure/metadata/infrastructure/webdriver/bidi/subscription.html.ini
+++ b/infrastructure/metadata/infrastructure/webdriver/bidi/subscription.html.ini
@@ -1 +1,2 @@
-disabled: https://github.com/web-platform-tests/wpt/issues/47544
+disabled:
+  if product != "chrome": @True
diff --git a/infrastructure/metadata/infrastructure/webdriver/bidi/subscription.window.js.ini b/infrastructure/metadata/infrastructure/webdriver/bidi/subscription.window.js.ini
new file mode 100644
index 00000000000000..1a5ac47c67d942
--- /dev/null
+++ b/infrastructure/metadata/infrastructure/webdriver/bidi/subscription.window.js.ini
@@ -0,0 +1,2 @@
+disabled:
+  if product != "chrome": @True
diff --git a/infrastructure/testdriver/bidi/bluetooth/simulate_adapter.https.html b/infrastructure/testdriver/bidi/bluetooth/simulate_adapter.https.html
index 3850fa26974a32..ed60285c41fabe 100644
--- a/infrastructure/testdriver/bidi/bluetooth/simulate_adapter.https.html
+++ b/infrastructure/testdriver/bidi/bluetooth/simulate_adapter.https.html
@@ -3,7 +3,7 @@
 <title>TestDriver bidi.bluetooth.simulate_adapter method</title>
 <script src="/resources/testharness.js"></script>
 <script src="/resources/testharnessreport.js"></script>
-<script src="/resources/testdriver.js"></script>
+<script src="/resources/testdriver.js?feature=bidi"></script>
 <script src="/resources/testdriver-vendor.js"></script>
 
 <script>
diff --git a/infrastructure/testdriver/bidi/permissions/set_permission.https.html b/infrastructure/testdriver/bidi/permissions/set_permission.https.html
index 692977bb1cbe8e..5b05e2374e2b4b 100644
--- a/infrastructure/testdriver/bidi/permissions/set_permission.https.html
+++ b/infrastructure/testdriver/bidi/permissions/set_permission.https.html
@@ -3,7 +3,7 @@
 <title>TestDriver bidi.permissions.set_permission method</title>
 <script src="/resources/testharness.js"></script>
 <script src="/resources/testharnessreport.js"></script>
-<script src="/resources/testdriver.js"></script>
+<script src="/resources/testdriver.js?feature=bidi"></script>
 <script src="/resources/testdriver-vendor.js"></script>
 
 <script>
diff --git a/infrastructure/webdriver/bidi/subscription.html b/infrastructure/webdriver/bidi/subscription.html
index 056c2e5f77fb95..ad7724427fcff8 100644
--- a/infrastructure/webdriver/bidi/subscription.html
+++ b/infrastructure/webdriver/bidi/subscription.html
@@ -1,9 +1,9 @@
 <!DOCTYPE html>
 <meta charset="utf-8">
-<title>Test console log are present</title>
+<title>Can subscribe and receive WebDriver BiDi events</title>
 <script src="/resources/testharness.js"></script>
 <script src="/resources/testharnessreport.js"></script>
-<script src="/resources/testdriver.js"></script>
+<script src="/resources/testdriver.js?feature=bidi"></script>
 <script src="/resources/testdriver-vendor.js"></script>
 <script>
     promise_test(async () => {
diff --git a/infrastructure/webdriver/bidi/subscription.window.js b/infrastructure/webdriver/bidi/subscription.window.js
new file mode 100644
index 00000000000000..5233cdf1e997cd
--- /dev/null
+++ b/infrastructure/webdriver/bidi/subscription.window.js
@@ -0,0 +1,22 @@
+// META: title=Can subscribe and receive WebDriver BiDi events
+// META: script=/resources/testdriver.js?feature=bidi
+
+'use strict';
+
+promise_test(async () => {
+    const some_message = "SOME MESSAGE";
+    // Subscribe to `log.entryAdded` BiDi events. This will not add a listener to the page.
+    await test_driver.bidi.log.entry_added.subscribe();
+    // Add a listener for the log.entryAdded event. This will not subscribe to the event, so the subscription is
+    // required before. The cleanup is done automatically after the test is finished.
+    const log_entry_promise = test_driver.bidi.log.entry_added.once();
+    // Emit a console.log message.
+    // Note: Lint rule is disabled in `lint.ignore` file.
+    console.log(some_message);
+    // Wait for the log.entryAdded event to be received.
+    const event = await log_entry_promise;
+    // Assert the log.entryAdded event has the expected message.
+    assert_equals(event.args.length, 1);
+    const event_message = event.args[0];
+    assert_equals(event_message.value, some_message);
+}, "Assert testdriver can subscribe and receive events");
diff --git a/lint.ignore b/lint.ignore
index ee19f061a74aa8..e59a64d2ed7197 100644
--- a/lint.ignore
+++ b/lint.ignore
@@ -122,6 +122,7 @@ CONSOLE: webaudio/resources/audit.js:41
 
 # Intentional use of console.*
 CONSOLE: infrastructure/webdriver/bidi/subscription.html
+CONSOLE: infrastructure/webdriver/bidi/subscription.window.js
 
 # use of console in a public library - annotation-model ensures
 # it is not actually used
diff --git a/resources/testdriver.js b/resources/testdriver.js
index bdd18cc0e7d063..526bc7f699025d 100644
--- a/resources/testdriver.js
+++ b/resources/testdriver.js
@@ -3,6 +3,26 @@
     var idCounter = 0;
     let testharness_context = null;
 
+    const features = (() => {
+        function getFeatures(scriptSrc) {
+            try {
+                const url = new URL(scriptSrc);
+                return url.searchParams.getAll('feature');
+            } catch (e) {
+                return [];
+            }
+        }
+
+        return getFeatures(document?.currentScript?.src ?? '');
+    })();
+
+    function assertBidiIsEnabled(){
+        if (!features.includes('bidi')) {
+            throw new Error(
+                "`?feature=bidi` is missing when importing testdriver.js but the test is using WebDriver BiDi APIs");
+        }
+    }
+
     function getInViewCenterPoint(rect) {
         var left = Math.max(0, rect.left);
         var right = Math.min(window.innerWidth, rect.right);
@@ -115,6 +135,7 @@
                      * is successfully done.
                      */
                     subscribe: async function (params = {}) {
+                        assertBidiIsEnabled();
                         return window.test_driver_internal.bidi.log.entry_added.subscribe(params);
                     },
                     /**
@@ -127,6 +148,7 @@
                      * added event listener when called.
                      */
                     on: function (callback) {
+                        assertBidiIsEnabled();
                         return window.test_driver_internal.bidi.log.entry_added.on(callback);
                     },
                     /**
@@ -137,6 +159,7 @@
                      * with the event object when the event is emitted.
                      */
                     once: function () {
+                        assertBidiIsEnabled();
                         return new Promise(resolve => {
                             const remove_handler = window.test_driver_internal.bidi.log.entry_added.on(
                                 event => {
@@ -181,6 +204,7 @@
                  *                    the permission fails.
                  */
                 set_permission: function (params) {
+                    assertBidiIsEnabled();
                     return window.test_driver_internal.bidi.permissions.set_permission(
                         params);
                 }
diff --git a/tools/lint/lint.py b/tools/lint/lint.py
index 32bc7ece7b12f3..ce0168586c00cc 100644
--- a/tools/lint/lint.py
+++ b/tools/lint/lint.py
@@ -523,20 +523,34 @@ def check_parsed(repo_root: Text, path: Text, f: IO[bytes]) -> List[rules.Error]
     for element in source_file.root.findall(".//{http://www.w3.org/1999/xhtml}script[@src]"):
         src = element.attrib["src"]
 
-        def incorrect_path(script: Text, src: Text) -> bool:
-            return (script == src or
-                ("/%s" % script in src and src != "/resources/%s" % script))
+        def is_path_correct(script: Text, src: Text,
+              allow_query_params: bool = False) -> bool:
+            if script == src:
+                # The src does not provide the full path.
+                return False
 
-        if incorrect_path("testharness.js", src):
+            if "/%s" % script not in src:
+                # The src is not related to the script.
+                return True
+
+            if src == "/resources/%s" % script:
+                # The src is properly included.
+                return True
+
+            # Check if the script is properly included with query parameters.
+            return allow_query_params and ("%s" % src).startswith(
+                "/resources/%s?" % script)
+
+        if not is_path_correct("testharness.js", src):
             errors.append(rules.TestharnessPath.error(path))
 
-        if incorrect_path("testharnessreport.js", src):
+        if not is_path_correct("testharnessreport.js", src):
             errors.append(rules.TestharnessReportPath.error(path))
 
-        if incorrect_path("testdriver.js", src):
+        if not is_path_correct("testdriver.js", src, True):
             errors.append(rules.TestdriverPath.error(path))
 
-        if incorrect_path("testdriver-vendor.js", src):
+        if not is_path_correct("testdriver-vendor.js", src):
             errors.append(rules.TestdriverVendorPath.error(path))
 
         script_path = None
diff --git a/tools/manifest/item.py b/tools/manifest/item.py
index e25f7ca2c29adc..17cfeb8ebb7356 100644
--- a/tools/manifest/item.py
+++ b/tools/manifest/item.py
@@ -165,6 +165,10 @@ def timeout(self) -> Optional[Text]:
     def pac(self) -> Optional[Text]:
         return self._extras.get("pac")
 
+    @property
+    def testdriver_features(self) -> Optional[List[Text]]:
+        return self._extras.get("testdriver_features")
+
     @property
     def testdriver(self) -> Optional[Text]:
         return self._extras.get("testdriver")
@@ -183,6 +187,8 @@ def to_json(self) -> Tuple[Optional[Text], Dict[Text, Any]]:
             rv[-1]["timeout"] = self.timeout
         if self.pac is not None:
             rv[-1]["pac"] = self.pac
+        if self.testdriver_features is not None:
+            rv[-1]["testdriver_features"] = self.testdriver_features
         if self.testdriver:
             rv[-1]["testdriver"] = self.testdriver
         if self.jsshell:
diff --git a/tools/manifest/sourcefile.py b/tools/manifest/sourcefile.py
index 02ab1ad4fe6017..cef5ef40a9b98b 100644
--- a/tools/manifest/sourcefile.py
+++ b/tools/manifest/sourcefile.py
@@ -6,7 +6,7 @@
 from io import BytesIO
 from typing import (Any, BinaryIO, Callable, Deque, Dict, Iterable, List,
                     Optional, Pattern, Set, Text, Tuple, TypedDict, Union, cast)
-from urllib.parse import urljoin
+from urllib.parse import urlparse, parse_qs, urljoin
 
 try:
     from xml.etree import cElementTree as ElementTree
@@ -723,7 +723,15 @@ def testdriver_nodes(self) -> List[ElementTree.Element]:
         """List of ElementTree Elements corresponding to nodes representing a
         testdriver.js script"""
         assert self.root is not None
-        return self.root.findall(".//{http://www.w3.org/1999/xhtml}script[@src='/resources/testdriver.js']")
+        # `xml.etree.ElementTree.findall` has a limited support of xPath, so
+        # explicit filter is required.
+        return [node for node in
+                self.root.findall(".//{http://www.w3.org/1999/xhtml}script")
+                if node.attrib.get('src',
+                                   "") == '/resources/testdriver.js'
+                or
+                node.attrib.get('src', "").startswith(
+                    '/resources/testdriver.js?')]
 
     @cached_property
     def has_testdriver(self) -> Optional[bool]:
@@ -733,6 +741,46 @@ def has_testdriver(self) -> Optional[bool]:
             return None
         return bool(self.testdriver_nodes)
 
+    def ___get_testdriver_include_path(self):
+        if self.script_metadata:
+            for (meta, content) in self.script_metadata:
+                if meta.strip() == 'script' and content.startswith(
+                      '/resources/testdriver.js'):
+                    return content.strip()
+
+        if self.root is None:
+            return None
+
+        for node in self.testdriver_nodes:
+            if "src" in node.attrib:
+                return node.attrib.get("src")
+
+        return None
+
+
+    @cached_property
+    def testdriver_features(self) -> Optional[Text]:
+        """
+        List of requested testdriver features.
+        """
+
+        testdriver_include_url = self.___get_testdriver_include_path()
+
+        if testdriver_include_url is None:
+            return None
+
+        # Parse the URL
+        parsed_url = urlparse(testdriver_include_url)
+        # Extract query parameters
+        query_params = parse_qs(parsed_url.query)
+        # Get the values for the 'feature' parameter
+        feature_values = query_params.get('feature', [])
+
+        if len(feature_values) > 0:
+            return feature_values
+
+        return None
+
     @cached_property
     def reftest_nodes(self) -> List[ElementTree.Element]:
         """List of ElementTree Elements corresponding to nodes representing a
@@ -984,6 +1032,7 @@ def manifest_items(self) -> Tuple[Text, List[ManifestItem]]:
                     global_variant_url(self.rel_url, suffix) + variant,
                     timeout=self.timeout,
                     pac=self.pac,
+                    testdriver_features=self.testdriver_features,
                     jsshell=jsshell,
                     script_metadata=self.script_metadata
                 )
@@ -1002,6 +1051,7 @@ def manifest_items(self) -> Tuple[Text, List[ManifestItem]]:
                     test_url + variant,
                     timeout=self.timeout,
                     pac=self.pac,
+                    testdriver_features=self.testdriver_features,
                     script_metadata=self.script_metadata
                 )
                 for variant in self.test_variants
@@ -1018,6 +1068,7 @@ def manifest_items(self) -> Tuple[Text, List[ManifestItem]]:
                     test_url + variant,
                     timeout=self.timeout,
                     pac=self.pac,
+                    testdriver_features=self.testdriver_features,
                     script_metadata=self.script_metadata
                 )
                 for variant in self.test_variants
@@ -1045,6 +1096,7 @@ def manifest_items(self) -> Tuple[Text, List[ManifestItem]]:
                     url,
                     timeout=self.timeout,
                     pac=self.pac,
+                    testdriver_features=self.testdriver_features,
                     testdriver=testdriver,
                     script_metadata=self.script_metadata
                 ))
diff --git a/tools/manifest/tests/test_manifest.py b/tools/manifest/tests/test_manifest.py
index fc2314b8356a21..762b9f9a4477d2 100644
--- a/tools/manifest/tests/test_manifest.py
+++ b/tools/manifest/tests/test_manifest.py
@@ -294,8 +294,11 @@ def test_update_from_json_modified():
     # Reload it from JSON
     m = manifest.Manifest.from_json("/", json_str)
 
-    # Update it with timeout="long"
-    s2 = SourceFileWithTest("test1", "1"*40, item.TestharnessTest, timeout="long", pac="proxy.pac")
+    testdriver_features = ['feature_1', 'feature_2']
+
+    # Update timeout, pac and testdriver_features
+    s2 = SourceFileWithTest("test1", "1" * 40, item.TestharnessTest, timeout="long", pac="proxy.pac",
+                            testdriver_features=testdriver_features)
     tree, sourcefile_mock = tree_and_sourcefile_mocks([(s2, None, True)])
     with mock.patch("tools.manifest.manifest.SourceFile", side_effect=sourcefile_mock):
         m.update(tree)
@@ -303,7 +306,8 @@ def test_update_from_json_modified():
     assert json_str == {
         'items': {'testharness': {'test1': [
             "1"*40,
-            (None, {'timeout': 'long', 'pac': 'proxy.pac'})
+            (None, {'timeout': 'long', 'pac': 'proxy.pac',
+                    'testdriver_features': testdriver_features})
         ]}},
         'url_base': '/',
         'version': 8
diff --git a/tools/manifest/tests/test_sourcefile.py b/tools/manifest/tests/test_sourcefile.py
index 8a9d8c36ee1c62..cc221888473db2 100644
--- a/tools/manifest/tests/test_sourcefile.py
+++ b/tools/manifest/tests/test_sourcefile.py
@@ -960,3 +960,56 @@ def test_page_ranges_invalid(page_ranges):
 def test_hash():
     s = SourceFile("/", "foo", "/", contents=b"Hello, World!")
     assert "b45ef6fec89518d314f546fd6c3025367b721684" == s.hash
+
+
+@pytest.mark.parametrize("file_name",
+                         ["html/test.worker.js", "html/test.window.js"])
+def test_script_testdriver_missing_features(file_name):
+    contents = f"""// META: title=TEST_TITLE
+// META: script=/resources/testdriver.js
+    test()""".encode("utf-8")
+
+    s = create(file_name, contents=contents)
+    item_type, items = s.manifest_items()
+    for item in items:
+        assert item.testdriver_features is None
+
+
+@pytest.mark.parametrize("features",
+                         [[], ['feature_1'], ['feature_1', 'feature_2']])
+@pytest.mark.parametrize("file_name",
+                         ["html/test.worker.js", "html/test.window.js"])
+def test_script_testdriver_features(file_name, features):
+    contents = f"""// META: title=TEST_TITLE
+// META: script=/resources/testdriver.js?{"&".join('feature=' + f for f in features)}             
+    test()""".encode("utf-8")
+
+    s = create(file_name, contents=contents)
+    item_type, items = s.manifest_items()
+    for item in items:
+        assert item.testdriver_features == (
+            features if len(features) > 0 else None)
+
+
+def test_html_testdriver_missing_features():
+    contents = f"""
+<!--Required to make test type `testharness` -->
+<script src="/resources/testharness.js"></script> 
+<script src="/resources/testdriver.js?feature=bidi"></script>
+    """.encode("utf-8")
+
+    s = create("html/test.html", contents=contents)
+    s.testdriver_features == None
+
+
+@pytest.mark.parametrize("features",
+                         [[], ['feature_1'], ['feature_1', 'feature_2']])
+def test_html_testdriver_features(features):
+    contents = f"""
+<script src="/resources/testdriver.js?{"&".join('feature=' + f for f in features)}"></script>
+<!--Required to make test type `testharness` -->
+<script src="/resources/testharness.js?feature=bidi"></script> 
+    """.encode("utf-8")
+
+    s = create("html/test.html", contents=contents)
+    s.testdriver_features == ['bidi']
diff --git a/tools/wptrunner/wptrunner/browsers/chrome.py b/tools/wptrunner/wptrunner/browsers/chrome.py
index b39b8deb76c958..d9d3ed166e10c5 100644
--- a/tools/wptrunner/wptrunner/browsers/chrome.py
+++ b/tools/wptrunner/wptrunner/browsers/chrome.py
@@ -6,7 +6,7 @@
 from mozlog.structuredlog import StructuredLogger
 
 from . import chrome_spki_certs
-from .base import BrowserError
+from .base import BrowserError, BrowserSettings
 from .base import WebDriverBrowser, require_arg
 from .base import NullBrowser  # noqa: F401
 from .base import OutputHandler
@@ -41,6 +41,8 @@
                  "update_properties": "update_properties",
                  "timeout_multiplier": "get_timeout_multiplier",}
 
+from ..wpttest import Test
+
 
 def debug_args(debug_info):
     if debug_info.interactive:
@@ -214,6 +216,7 @@ def __init__(self,
         super().__init__(logger, **kwargs)
         self._leak_check = leak_check
         self._actual_port = None
+        self._require_webdriver_bidi = None
 
     def restart_on_test_type_change(self, new_test_type: str, old_test_type: str) -> bool:
         # Restart the test runner when switch from/to wdspec tests. Wdspec test
@@ -262,6 +265,20 @@ def executor_browser(self):
         browser_cls, browser_kwargs = super().executor_browser()
         return browser_cls, {**browser_kwargs, "leak_check": self._leak_check}
 
+    @property
+    def require_webdriver_bidi(self) -> Optional[bool]:
+        return self._require_webdriver_bidi
+
+    def settings(self, test: Test) -> BrowserSettings:
+        """ Required to store `require_webdriver_bidi` in browser settings."""
+        settings = super().settings(test)
+        self._require_webdriver_bidi = test.testdriver_features is not None and 'bidi' in test.testdriver_features
+
+        return {
+            **settings,
+            "require_webdriver_bidi": self._require_webdriver_bidi
+        }
+
 
 class ChromeDriverOutputHandler(OutputHandler):
     PORT_RE = re.compile(rb'.*was started successfully on port (\d+)\.')
diff --git a/tools/wptrunner/wptrunner/executors/executorchrome.py b/tools/wptrunner/wptrunner/executors/executorchrome.py
index fa593ce4023106..c46b8e978c4d26 100644
--- a/tools/wptrunner/wptrunner/executors/executorchrome.py
+++ b/tools/wptrunner/wptrunner/executors/executorchrome.py
@@ -20,6 +20,7 @@
     WebDriverFedCMProtocolPart,
     WebDriverPrintRefTestExecutor,
     WebDriverProtocol,
+    WebDriverBidiProtocol,
     WebDriverRefTestExecutor,
     WebDriverTestharnessExecutor,
     WebDriverTestharnessProtocolPart,
@@ -211,6 +212,27 @@ def __init__(self, executor, browser, capabilities, **kwargs):
         super().__init__(executor, browser, capabilities, **kwargs)
 
 
+class ChromeDriverBidiProtocol(WebDriverBidiProtocol):
+    implements = [
+        ChromeDriverBaseProtocolPart,
+        ChromeDriverDevToolsProtocolPart,
+        ChromeDriverFedCMProtocolPart,
+        ChromeDriverTestharnessProtocolPart,
+    ]
+    for base_part in WebDriverBidiProtocol.implements:
+        if base_part.name not in {part.name for part in implements}:
+            implements.append(base_part)
+
+    # Prefix to apply to vendor-specific WebDriver extension commands.
+    vendor_prefix = "goog"
+
+    def __init__(self, executor, browser, capabilities, **kwargs):
+        self.implements = list(ChromeDriverBidiProtocol.implements)
+        if getattr(browser, "leak_check", False):
+            self.implements.append(ChromeDriverLeakProtocolPart)
+        super().__init__(executor, browser, capabilities, **kwargs)
+
+
 def _evaluate_leaks(executor_cls):
     if hasattr(executor_cls, "base_convert_result"):
         # Don't wrap more than once, which can cause unbounded recursion.
@@ -244,9 +266,15 @@ class ChromeDriverRefTestExecutor(WebDriverRefTestExecutor, _SanitizerMixin):  #
 
 @_evaluate_leaks
 class ChromeDriverTestharnessExecutor(WebDriverTestharnessExecutor, _SanitizerMixin):  # type: ignore
-    protocol_cls = ChromeDriverProtocol
-
     def __init__(self, *args, reuse_window=False, **kwargs):
+        require_webdriver_bidi = kwargs.get("browser_settings", {}).get(
+            "require_webdriver_bidi", None)
+
+        if require_webdriver_bidi:
+            self.protocol_cls = ChromeDriverBidiProtocol
+        else:
+            self.protocol_cls = ChromeDriverProtocol
+
         super().__init__(*args, **kwargs)
         self.reuse_window = reuse_window
 
diff --git a/tools/wptrunner/wptrunner/wpttest.py b/tools/wptrunner/wptrunner/wpttest.py
index 2e3fd974d4d43e..2646fc4449868a 100644
--- a/tools/wptrunner/wptrunner/wpttest.py
+++ b/tools/wptrunner/wptrunner/wpttest.py
@@ -216,7 +216,8 @@ class Test(ABC):
     long_timeout = 60  # seconds
 
     def __init__(self, url_base, tests_root, url, inherit_metadata, test_metadata,
-                 timeout=None, path=None, protocol="http", subdomain=False, pac=None):
+          timeout=None, path=None, protocol="http", subdomain=False, pac=None,
+          testdriver_features=None):
         self.url_base = url_base
         self.tests_root = tests_root
         self.url = url
@@ -224,6 +225,7 @@ def __init__(self, url_base, tests_root, url, inherit_metadata, test_metadata,
         self._test_metadata = test_metadata
         self.timeout = timeout if timeout is not None else self.default_timeout
         self.path = path
+        self.testdriver_features = testdriver_features
         self.subdomain = subdomain
         self.environment = {"url_base": url_base,
                             "protocol": protocol,
@@ -482,9 +484,10 @@ class TestharnessTest(Test):
 
     def __init__(self, url_base, tests_root, url, inherit_metadata, test_metadata,
                  timeout=None, path=None, protocol="http", testdriver=False,
-                 jsshell=False, scripts=None, subdomain=False, pac=None):
+                 jsshell=False, scripts=None, subdomain=False, pac=None,
+                 testdriver_features=None):
         Test.__init__(self, url_base, tests_root, url, inherit_metadata, test_metadata, timeout,
-                      path, protocol, subdomain, pac)
+                      path, protocol, subdomain, pac, testdriver_features)
 
         self.testdriver = testdriver
         self.jsshell = jsshell
@@ -494,6 +497,7 @@ def __init__(self, url_base, tests_root, url, inherit_metadata, test_metadata,
     def from_manifest(cls, manifest_file, manifest_item, inherit_metadata, test_metadata):
         timeout = cls.long_timeout if manifest_item.timeout == "long" else cls.default_timeout
         pac = manifest_item.pac
+        testdriver_features = manifest_item.testdriver_features
         testdriver = manifest_item.testdriver if hasattr(manifest_item, "testdriver") else False
         jsshell = manifest_item.jsshell if hasattr(manifest_item, "jsshell") else False
         script_metadata = manifest_item.script_metadata or []
@@ -506,6 +510,7 @@ def from_manifest(cls, manifest_file, manifest_item, inherit_metadata, test_meta
                    test_metadata,
                    timeout=timeout,
                    pac=pac,
+                   testdriver_features=testdriver_features,
                    path=os.path.join(manifest_file.tests_root, manifest_item.path),
                    protocol=server_protocol(manifest_item),
                    testdriver=testdriver,