load page and check extension install

Author: navin772

py/selenium/webdriver/remote/webdriver.py

@@ -1350,7 +1350,7 @@ def webextension(self):
         Examples:
         ---------
         >>> extension_path = "/path/to/extension"
-        >>> extension_result = driver.webextension.install(path=extension_path)))
+        >>> extension_result = driver.webextension.install(path=extension_path)
         >>> driver.webextension.uninstall(extension_result)
         """
         if not self._websocket_connection:

py/test/selenium/webdriver/common/bidi_webextension_tests.py

@@ -19,29 +19,15 @@
 import os
 import pytest
 
-EXTENSION_ID = "[email protected]"
-EXTENSION_PATH = "common/extensions/webextensions-selenium-example-signed"
-EXTENSION_ARCHIVE_PATH = "common/extensions/webextensions-selenium-example.xpi"
-
-
-# Function to find the project root directory
-def find_project_root():
-    current_dir = os.path.dirname(os.path.abspath(__file__))
-
-    while current_dir != os.path.dirname(current_dir):
-        extensions_dir = os.path.join(current_dir, "common", "extensions")
-        if os.path.isdir(extensions_dir):
-            return current_dir
-        current_dir = os.path.dirname(current_dir)
+from selenium.webdriver.common.by import By
+from selenium.webdriver.support.wait import WebDriverWait
 
-    return os.path.dirname(os.path.abspath(__file__))
 
+EXTENSION_ID = "[email protected]"
+EXTENSION_PATH = "webextensions-selenium-example"
+EXTENSION_ARCHIVE_PATH = "webextensions-selenium-example.xpi"
 
-@pytest.fixture
-def locate_project_path():
-    """Locate the project path for the extension files."""
-    project_root = find_project_root()
-    return project_root
+extensions = os.path.abspath("../../../../../../test/extensions/")
 
 
 def test_webextension_initialized(driver):
@@ -51,52 +37,114 @@ def test_webextension_initialized(driver):
 
 @pytest.mark.xfail_chrome
 @pytest.mark.xfail_edge
-def test_install_extension_path(driver, locate_project_path):
+def test_install_extension_path(driver, pages):
     """Test installing an extension from a directory path."""
-    path = os.path.join(locate_project_path, EXTENSION_PATH)
+    path = os.path.join(extensions, EXTENSION_PATH)
 
     ex_in = driver.webextension.install(path=path)
     assert ex_in.get("extension") == EXTENSION_ID
 
+    pages.load("blank.html")
+    injected = WebDriverWait(driver, timeout=2).until(
+        lambda dr: dr.find_element(By.ID, "webextensions-selenium-example")
+    )
+    assert injected.text == "Content injected by webextensions-selenium-example"
+
     driver.webextension.uninstall(ex_in)
 
+    context_id = driver.current_window_handle
+    driver.browsing_context.reload(context_id)
+    assert len(driver.find_elements(By.ID, "webextensions-selenium-example")) == 0
+
 
 @pytest.mark.xfail_chrome
 @pytest.mark.xfail_edge
-def test_install_archive_extension_path(driver, locate_project_path):
+def test_install_archive_extension_path(driver, pages):
     """Test installing an extension from an archive path."""
-    path = os.path.join(locate_project_path, EXTENSION_ARCHIVE_PATH)
+    path = os.path.join(extensions, EXTENSION_ARCHIVE_PATH)
 
     ex = driver.webextension.install(archive_path=path)
     assert ex.get("extension") == EXTENSION_ID
 
+    pages.load("blank.html")
+    injected = WebDriverWait(driver, timeout=2).until(
+        lambda dr: dr.find_element(By.ID, "webextensions-selenium-example")
+    )
+    assert injected.text == "Content injected by webextensions-selenium-example"
+
     driver.webextension.uninstall(ex)
 
+    context_id = driver.current_window_handle
+    driver.browsing_context.reload(context_id)
+
+    assert len(driver.find_elements(By.ID, "webextensions-selenium-example")) == 0
+
 
 @pytest.mark.xfail_chrome
 @pytest.mark.xfail_edge
-def test_install_base64_extension_path(driver, locate_project_path):
+def test_install_base64_extension_path(driver, pages):
     """Test installing an extension from a base64 encoded string."""
-    path = os.path.join(locate_project_path, EXTENSION_ARCHIVE_PATH)
+    path = os.path.join(extensions, EXTENSION_ARCHIVE_PATH)
 
     with open(path, "rb") as file:
         base64_encoded = base64.b64encode(file.read()).decode("utf-8")
 
     ex = driver.webextension.install(base64_value=base64_encoded)
     assert ex.get("extension") == EXTENSION_ID
 
+    pages.load("blank.html")
+    injected = WebDriverWait(driver, timeout=2).until(
+        lambda dr: dr.find_element(By.ID, "webextensions-selenium-example")
+    )
+    assert injected.text == "Content injected by webextensions-selenium-example"
+
+    driver.webextension.uninstall(ex)
+
+    context_id = driver.current_window_handle
+    driver.browsing_context.reload(context_id)
+
+    assert len(driver.find_elements(By.ID, "webextensions-selenium-example")) == 0
+
+
+@pytest.mark.xfail_chrome
+@pytest.mark.xfail_edge
+def test_install_unsigned_extension(driver, pages):
+    """Test installing an unsigned extension."""
+    path = os.path.join(extensions, "webextensions-selenium-example-unsigned")
+
+    ex = driver.webextension.install(path=path)
+    assert ex.get("extension") == EXTENSION_ID
+
+    pages.load("blank.html")
+    injected = WebDriverWait(driver, timeout=2).until(
+        lambda dr: dr.find_element(By.ID, "webextensions-selenium-example")
+    )
+    assert injected.text == "Content injected by webextensions-selenium-example"
+
     driver.webextension.uninstall(ex)
 
+    context_id = driver.current_window_handle
+    driver.browsing_context.reload(context_id)
+
+    assert len(driver.find_elements(By.ID, "webextensions-selenium-example")) == 0
+
 
 @pytest.mark.xfail_chrome
 @pytest.mark.xfail_edge
-def test_install_with_extension_id_uninstall(driver, locate_project_path):
+def test_install_with_extension_id_uninstall(driver, pages):
     """Test uninstalling an extension using just the extension ID."""
-    path = os.path.join(locate_project_path, EXTENSION_PATH)
+    path = os.path.join(extensions, EXTENSION_PATH)
 
     ex = driver.webextension.install(path=path)
     extension_id = ex.get("extension")
     assert extension_id == EXTENSION_ID
 
+    pages.load("blank.html")
+
     # Uninstall using the extension ID
     driver.webextension.uninstall(extension_id)
+
+    context_id = driver.current_window_handle
+    driver.browsing_context.reload(context_id)
+
+    assert len(driver.find_elements(By.ID, "webextensions-selenium-example")) == 0