1 files changed, 62 insertions, 0 deletions
diff --git a/toolkit/components/cleardata/tests/marionette/test_service_worker_at_shutdown.py b/toolkit/components/cleardata/tests/marionette/test_service_worker_at_shutdown.py
new file mode 100644
index 0000000000..9738c0ea17
--- /dev/null
+++ b/toolkit/components/cleardata/tests/marionette/test_service_worker_at_shutdown.py
@@ -0,0 +1,62 @@
+# 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_driver import Wait
+from marionette_harness import MarionetteTestCase
+
+
+class ServiceWorkerAtShutdownTestCase(MarionetteTestCase):
+    def setUp(self):
+        super(ServiceWorkerAtShutdownTestCase, self).setUp()
+        self.install_service_worker()
+        self.set_pref_to_delete_site_data_on_shutdown()
+
+    def tearDown(self):
+        self.marionette.restart(in_app=False, clean=True)
+        super(ServiceWorkerAtShutdownTestCase, self).tearDown()
+
+    def install_service_worker(self):
+        install_url = self.marionette.absolute_url(
+            "serviceworker/install_serviceworker.html"
+        )
+        self.marionette.navigate(install_url)
+        # Make sure 'install_url' is not loaded on startup, it would reinstall the service worker
+        dummy_url = self.marionette.absolute_url("dummy.html")
+        self.marionette.navigate(dummy_url)
+        Wait(self.marionette).until(lambda _: self.is_service_worker_registered)
+
+    def set_pref_to_delete_site_data_on_shutdown(self):
+        self.marionette.set_pref("privacy.sanitize.sanitizeOnShutdown", True)
+        self.marionette.set_pref("privacy.clearOnShutdown.offlineApps", True)
+
+    def test_unregistering_service_worker_when_clearing_data(self):
+        self.marionette.restart(clean=False, in_app=True)
+        self.assertFalse(self.is_service_worker_registered)
+
+    @property
+    def is_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 principal =
+                    Services.scriptSecurityManager.createContentPrincipalFromOrigin(arguments[0]);
+
+                let serviceWorkers = serviceWorkerManager.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(""),),
+            )