Adding upstream version 86.0.1.upstream/86.0.1upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

4 files changed, 418 insertions, 0 deletions

diff --git a/toolkit/components/cleardata/tests/browser/browser.ini b/toolkit/components/cleardata/tests/browser/browser.ini
new file mode 100644
index 0000000000..92373c3ec7
--- /dev/null
+++ b/toolkit/components/cleardata/tests/browser/browser.ini
@@ -0,0 +1,3 @@
+[browser_serviceworkers.js]
+[browser_quota.js]
+support-files = worker.js
diff --git a/toolkit/components/cleardata/tests/browser/browser_quota.js b/toolkit/components/cleardata/tests/browser/browser_quota.js
new file mode 100644
index 0000000000..bbbe986fa7
--- /dev/null
+++ b/toolkit/components/cleardata/tests/browser/browser_quota.js
@@ -0,0 +1,238 @@
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+// This function adds the quota storage by simpleDB (one of quota clients
+// managed by the QuotaManager). In this function, a directory
+// ${profile}/storage/default/${origin}/sdb/ and a file inside are expected to
+// be added.
+async function addQuotaStorage(principal) {
+  let connection = Cc["@mozilla.org/dom/sdb-connection;1"].createInstance(
+    Ci.nsISDBConnection
+  );
+
+  connection.init(principal);
+
+  await new Promise((aResolve, aReject) => {
+    let request = connection.open("db");
+    request.callback = request => {
+      if (request.resultCode == Cr.NS_OK) {
+        aResolve();
+      } else {
+        aReject(request.resultCode);
+      }
+    };
+  });
+
+  await new Promise((aResolve, aReject) => {
+    let request = connection.write(new ArrayBuffer(1));
+    request.callback = request => {
+      if (request.resultCode == Cr.NS_OK) {
+        aResolve();
+      } else {
+        aReject(request.resultCode);
+      }
+    };
+  });
+}
+
+function getPrincipal(url, attr = {}) {
+  let uri = Services.io.newURI(url);
+  let ssm = Services.scriptSecurityManager;
+
+  return ssm.createContentPrincipal(uri, attr);
+}
+
+function getProfileDir() {
+  let directoryService = Services.dirsvc;
+
+  return directoryService.get("ProfD", Ci.nsIFile);
+}
+
+function getRelativeFile(relativePath) {
+  let profileDir = getProfileDir();
+
+  let file = profileDir.clone();
+  relativePath.split("/").forEach(function(component) {
+    file.append(component);
+  });
+
+  return file;
+}
+
+function getPath(origin) {
+  // Santizing
+  let regex = /[:\/]/g;
+  return "storage/default/" + origin.replace(regex, "+");
+}
+
+// This function checks if the origin has the quota storage by checking whether
+// the origin directory of that origin exists or not.
+function hasQuotaStorage(origin, attr) {
+  let path = getPath(origin);
+  if (attr) {
+    path = path + "^userContextId=" + attr.userContextId;
+  }
+
+  let file = getRelativeFile(path);
+  return file.exists();
+}
+
+async function runTest(sites, deleteDataFunc) {
+  info(`Adding quota storage`);
+  for (let site of sites) {
+    const principal = getPrincipal(site.origin, site.originAttributes);
+    await addQuotaStorage(principal);
+  }
+
+  info(`Verifying ${deleteDataFunc.name}`);
+  let site;
+  while ((site = sites.shift())) {
+    await new Promise(aResolve => {
+      deleteDataFunc(...site.args, value => {
+        Assert.equal(value, 0);
+        aResolve();
+      });
+    });
+
+    ok(
+      !hasQuotaStorage(site.origin, site.originAttributes),
+      `${site.origin} has no quota storage`
+    );
+    sites.forEach(remainSite =>
+      ok(
+        hasQuotaStorage(remainSite.origin, remainSite.originAttributes),
+        `${remainSite.origin} has quota storage`
+      )
+    );
+  }
+}
+
+add_task(async function setup() {
+  await SpecialPowers.pushPrefEnv({
+    set: [
+      ["dom.quotaManager.testing", true],
+      ["dom.simpleDB.enabled", true],
+    ],
+  });
+});
+
+const ORG_DOMAIN = "example.com";
+const ORG_ORIGIN = `https://${ORG_DOMAIN}`;
+const COM_DOMAIN = "example.org";
+const COM_ORIGIN = `https://${COM_DOMAIN}`;
+const LH_DOMAIN = "localhost";
+const FOO_DOMAIN = "foo.com";
+
+add_task(async function test_deleteFromHost() {
+  const sites = [
+    {
+      args: [ORG_DOMAIN, true, Ci.nsIClearDataService.CLEAR_DOM_QUOTA],
+      origin: ORG_ORIGIN,
+    },
+    {
+      args: [COM_DOMAIN, true, Ci.nsIClearDataService.CLEAR_DOM_QUOTA],
+      origin: COM_ORIGIN,
+    },
+    {
+      args: [LH_DOMAIN, true, Ci.nsIClearDataService.CLEAR_DOM_QUOTA],
+      origin: `http://${LH_DOMAIN}:8000`,
+    },
+    {
+      args: [FOO_DOMAIN, true, Ci.nsIClearDataService.CLEAR_DOM_QUOTA],
+      origin: `http://${FOO_DOMAIN}`,
+      originAttributes: { userContextId: 1 },
+    },
+  ];
+
+  await runTest(sites, Services.clearData.deleteDataFromHost);
+});
+
+add_task(async function test_deleteFromPrincipal() {
+  const sites = [
+    {
+      args: [
+        getPrincipal(ORG_ORIGIN),
+        true,
+        Ci.nsIClearDataService.CLEAR_DOM_QUOTA,
+      ],
+      origin: ORG_ORIGIN,
+    },
+    {
+      args: [
+        getPrincipal(COM_ORIGIN),
+        true,
+        Ci.nsIClearDataService.CLEAR_DOM_QUOTA,
+      ],
+      origin: COM_ORIGIN,
+    },
+  ];
+
+  await runTest(sites, Services.clearData.deleteDataFromPrincipal);
+});
+
+add_task(async function test_deleteFromOriginAttributes() {
+  const ORG_OA = { userContextId: 1 };
+  const COM_OA = { userContextId: 2 };
+  const sites = [
+    {
+      args: [ORG_OA],
+      origin: ORG_ORIGIN,
+      originAttributes: ORG_OA,
+    },
+    {
+      args: [COM_OA],
+      origin: COM_ORIGIN,
+      originAttributes: COM_OA,
+    },
+  ];
+
+  await runTest(
+    sites,
+    Services.clearData.deleteDataFromOriginAttributesPattern
+  );
+});
+
+add_task(async function test_deleteAll() {
+  info(`Adding quota storage`);
+  await addQuotaStorage(getPrincipal(ORG_ORIGIN));
+  await addQuotaStorage(getPrincipal(COM_ORIGIN));
+
+  info(`Verifying deleteData`);
+  await new Promise(aResolve => {
+    Services.clearData.deleteData(
+      Ci.nsIClearDataService.CLEAR_DOM_QUOTA,
+      value => {
+        Assert.equal(value, 0);
+        aResolve();
+      }
+    );
+  });
+
+  ok(!hasQuotaStorage(ORG_ORIGIN), `${ORG_ORIGIN} has no quota storage`);
+  ok(!hasQuotaStorage(COM_ORIGIN), `${COM_ORIGIN} has no quota storage`);
+});
+
+add_task(async function test_deleteSubdomain() {
+  const ANOTHER_ORIGIN = `https://wwww.${ORG_DOMAIN}`;
+  info(`Adding quota storage`);
+  await addQuotaStorage(getPrincipal(ORG_ORIGIN));
+  await addQuotaStorage(getPrincipal(ANOTHER_ORIGIN));
+
+  info(`Verifying deleteDataFromHost for subdomain`);
+  await new Promise(aResolve => {
+    Services.clearData.deleteDataFromHost(
+      ORG_DOMAIN,
+      true,
+      Ci.nsIClearDataService.CLEAR_DOM_QUOTA,
+      value => {
+        Assert.equal(value, 0);
+        aResolve();
+      }
+    );
+  });
+
+  ok(!hasQuotaStorage(ORG_ORIGIN), `${ORG_ORIGIN} has no quota storage`);
+  ok(!hasQuotaStorage(COM_ORIGIN), `${ANOTHER_ORIGIN} has no quota storage`);
+});
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..19a44cc11b
--- /dev/null
+++ b/toolkit/components/cleardata/tests/browser/browser_serviceworkers.js
@@ -0,0 +1,176 @@
+/* Any copyright is dedicated to the Public Domain.
+   http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+const { SiteDataTestUtils } = ChromeUtils.import(
+  "resource://testing-common/SiteDataTestUtils.jsm"
+);
+
+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_task(async function setup() {
+  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");
+
+  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"
+  );
+
+  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_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();
+});
diff --git a/toolkit/components/cleardata/tests/browser/worker.js b/toolkit/components/cleardata/tests/browser/worker.js
new file mode 100644
index 0000000000..aa8a83a4ce
--- /dev/null
+++ b/toolkit/components/cleardata/tests/browser/worker.js
@@ -0,0 +1 @@
+// Empty script for testing service workers