summaryrefslogtreecommitdiffstats
path: root/toolkit/components/extensions/test/marionette
diff options
context:
space:
mode:
Diffstat (limited to 'toolkit/components/extensions/test/marionette')
-rw-r--r--toolkit/components/extensions/test/marionette/data/extension-with-bg-sw/manifest.json11
-rw-r--r--toolkit/components/extensions/test/marionette/data/extension-with-bg-sw/sw.js3
-rw-r--r--toolkit/components/extensions/test/marionette/manifest-serviceworker.toml4
-rw-r--r--toolkit/components/extensions/test/marionette/service_worker_testutils.py48
-rw-r--r--toolkit/components/extensions/test/marionette/test_extension_serviceworkers_purged_on_pref_disabled.py56
-rw-r--r--toolkit/components/extensions/test/marionette/test_temporary_extension_serviceworkers_not_persisted.py54
6 files changed, 176 insertions, 0 deletions
diff --git a/toolkit/components/extensions/test/marionette/data/extension-with-bg-sw/manifest.json b/toolkit/components/extensions/test/marionette/data/extension-with-bg-sw/manifest.json
new file mode 100644
index 0000000000..38a5c3f027
--- /dev/null
+++ b/toolkit/components/extensions/test/marionette/data/extension-with-bg-sw/manifest.json
@@ -0,0 +1,11 @@
+{
+ "manifest_version": 2,
+ "name": "Test Extension with Background Service Worker",
+ "version": "1",
+ "browser_specific_settings": {
+ "gecko": { "id": "extension-with-bg-sw@test" }
+ },
+ "background": {
+ "service_worker": "sw.js"
+ }
+}
diff --git a/toolkit/components/extensions/test/marionette/data/extension-with-bg-sw/sw.js b/toolkit/components/extensions/test/marionette/data/extension-with-bg-sw/sw.js
new file mode 100644
index 0000000000..2282e6a64b
--- /dev/null
+++ b/toolkit/components/extensions/test/marionette/data/extension-with-bg-sw/sw.js
@@ -0,0 +1,3 @@
+"use strict";
+
+dump("extension-with-bg-sw: sw.js loaded");
diff --git a/toolkit/components/extensions/test/marionette/manifest-serviceworker.toml b/toolkit/components/extensions/test/marionette/manifest-serviceworker.toml
new file mode 100644
index 0000000000..c8035f80c2
--- /dev/null
+++ b/toolkit/components/extensions/test/marionette/manifest-serviceworker.toml
@@ -0,0 +1,4 @@
+[DEFAULT]
+
+["test_extension_serviceworkers_purged_on_pref_disabled.py"]
+["test_temporary_extension_serviceworkers_not_persisted.py"]
diff --git a/toolkit/components/extensions/test/marionette/service_worker_testutils.py b/toolkit/components/extensions/test/marionette/service_worker_testutils.py
new file mode 100644
index 0000000000..b1fda926c0
--- /dev/null
+++ b/toolkit/components/extensions/test/marionette/service_worker_testutils.py
@@ -0,0 +1,48 @@
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+from marionette_harness import MarionetteTestCase
+
+EXT_ID = "extension-with-bg-sw@test"
+EXT_DIR_PATH = "extension-with-bg-sw"
+PREF_BG_SW_ENABLED = "extensions.backgroundServiceWorker.enabled"
+PREF_PERSIST_TEMP_ADDONS = (
+ "dom.serviceWorkers.testing.persistTemporarilyInstalledAddons"
+)
+
+
+class MarionetteServiceWorkerTestCase(MarionetteTestCase):
+ def get_extension_url(self, path="/"):
+ with self.marionette.using_context("chrome"):
+ return self.marionette.execute_script(
+ """
+ let policy = WebExtensionPolicy.getByID(arguments[0]);
+ return policy.getURL(arguments[1])
+ """,
+ script_args=(self.test_extension_id, path),
+ )
+
+ @property
+ def is_extension_service_worker_registered(self):
+ with self.marionette.using_context("chrome"):
+ return self.marionette.execute_script(
+ """
+ let serviceWorkerManager = Cc["@mozilla.org/serviceworkers/manager;1"].getService(
+ Ci.nsIServiceWorkerManager
+ );
+
+ let serviceWorkers = serviceWorkerManager.getAllRegistrations();
+ for (let i = 0; i < serviceWorkers.length; i++) {
+ let sw = serviceWorkers.queryElementAt(
+ i,
+ Ci.nsIServiceWorkerRegistrationInfo
+ );
+ if (sw.scope == arguments[0]) {
+ return true;
+ }
+ }
+ return false;
+ """,
+ script_args=(self.test_extension_base_url,),
+ )
diff --git a/toolkit/components/extensions/test/marionette/test_extension_serviceworkers_purged_on_pref_disabled.py b/toolkit/components/extensions/test/marionette/test_extension_serviceworkers_purged_on_pref_disabled.py
new file mode 100644
index 0000000000..ff2184c692
--- /dev/null
+++ b/toolkit/components/extensions/test/marionette/test_extension_serviceworkers_purged_on_pref_disabled.py
@@ -0,0 +1,56 @@
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+import os
+import sys
+
+from marionette_driver import Wait
+from marionette_driver.addons import Addons
+
+# Add this directory to the import path.
+sys.path.append(os.path.dirname(__file__))
+
+from service_worker_testutils import (
+ EXT_DIR_PATH,
+ EXT_ID,
+ PREF_BG_SW_ENABLED,
+ PREF_PERSIST_TEMP_ADDONS,
+ MarionetteServiceWorkerTestCase,
+)
+
+
+class PurgeExtensionServiceWorkersOnPrefDisabled(MarionetteServiceWorkerTestCase):
+ def setUp(self):
+ super(PurgeExtensionServiceWorkersOnPrefDisabled, self).setUp()
+ self.test_extension_id = EXT_ID
+ # Flip the "mirror: once" pref and restart Firefox to be able
+ # to run the extension successfully.
+ self.marionette.set_pref(PREF_BG_SW_ENABLED, True)
+ self.marionette.set_pref(PREF_PERSIST_TEMP_ADDONS, True)
+ self.marionette.restart(in_app=True)
+
+ def tearDown(self):
+ self.marionette.restart(in_app=False, clean=True)
+ super(PurgeExtensionServiceWorkersOnPrefDisabled, self).tearDown()
+
+ def test_unregistering_service_worker_when_clearing_data(self):
+ self.install_extension_with_service_worker()
+
+ # Flip the pref to false and restart again to verify that the
+ # service worker registration has been removed as expected.
+ self.marionette.set_pref(PREF_BG_SW_ENABLED, False)
+ self.marionette.restart(in_app=True)
+ self.assertFalse(self.is_extension_service_worker_registered)
+
+ def install_extension_with_service_worker(self):
+ addons = Addons(self.marionette)
+ test_extension_path = os.path.join(
+ os.path.dirname(self.filepath), "data", EXT_DIR_PATH
+ )
+ addons.install(test_extension_path, temp=True)
+ self.test_extension_base_url = self.get_extension_url()
+ Wait(self.marionette).until(
+ lambda _: self.is_extension_service_worker_registered,
+ message="Wait the extension service worker to be registered",
+ )
diff --git a/toolkit/components/extensions/test/marionette/test_temporary_extension_serviceworkers_not_persisted.py b/toolkit/components/extensions/test/marionette/test_temporary_extension_serviceworkers_not_persisted.py
new file mode 100644
index 0000000000..57c0696385
--- /dev/null
+++ b/toolkit/components/extensions/test/marionette/test_temporary_extension_serviceworkers_not_persisted.py
@@ -0,0 +1,54 @@
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+import os
+import sys
+
+from marionette_driver import Wait
+from marionette_driver.addons import Addons
+
+# Add this directory to the import path.
+sys.path.append(os.path.dirname(__file__))
+
+from service_worker_testutils import (
+ EXT_DIR_PATH,
+ EXT_ID,
+ PREF_BG_SW_ENABLED,
+ MarionetteServiceWorkerTestCase,
+)
+
+
+class TemporarilyInstalledAddonServiceWorkerNotPersisted(
+ MarionetteServiceWorkerTestCase
+):
+ def setUp(self):
+ super(TemporarilyInstalledAddonServiceWorkerNotPersisted, self).setUp()
+ self.test_extension_id = EXT_ID
+ # Flip the "mirror: once" pref and restart Firefox to be able
+ # to run the extension successfully.
+ self.marionette.set_pref(PREF_BG_SW_ENABLED, True)
+ self.marionette.restart(in_app=True)
+
+ def tearDown(self):
+ self.marionette.restart(in_app=False, clean=True)
+ super(TemporarilyInstalledAddonServiceWorkerNotPersisted, self).tearDown()
+
+ def test_temporarily_installed_addon_serviceWorkers_not_persisted(self):
+ self.install_temporary_extension_with_service_worker()
+ # Make sure the extension worker registration is persisted
+ # across restarts when the pref stays set to true.
+ self.marionette.restart(in_app=True)
+ self.assertFalse(self.is_extension_service_worker_registered)
+
+ def install_temporary_extension_with_service_worker(self):
+ addons = Addons(self.marionette)
+ test_extension_path = os.path.join(
+ os.path.dirname(self.filepath), "data", EXT_DIR_PATH
+ )
+ addons.install(test_extension_path, temp=True)
+ self.test_extension_base_url = self.get_extension_url()
+ Wait(self.marionette).until(
+ lambda _: self.is_extension_service_worker_registered,
+ message="Wait the extension service worker to be registered",
+ )