facilitate running test environment with Haketilo loaded into browser

3 files changed, 67 insertions, 32 deletions

diff --git a/test/__main__.py b/test/__main__.py
index f8b8976..7afda55 100644
--- a/test/__main__.py
+++ b/test/__main__.py
@@ -36,23 +36,40 @@ import readline
 from .server import do_an_internet
 from .misc_constants import *
 from .profiles import firefox_safe_mode
+from .extension_crafting import get_extension_base_url
 
 def fail(msg, error_code):
     print('Error:', msg)
-    print('Usage:', sys.argv[0], '[certificates_directory] [proxy_port]')
+    print('Usage:', sys.argv[0], '[--load-haketilo]', '[certificates_directory] [proxy_port]')
     sys.exit(error_code)
 
-certdir = Path(sys.argv[1]).resolve() if len(sys.argv) > 1 else default_cert_dir
+load_haketilo = False
+argv_idx = 1
+if len(sys.argv) > argv_idx and sys.argv[argv_idx] == '--load-haketilo':
+    load_haketilo = True
+    argv_idx += 1
+
+certdir = Path(sys.argv[argv_idx]).resolve() if len(sys.argv) > argv_idx \
+    else default_cert_dir
+
 if not certdir.is_dir():
     fail('selected certificate directory does not exist.', 2)
 
-port = sys.argv[2] if len(sys.argv) > 2 else str(default_proxy_port)
+argv_idx += 1
+
+port = sys.argv[argv_idx] if len(sys.argv) > argv_idx \
+    else str(default_proxy_port)
+
 if not port.isnumeric():
     fail('port must be an integer.', 3)
 
 httpd = do_an_internet(certdir, int(port))
 driver = firefox_safe_mode(proxy_port=int(port))
 
+if load_haketilo:
+    driver.install_addon(str(here.parent / 'mozilla-build.zip'), temporary=True)
+    driver.get(get_extension_base_url(driver) + 'html/settings.html')
+
 print("You can now control the browser through 'driver' object")
 
 # Here we enable readline-enhanced editing:
diff --git a/test/extension_crafting.py b/test/extension_crafting.py
index bf54691..680c45e 100644
--- a/test/extension_crafting.py
+++ b/test/extension_crafting.py
@@ -6,7 +6,7 @@ Making temporary WebExtensions for use in the test suite
 
 # This file is part of Haketilo.
 #
-# Copyright (C) 2021 Wojtek Kosior <koszko@koszko.org>
+# Copyright (C) 2021, 2022 Wojtek Kosior <koszko@koszko.org>
 #
 # This program is free software: you can redistribute it and/or modify
 # it under the terms of the GNU General Public License as published by
@@ -27,11 +27,16 @@ Making temporary WebExtensions for use in the test suite
 
 import json
 import zipfile
+import re
+import shutil
+import subprocess
+
 from pathlib import Path
 from uuid import uuid4
 from tempfile import TemporaryDirectory
-import shutil
-import subprocess
+
+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.common.exceptions import NoSuchElementException
 
 from .misc_constants import *
 
@@ -170,3 +175,41 @@ def make_extension(destination_dir,
             html.add_to_xpi(xpi)
 
     return destination_path
+
+extract_base_url_re = re.compile(r'^(.*)manifest.json$')
+
+def get_extension_base_url(driver):
+    """
+    Extension's internall UUID is not directly exposed in Selenium. Instead, we
+    can navigate to about:debugging and inspect the manifest URL present there
+    to get the base url like:
+        moz-extension://b225c78f-d108-4caa-8406-f38b37d8dee5/
+    which can then be used to navigate to extension-bundled pages.
+    """
+    # For newer Firefoxes
+    driver.get('about:debugging#/runtime/this-firefox')
+
+    def get_manifest_link_newer_ff(driver):
+        try:
+            return driver.find_element_by_class_name('qa-manifest-url')
+        except NoSuchElementException:
+            pass
+
+        try:
+            details = driver.find_element_by_class_name('error-page-details')
+        except NoSuchElementException:
+            return False
+
+        if '#/runtime/this-firefox' in details.text:
+            return "not_newer_ff"
+
+    manifest_link = WebDriverWait(driver, 10).until(get_manifest_link_newer_ff)
+
+    if manifest_link == "not_newer_ff":
+        driver.get("about:debugging#addons")
+        driver.implicitly_wait(10)
+        manifest_link = driver.find_element_by_class_name('manifest-url')
+        driver.implicitly_wait(0)
+
+    manifest_url = manifest_link.get_attribute('href')
+    return extract_base_url_re.match(manifest_url).group(1)
diff --git a/test/test_integration.py b/test/test_integration.py
index 8d8d08d..87d1827 100644
--- a/test/test_integration.py
+++ b/test/test_integration.py
@@ -18,33 +18,8 @@ Haketilo integration tests
 # CC0 1.0 Universal License for more details.
 
 import pytest
-import re
 
-from selenium.webdriver.support.ui import WebDriverWait
-from selenium.common.exceptions import NoSuchElementException
-
-extract_base_url_re = re.compile(r'^(.*)manifest.json$')
-
-def get_extension_base_url(driver):
-    """
-    Extension's internall UUID is not directly exposed in Selenium. Instead, we
-    can navigate to about:debugging and inspect the manifest URL present there
-    to get the base url like:
-        moz-extension://b225c78f-d108-4caa-8406-f38b37d8dee5/
-    which can then be used to navigate to extension-bundled pages.
-    """
-    driver.implicitly_wait(10)
-    try:
-        # For newer Firefoxes
-        driver.get('about:debugging#/runtime/this-firefox')
-        manifest_link = driver.find_element_by_class_name('qa-manifest-url')
-    except NoSuchElementException:
-        driver.get("about:debugging#addons")
-        manifest_link = driver.find_element_by_class_name('manifest-url')
-    driver.implicitly_wait(0)
-
-    manifest_url = manifest_link.get_attribute('href')
-    return extract_base_url_re.match(manifest_url).group(1)
+from .extension_crafting import get_extension_base_url
 
 @pytest.mark.usefixtures('haketilo')
 def test_integration(driver):