[wdspec] Add additional wdspec tests for removeDataCollector

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

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

Author: juliandescottes

webdriver/tests/bidi/network/remove_data_collector/collector.py

@@ -1,13 +1,43 @@
 import pytest
 import webdriver.bidi.error as error
 
+from .. import PAGE_EMPTY_TEXT
+
 pytestmark = pytest.mark.asyncio
 
 
-async def test_return_value(bidi_session, add_intercept):
+async def test_return_value(bidi_session):
     collector = await bidi_session.network.add_data_collector(
         data_types=["response"], max_encoded_data_size=1000
     )
 
     result = await bidi_session.network.remove_data_collector(collector=collector)
     assert result == {}
+
+
+async def test_data_not_available_after_remove(
+    bidi_session, url, add_data_collector, setup_collected_response
+):
+    # Collect a network response with 2 collectors
+    collector = await add_data_collector(
+        collector_type="blob", data_types=["response"], max_encoded_data_size=1000
+    )
+    [request, other_collector] = await setup_collected_response(
+        fetch_url=url(PAGE_EMPTY_TEXT)
+    )
+
+    # Remove the first collector.
+    await bidi_session.network.remove_data_collector(collector=collector)
+
+    # Data still available from other collector and globally
+    await bidi_session.network.get_data(
+        request=request, data_type="response", collector=other_collector
+    )
+    await bidi_session.network.get_data(request=request, data_type="response")
+
+    # Remove the other collector.
+    await bidi_session.network.remove_data_collector(collector=other_collector)
+
+    # Data no longer available globally
+    with pytest.raises(error.NoSuchNetworkDataException):
+        await bidi_session.network.get_data(request=request, data_type="response")