summaryrefslogtreecommitdiffstats
path: root/dom/workers/test/marionette
diff options
context:
space:
mode:
Diffstat (limited to 'dom/workers/test/marionette')
-rw-r--r--dom/workers/test/marionette/manifest.ini2
-rw-r--r--dom/workers/test/marionette/service_worker_utils.py63
-rw-r--r--dom/workers/test/marionette/test_service_workers_at_startup.py31
-rw-r--r--dom/workers/test/marionette/test_service_workers_disabled.py37
4 files changed, 133 insertions, 0 deletions
diff --git a/dom/workers/test/marionette/manifest.ini b/dom/workers/test/marionette/manifest.ini
new file mode 100644
index 0000000000..05aa421e49
--- /dev/null
+++ b/dom/workers/test/marionette/manifest.ini
@@ -0,0 +1,2 @@
+[test_service_workers_at_startup.py]
+[test_service_workers_disabled.py]
diff --git a/dom/workers/test/marionette/service_worker_utils.py b/dom/workers/test/marionette/service_worker_utils.py
new file mode 100644
index 0000000000..b29789b6f0
--- /dev/null
+++ b/dom/workers/test/marionette/service_worker_utils.py
@@ -0,0 +1,63 @@
+# 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
+
+from marionette_driver import Wait
+from marionette_harness import MarionetteTestCase
+
+
+class MarionetteServiceWorkerTestCase(MarionetteTestCase):
+ def install_service_worker(self, path):
+ install_url = self.marionette.absolute_url(path)
+ self.marionette.navigate(install_url)
+ Wait(self.marionette).until(
+ lambda _: self.is_service_worker_registered,
+ message="Service worker not successfully installed",
+ )
+
+ # Wait for the registered service worker to be stored in the Firefox
+ # profile before restarting the instance to prevent intermittent
+ # failures (Bug 1665184).
+ Wait(self.marionette, timeout=10).until(
+ lambda _: self.profile_serviceworker_txt_exists,
+ message="Service worker not stored in profile",
+ )
+
+ # self.marionette.restart(in_app=True) will restore service workers if
+ # we don't navigate away before restarting.
+ self.marionette.navigate("about:blank")
+
+ # Using @property helps avoid the case where missing parens at the call site
+ # yields an unvarying 'true' value.
+ @property
+ def profile_serviceworker_txt_exists(self):
+ return "serviceworker.txt" in os.listdir(self.marionette.profile_path)
+
+ @property
+ def is_service_worker_registered(self):
+ with self.marionette.using_context("chrome"):
+ return self.marionette.execute_script(
+ """
+ let swm = Cc["@mozilla.org/serviceworkers/manager;1"].getService(
+ Ci.nsIServiceWorkerManager
+ );
+ let ssm = Services.scriptSecurityManager;
+
+ let principal = ssm.createContentPrincipalFromOrigin(arguments[0]);
+
+ let serviceWorkers = swm.getAllRegistrations();
+ for (let i = 0; i < serviceWorkers.length; i++) {
+ let sw = serviceWorkers.queryElementAt(
+ i,
+ Ci.nsIServiceWorkerRegistrationInfo
+ );
+ if (sw.principal.origin == principal.origin) {
+ return true;
+ }
+ }
+ return false;
+ """,
+ script_args=(self.marionette.absolute_url(""),),
+ )
diff --git a/dom/workers/test/marionette/test_service_workers_at_startup.py b/dom/workers/test/marionette/test_service_workers_at_startup.py
new file mode 100644
index 0000000000..14ff32f87f
--- /dev/null
+++ b/dom/workers/test/marionette/test_service_workers_at_startup.py
@@ -0,0 +1,31 @@
+# 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
+
+# Add this directory to the import path.
+sys.path.append(os.path.dirname(__file__))
+
+from marionette_driver import Wait
+from service_worker_utils import MarionetteServiceWorkerTestCase
+
+
+class ServiceWorkerAtStartupTestCase(MarionetteServiceWorkerTestCase):
+ def setUp(self):
+ super(ServiceWorkerAtStartupTestCase, self).setUp()
+ self.install_service_worker("serviceworker/install_serviceworker.html")
+
+ def tearDown(self):
+ self.marionette.restart(in_app=False, clean=True)
+ super(ServiceWorkerAtStartupTestCase, self).tearDown()
+
+ def test_registered_service_worker_after_restart(self):
+ self.marionette.restart()
+
+ Wait(self.marionette).until(
+ lambda _: self.is_service_worker_registered,
+ message="Service worker not registered after restart",
+ )
+ self.assertTrue(self.is_service_worker_registered)
diff --git a/dom/workers/test/marionette/test_service_workers_disabled.py b/dom/workers/test/marionette/test_service_workers_disabled.py
new file mode 100644
index 0000000000..deed164242
--- /dev/null
+++ b/dom/workers/test/marionette/test_service_workers_disabled.py
@@ -0,0 +1,37 @@
+# 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
+
+# Add this directory to the import path.
+sys.path.append(os.path.dirname(__file__))
+
+from service_worker_utils import MarionetteServiceWorkerTestCase
+
+
+class ServiceWorkersDisabledTestCase(MarionetteServiceWorkerTestCase):
+ def setUp(self):
+ super(ServiceWorkersDisabledTestCase, self).setUp()
+ self.install_service_worker("serviceworker/install_serviceworker.html")
+
+ def tearDown(self):
+ self.marionette.restart(in_app=False, clean=True)
+ super(ServiceWorkersDisabledTestCase, self).tearDown()
+
+ def test_service_workers_disabled_at_startup(self):
+ # self.marionette.set_pref sets preferences after startup. Using it
+ # here causes intermittent failures.
+ self.marionette.instance.profile.set_preferences(
+ {
+ "dom.serviceWorkers.enabled": False,
+ }
+ )
+
+ self.marionette.restart()
+
+ self.assertFalse(
+ self.is_service_worker_registered,
+ "Service worker registration should have been purged",
+ )