1 files changed, 287 insertions, 0 deletions
diff --git a/toolkit/components/cleardata/tests/browser/browser_serviceworkers.js b/toolkit/components/cleardata/tests/browser/browser_serviceworkers.js
new file mode 100644
index 0000000000..ad501aed70
--- /dev/null
+++ b/toolkit/components/cleardata/tests/browser/browser_serviceworkers.js
@@ -0,0 +1,287 @@
+/* Any copyright is dedicated to the Public Domain.
+   http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+const { SiteDataTestUtils } = ChromeUtils.importESModule(
+  "resource://testing-common/SiteDataTestUtils.sys.mjs"
+);
+
+async function addServiceWorker(origin) {
+  let swURL =
+    getRootDirectory(gTestPath).replace("chrome://mochitests/content", origin) +
+    "worker.js";
+
+  let registered = SiteDataTestUtils.promiseServiceWorkerRegistered(swURL);
+  await SiteDataTestUtils.addServiceWorker(swURL);
+  await registered;
+
+  ok(true, `${origin} has a service worker`);
+}
+
+add_setup(async function () {
+  await SpecialPowers.pushPrefEnv({
+    set: [
+      ["dom.serviceWorkers.enabled", true],
+      ["dom.serviceWorkers.exemptFromPerDomainMax", true],
+      ["dom.serviceWorkers.testing.enabled", true],
+    ],
+  });
+});
+
+add_task(async function test_deleteFromHost() {
+  await addServiceWorker("https://example.com");
+  await addServiceWorker("https://example.org");
+  await addServiceWorker("https://test1.example.org");
+
+  let unregistered = SiteDataTestUtils.promiseServiceWorkerUnregistered(
+    "https://example.com"
+  );
+  await new Promise(aResolve => {
+    Services.clearData.deleteDataFromHost(
+      "example.com",
+      true,
+      Ci.nsIClearDataService.CLEAR_DOM_QUOTA,
+      value => {
+        Assert.equal(value, 0);
+        aResolve();
+      }
+    );
+  });
+  await unregistered;
+
+  ok(
+    !SiteDataTestUtils.hasServiceWorkers("https://example.com"),
+    "example.com has no service worker"
+  );
+  ok(
+    SiteDataTestUtils.hasServiceWorkers("https://example.org"),
+    "example.org has a service worker"
+  );
+  ok(
+    SiteDataTestUtils.hasServiceWorkers("https://test1.example.org"),
+    "test1.example.org has a service worker"
+  );
+
+  // Clearing the subdomain should not clear the base domain.
+  unregistered = SiteDataTestUtils.promiseServiceWorkerUnregistered(
+    "https://test1.example.org"
+  );
+  await new Promise(aResolve => {
+    Services.clearData.deleteDataFromHost(
+      "test1.example.org",
+      true,
+      Ci.nsIClearDataService.CLEAR_DOM_QUOTA,
+      value => {
+        Assert.equal(value, 0);
+        aResolve();
+      }
+    );
+  });
+  await unregistered;
+
+  ok(
+    !SiteDataTestUtils.hasServiceWorkers("https://test1.example.org"),
+    "test1.example.org has no service worker"
+  );
+  ok(
+    SiteDataTestUtils.hasServiceWorkers("https://example.org"),
+    "example.org has a service worker"
+  );
+
+  unregistered = SiteDataTestUtils.promiseServiceWorkerUnregistered(
+    "https://example.org"
+  );
+  await new Promise(aResolve => {
+    Services.clearData.deleteDataFromHost(
+      "example.org",
+      true,
+      Ci.nsIClearDataService.CLEAR_DOM_QUOTA,
+      value => {
+        Assert.equal(value, 0);
+        aResolve();
+      }
+    );
+  });
+  await unregistered;
+
+  ok(
+    !SiteDataTestUtils.hasServiceWorkers("https://example.org"),
+    "example.org has no service worker"
+  );
+  ok(
+    !SiteDataTestUtils.hasServiceWorkers("https://example.com"),
+    "example.com has no service worker"
+  );
+});
+
+add_task(async function test_deleteFromBaseDomain() {
+  await addServiceWorker("https://example.com");
+  await addServiceWorker("https://test1.example.com");
+  await addServiceWorker("https://test2.example.com");
+  await addServiceWorker("https://example.org");
+
+  let unregistered = SiteDataTestUtils.promiseServiceWorkerUnregistered(
+    "https://example.com"
+  );
+  let unregisteredSub1 = SiteDataTestUtils.promiseServiceWorkerUnregistered(
+    "https://test1.example.com"
+  );
+  let unregisteredSub2 = SiteDataTestUtils.promiseServiceWorkerUnregistered(
+    "https://test1.example.com"
+  );
+  await new Promise(aResolve => {
+    Services.clearData.deleteDataFromBaseDomain(
+      "example.com",
+      true,
+      Ci.nsIClearDataService.CLEAR_DOM_QUOTA,
+      value => {
+        Assert.equal(value, 0);
+        aResolve();
+      }
+    );
+  });
+  await Promise.all([unregistered, unregisteredSub1, unregisteredSub2]);
+
+  ok(
+    !SiteDataTestUtils.hasServiceWorkers("https://example.com"),
+    "example.com has no service worker"
+  );
+  ok(
+    !SiteDataTestUtils.hasServiceWorkers("https://test1.example.com"),
+    "test1.example.com has no service worker"
+  );
+  ok(
+    !SiteDataTestUtils.hasServiceWorkers("https://test2.example.com"),
+    "test2.example.com has no service worker"
+  );
+  ok(
+    SiteDataTestUtils.hasServiceWorkers("https://example.org"),
+    "example.org has a service worker"
+  );
+
+  unregistered = SiteDataTestUtils.promiseServiceWorkerUnregistered(
+    "https://example.org"
+  );
+  await new Promise(aResolve => {
+    Services.clearData.deleteDataFromBaseDomain(
+      "example.org",
+      true,
+      Ci.nsIClearDataService.CLEAR_DOM_QUOTA,
+      value => {
+        Assert.equal(value, 0);
+        aResolve();
+      }
+    );
+  });
+  await unregistered;
+
+  ok(
+    !SiteDataTestUtils.hasServiceWorkers("https://example.org"),
+    "example.org has no service worker"
+  );
+  ok(
+    !SiteDataTestUtils.hasServiceWorkers("https://example.com"),
+    "example.com has no service worker"
+  );
+  ok(
+    !SiteDataTestUtils.hasServiceWorkers("https://test1.example.com"),
+    "test1.example.com has no service worker"
+  );
+  ok(
+    !SiteDataTestUtils.hasServiceWorkers("https://test2.example.com"),
+    "test2.example.com has no service worker"
+  );
+});
+
+add_task(async function test_deleteFromPrincipal() {
+  await addServiceWorker("https://test1.example.com");
+  await addServiceWorker("https://test1.example.org");
+
+  let unregistered = SiteDataTestUtils.promiseServiceWorkerUnregistered(
+    "https://test1.example.com"
+  );
+  let principal =
+    Services.scriptSecurityManager.createContentPrincipalFromOrigin(
+      "https://test1.example.com/"
+    );
+  await new Promise(aResolve => {
+    Services.clearData.deleteDataFromPrincipal(
+      principal,
+      true,
+      Ci.nsIClearDataService.CLEAR_DOM_QUOTA,
+      value => {
+        Assert.equal(value, 0);
+        aResolve();
+      }
+    );
+  });
+  await unregistered;
+
+  ok(
+    !SiteDataTestUtils.hasServiceWorkers("https://test1.example.com"),
+    "test1.example.com has no service worker"
+  );
+  ok(
+    SiteDataTestUtils.hasServiceWorkers("https://test1.example.org"),
+    "test1.example.org has a service worker"
+  );
+
+  unregistered = SiteDataTestUtils.promiseServiceWorkerUnregistered(
+    "https://test1.example.org"
+  );
+  principal = Services.scriptSecurityManager.createContentPrincipalFromOrigin(
+    "https://test1.example.org/"
+  );
+  await new Promise(aResolve => {
+    Services.clearData.deleteDataFromPrincipal(
+      principal,
+      true,
+      Ci.nsIClearDataService.CLEAR_DOM_QUOTA,
+      value => {
+        Assert.equal(value, 0);
+        aResolve();
+      }
+    );
+  });
+  await unregistered;
+
+  ok(
+    !SiteDataTestUtils.hasServiceWorkers("https://test1.example.org"),
+    "test1.example.org has no service worker"
+  );
+  ok(
+    !SiteDataTestUtils.hasServiceWorkers("https://test1.example.com"),
+    "test1.example.com has no service worker"
+  );
+});
+
+add_task(async function test_deleteAll() {
+  await addServiceWorker("https://test2.example.com");
+  await addServiceWorker("https://test2.example.org");
+
+  let unregistered = SiteDataTestUtils.promiseServiceWorkerUnregistered(
+    "https://test2.example.com"
+  );
+  await new Promise(aResolve => {
+    Services.clearData.deleteData(
+      Ci.nsIClearDataService.CLEAR_DOM_QUOTA,
+      value => {
+        Assert.equal(value, 0);
+        aResolve();
+      }
+    );
+  });
+  await unregistered;
+
+  ok(
+    !SiteDataTestUtils.hasServiceWorkers("https://test2.example.com"),
+    "test2.example.com has no service worker"
+  );
+  ok(
+    !SiteDataTestUtils.hasServiceWorkers("https://test2.example.org"),
+    "test2.example.org has no service worker"
+  );
+
+  await SiteDataTestUtils.clear();
+});