[wdspec] Add fixture to add network data collector

Depends on D259969

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

bugzilla-url: https://bugzilla.mozilla.org/show_bug.cgi?id=1971780
gecko-commit: 8f9e6b10369b0fe0c9e00fffae88bec4bd6a2b78
gecko-reviewers: webdriver-reviewers, Sasha

Author: juliandescottes

webdriver/tests/support/fixtures_bidi.py

@@ -15,6 +15,7 @@
     InvalidArgumentException,
     NoSuchFrameException,
     NoSuchInterceptException,
+    NoSuchNetworkCollectorException,
     NoSuchRequestException,
     NoSuchScriptException,
     NoSuchUserContextException,
@@ -736,6 +737,44 @@ async def on_event(method, data, event=event):
         remove_listener()
 
 
+@pytest_asyncio.fixture
+async def add_data_collector(bidi_session):
+    """Add a network data collector, and ensure the collector is removed at the
+    end of the test."""
+
+    collectors = []
+
+    async def add_data_collector(
+        collector_type="blob",
+        data_types=["response"],
+        max_encoded_data_size=1000,
+        contexts=None,
+        user_contexts=None,
+    ):
+        nonlocal collectors
+        collector = await bidi_session.network.add_data_collector(
+            collector_type=collector_type,
+            data_types=data_types,
+            max_encoded_data_size=max_encoded_data_size,
+            contexts=contexts,
+            user_contexts=user_contexts,
+        )
+        collectors.append(collector)
+
+        return collector
+
+    yield add_data_collector
+
+    # Remove all added collectors at the end of the test
+    for collector in collectors:
+        try:
+            await bidi_session.network.remove_data_collector(collector=collector)
+        except NoSuchNetworkCollectorException:
+            # Ignore exceptions in case a specific collector was already removed
+            # during the test.
+            pass
+
+
 @pytest_asyncio.fixture
 async def add_intercept(bidi_session):
     """Add a network intercept for the provided phases and url patterns, and