7 files changed, 331 insertions, 0 deletions
diff --git a/dom/quota/test/modules/system/ModuleLoader.sys.mjs b/dom/quota/test/modules/system/ModuleLoader.sys.mjs
new file mode 100644
index 0000000000..7bddfd2760
--- /dev/null
+++ b/dom/quota/test/modules/system/ModuleLoader.sys.mjs
@@ -0,0 +1,63 @@
+/**
+ * Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/
+ */
+
+export function ModuleLoader(base, depth, proto) {
+  const modules = {};
+
+  const principal = Cc["@mozilla.org/systemprincipal;1"].createInstance(
+    Ci.nsIPrincipal
+  );
+
+  const sharedGlobalsandbox = Cu.Sandbox(principal, {
+    invisibleToDebugger: true,
+    sandboxName: "FS Module Loader",
+    sandboxPrototype: proto,
+    wantComponents: false,
+    wantGlobalProperties: [],
+    wantXrays: false,
+  });
+
+  const require = async function(id) {
+    if (modules[id]) {
+      return modules[id].exported_symbols;
+    }
+
+    const url = new URL(depth + id, base);
+
+    const module = Object.create(null, {
+      exported_symbols: {
+        configurable: false,
+        enumerable: true,
+        value: Object.create(null),
+        writable: true,
+      },
+    });
+
+    modules[id] = module;
+
+    const properties = {
+      require_module: require,
+      exported_symbols: module.exported_symbols,
+    };
+
+    // Create a new object in this sandbox, that will be used as the scope
+    // object for this particular module.
+    const sandbox = sharedGlobalsandbox.Object();
+    Object.assign(sandbox, properties);
+
+    Services.scriptloader.loadSubScript(url.href, sandbox);
+
+    return module.exported_symbols;
+  };
+
+  const returnObj = {
+    require: {
+      enumerable: true,
+      value: require,
+    },
+  };
+
+  return Object.create(null, returnObj);
+}
diff --git a/dom/quota/test/modules/system/StorageUtils.sys.mjs b/dom/quota/test/modules/system/StorageUtils.sys.mjs
new file mode 100644
index 0000000000..bda40a4747
--- /dev/null
+++ b/dom/quota/test/modules/system/StorageUtils.sys.mjs
@@ -0,0 +1,65 @@
+/**
+ * Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/
+ */
+
+class RequestError extends Error {
+  constructor(resultCode, resultName) {
+    super(`Request failed (code: ${resultCode}, name: ${resultName})`);
+    this.name = "RequestError";
+    this.resultCode = resultCode;
+    this.resultName = resultName;
+  }
+}
+
+export function setStoragePrefs(optionalPrefsToSet) {
+  const prefsToSet = [["dom.quotaManager.testing", true]];
+
+  if (Services.appinfo.OS === "WINNT") {
+    prefsToSet.push(["dom.quotaManager.useDOSDevicePathSyntax", true]);
+  }
+
+  if (optionalPrefsToSet) {
+    prefsToSet.push(...optionalPrefsToSet);
+  }
+
+  for (const pref of prefsToSet) {
+    Services.prefs.setBoolPref(pref[0], pref[1]);
+  }
+}
+
+export function clearStoragePrefs(optionalPrefsToClear) {
+  const prefsToClear = [
+    "dom.quotaManager.testing",
+    "dom.simpleDB.enabled",
+    "dom.storageManager.enabled",
+  ];
+
+  if (Services.appinfo.OS === "WINNT") {
+    prefsToClear.push("dom.quotaManager.useDOSDevicePathSyntax");
+  }
+
+  if (optionalPrefsToClear) {
+    prefsToClear.push(...optionalPrefsToClear);
+  }
+
+  for (const pref of prefsToClear) {
+    Services.prefs.clearUserPref(pref);
+  }
+}
+
+export async function clearStoragesForOrigin(principal) {
+  const request = Services.qms.clearStoragesForPrincipal(principal);
+
+  await new Promise(function(resolve) {
+    request.callback = function() {
+      resolve();
+    };
+  });
+
+  if (request.resultCode != Cr.NS_OK) {
+    throw new RequestError(request.resultCode, request.resultName);
+  }
+
+  return request.result;
+}
diff --git a/dom/quota/test/modules/system/WorkerDriver.sys.mjs b/dom/quota/test/modules/system/WorkerDriver.sys.mjs
new file mode 100644
index 0000000000..c03a94cf56
--- /dev/null
+++ b/dom/quota/test/modules/system/WorkerDriver.sys.mjs
@@ -0,0 +1,52 @@
+/**
+ * Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/
+ */
+
+export async function runTestInWorker(script, base, listener) {
+  return new Promise(function(resolve) {
+    const globalHeadUrl = new URL(
+      "resource://testing-common/dom/quota/test/modules/worker/head.js"
+    );
+
+    const worker = new Worker(globalHeadUrl.href);
+
+    worker.onmessage = function(event) {
+      const data = event.data;
+
+      switch (data.op) {
+        case "ok":
+          listener.onOk(data.value, data.message);
+          break;
+
+        case "is":
+          listener.onIs(data.a, data.b, data.message);
+          break;
+
+        case "info":
+          listener.onInfo(data.message);
+          break;
+
+        case "finish":
+          resolve();
+          break;
+
+        case "failure":
+          listener.onOk(false, "Worker had a failure: " + data.message);
+          resolve();
+          break;
+      }
+    };
+
+    worker.onerror = function(event) {
+      listener.onOk(false, "Worker had an error: " + event.data);
+      resolve();
+    };
+
+    const scriptUrl = new URL(script, base);
+
+    const localHeadUrl = new URL("head.js", scriptUrl);
+
+    worker.postMessage([localHeadUrl.href, scriptUrl.href]);
+  });
+}
diff --git a/dom/quota/test/modules/system/worker/.eslintrc.js b/dom/quota/test/modules/system/worker/.eslintrc.js
new file mode 100644
index 0000000000..505e079c57
--- /dev/null
+++ b/dom/quota/test/modules/system/worker/.eslintrc.js
@@ -0,0 +1,21 @@
+/**
+ * Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/
+ */
+
+"use strict";
+
+module.exports = {
+  env: {
+    worker: true,
+  },
+
+  overrides: [
+    {
+      files: ["head.js"],
+      env: {
+        worker: true,
+      },
+    },
+  ],
+};
diff --git a/dom/quota/test/modules/system/worker/Assert.js b/dom/quota/test/modules/system/worker/Assert.js
new file mode 100644
index 0000000000..7c7e2683ea
--- /dev/null
+++ b/dom/quota/test/modules/system/worker/Assert.js
@@ -0,0 +1,22 @@
+/**
+ * Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/
+ */
+
+const Assert = {
+  ok(value, message) {
+    postMessage({
+      op: "ok",
+      value: !!value,
+      message,
+    });
+  },
+  equal(a, b, message) {
+    postMessage({
+      op: "is",
+      a,
+      b,
+      message,
+    });
+  },
+};
diff --git a/dom/quota/test/modules/system/worker/ModuleLoader.js b/dom/quota/test/modules/system/worker/ModuleLoader.js
new file mode 100644
index 0000000000..6de1fbc299
--- /dev/null
+++ b/dom/quota/test/modules/system/worker/ModuleLoader.js
@@ -0,0 +1,52 @@
+/**
+ * Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/
+ */
+
+function ModuleLoader(base, depth, proto) {
+  const modules = {};
+
+  const require = async function(id) {
+    if (modules[id]) {
+      return modules[id].exported_symbols;
+    }
+
+    const url = new URL(depth + id, base);
+
+    const module = Object.create(null, {
+      exported_symbols: {
+        configurable: false,
+        enumerable: true,
+        value: Object.create(null),
+        writable: true,
+      },
+    });
+
+    modules[id] = module;
+
+    const xhr = new XMLHttpRequest();
+    xhr.open("GET", url.href, false);
+    xhr.responseType = "text";
+    xhr.send();
+
+    let source = xhr.responseText;
+
+    let code = new Function(
+      "require_module",
+      "exported_symbols",
+      `eval(arguments[2] + "\\n//# sourceURL=" + arguments[3] + "\\n")`
+    );
+    code(require, module.exported_symbols, source, url.href);
+
+    return module.exported_symbols;
+  };
+
+  const returnObj = {
+    require: {
+      enumerable: true,
+      value: require,
+    },
+  };
+
+  return Object.create(null, returnObj);
+}
diff --git a/dom/quota/test/modules/system/worker/head.js b/dom/quota/test/modules/system/worker/head.js
new file mode 100644
index 0000000000..1308508918
--- /dev/null
+++ b/dom/quota/test/modules/system/worker/head.js
@@ -0,0 +1,56 @@
+/**
+ * Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/
+ */
+
+// eslint-disable-next-line mozilla/no-define-cc-etc
+const Cr = {
+  NS_ERROR_NOT_IMPLEMENTED: 2147500033,
+};
+
+function add_task(func) {
+  if (!add_task.tasks) {
+    add_task.tasks = [];
+    add_task.index = 0;
+  }
+
+  add_task.tasks.push(func);
+}
+
+addEventListener("message", async function onMessage(event) {
+  function info(message) {
+    postMessage({ op: "info", message });
+  }
+
+  function executeSoon(callback) {
+    const channel = new MessageChannel();
+    channel.port1.postMessage("");
+    channel.port2.onmessage = function() {
+      callback();
+    };
+  }
+
+  function runNextTest() {
+    if (add_task.index < add_task.tasks.length) {
+      const task = add_task.tasks[add_task.index++];
+      info("add_task | Entering test " + task.name);
+      task()
+        .then(function() {
+          executeSoon(runNextTest);
+          info("add_task | Leaving test " + task.name);
+        })
+        .catch(function(ex) {
+          postMessage({ op: "failure", message: "" + ex });
+        });
+    } else {
+      postMessage({ op: "finish" });
+    }
+  }
+
+  removeEventListener("message", onMessage);
+
+  const data = event.data;
+  importScripts(...data);
+
+  executeSoon(runNextTest);
+});