Add a fixture for WebDriver BiDi tests to keep a list of installed web extensions and remove them at the end of the test.

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

bugzilla-url: https://bugzilla.mozilla.org/show_bug.cgi?id=1956885
gecko-commit: 204f50c78893884b11824efe542957ad02b29d18
gecko-reviewers: whimboo

Author: elkal98

webdriver/tests/bidi/web_extension/install/install.py

@@ -4,45 +4,33 @@
 
 
 @pytest.mark.asyncio
-async def test_install_from_base64(bidi_session, extension_data):
-    web_extension = await bidi_session.web_extension.install(
+async def test_install_from_base64(bidi_session, install_webextension, extension_data):
+    web_extension = await install_webextension(
         extension_data={
             "type": "base64",
             "value": extension_data["base64"],
         }
     )
-    try:
-        assert_extension_id(web_extension, extension_data)
-    finally:
-        # Clean up the extension.
-        await bidi_session.web_extension.uninstall(extension=web_extension)
+    assert_extension_id(web_extension, extension_data)
 
 
 @pytest.mark.asyncio
-async def test_install_from_path(bidi_session, extension_data):
-    web_extension = await bidi_session.web_extension.install(
+async def test_install_from_path(bidi_session, install_webextension, extension_data):
+    web_extension = await install_webextension(
         extension_data={
             "type": "path",
             "path": extension_data["path"],
         }
     )
-    try:
-        assert_extension_id(web_extension, extension_data)
-    finally:
-        # Clean up the extension.
-        await bidi_session.web_extension.uninstall(extension=web_extension)
+    assert_extension_id(web_extension, extension_data)
 
 
 @pytest.mark.asyncio
-async def test_install_from_archive_path(bidi_session, extension_data):
-    web_extension = await bidi_session.web_extension.install(
+async def test_install_from_archive_path(bidi_session, install_webextension, extension_data):
+    web_extension = await install_webextension(
         extension_data={
             "type": "archivePath",
             "path": extension_data["archivePath"],
         }
     )
-    try:
-        assert_extension_id(web_extension, extension_data)
-    finally:
-        # Clean up the extension.
-        await bidi_session.web_extension.uninstall(extension=web_extension)
+    assert_extension_id(web_extension, extension_data)

webdriver/tests/support/fixtures_bidi.py

@@ -1226,3 +1226,26 @@ async def match_media_query(context, media_query_type, media_query_value):
         return result["value"]
 
     return match_media_query
+
+
+@pytest_asyncio.fixture
+async def install_webextension(bidi_session):
+    installed_extensions = []
+
+    async def install(extension_data, _extension_params=None):
+        if _extension_params is None:
+            _extension_params = {}
+        extension = await bidi_session.web_extension.install(
+            extension_data=extension_data,
+            _extension_params=_extension_params
+        )
+        installed_extensions.append(extension)
+        return extension
+
+    yield install
+
+    for extension in installed_extensions:
+        try:
+            await bidi_session.web_extension.uninstall(extension=extension)
+        except Exception as e:
+            print(f"Failed to uninstall extension {extension}. Error: {e}")