1 files changed, 119 insertions, 0 deletions
diff --git a/toolkit/components/extensions/test/browser/head_serviceworker.js b/toolkit/components/extensions/test/browser/head_serviceworker.js
new file mode 100644
index 0000000000..b2f9512e5a
--- /dev/null
+++ b/toolkit/components/extensions/test/browser/head_serviceworker.js
@@ -0,0 +1,119 @@
+"use strict";
+
+/* exported assert_background_serviceworker_pref_enabled,
+ *          getBackgroundServiceWorkerRegistration,
+ *          getServiceWorkerInfo, getServiceWorkerState,
+ *          waitForServiceWorkerRegistrationsRemoved, waitForServiceWorkerTerminated
+ */
+
+async function assert_background_serviceworker_pref_enabled() {
+  is(
+    WebExtensionPolicy.backgroundServiceWorkerEnabled,
+    true,
+    "Expect extensions.backgroundServiceWorker.enabled to be true"
+  );
+}
+
+// Return the name of the enum corresponding to the worker's state (ex: "STATE_ACTIVATED")
+// because nsIServiceWorkerInfo doesn't currently provide a comparable string-returning getter.
+function getServiceWorkerState(workerInfo) {
+  const map = Object.keys(workerInfo)
+    .filter(k => k.startsWith("STATE_"))
+    .reduce((map, name) => {
+      map.set(workerInfo[name], name);
+      return map;
+    }, new Map());
+  return map.has(workerInfo.state)
+    ? map.get(workerInfo.state)
+    : "state: ${workerInfo.state}";
+}
+
+function getServiceWorkerInfo(swRegInfo) {
+  const { evaluatingWorker, installingWorker, waitingWorker, activeWorker } =
+    swRegInfo;
+  return evaluatingWorker || installingWorker || waitingWorker || activeWorker;
+}
+
+async function waitForServiceWorkerTerminated(swRegInfo) {
+  info(`Wait all ${swRegInfo.scope} workers to be terminated`);
+
+  try {
+    await BrowserTestUtils.waitForCondition(
+      () => !getServiceWorkerInfo(swRegInfo)
+    );
+  } catch (err) {
+    const workerInfo = getServiceWorkerInfo(swRegInfo);
+    if (workerInfo) {
+      ok(
+        false,
+        `Error while waiting for workers for scope ${swRegInfo.scope} to be terminated. ` +
+          `Found a worker in state: ${getServiceWorkerState(workerInfo)}`
+      );
+      return;
+    }
+
+    throw err;
+  }
+}
+
+function getBackgroundServiceWorkerRegistration(extension) {
+  const policy = WebExtensionPolicy.getByHostname(extension.uuid);
+  const expectedSWScope = policy.getURL("/");
+  const expectedScriptURL = policy.extension.backgroundWorkerScript || "";
+
+  ok(
+    expectedScriptURL.startsWith(expectedSWScope),
+    `Extension does include a valid background.service_worker: ${expectedScriptURL}`
+  );
+
+  const swm = Cc["@mozilla.org/serviceworkers/manager;1"].getService(
+    Ci.nsIServiceWorkerManager
+  );
+
+  let swReg;
+  let regs = swm.getAllRegistrations();
+
+  for (let i = 0; i < regs.length; i++) {
+    let reg = regs.queryElementAt(i, Ci.nsIServiceWorkerRegistrationInfo);
+    if (reg.scriptSpec === expectedScriptURL) {
+      swReg = reg;
+      break;
+    }
+  }
+
+  ok(swReg, `Found service worker registration for ${expectedScriptURL}`);
+
+  is(
+    swReg.scope,
+    expectedSWScope,
+    "The extension background worker registration has the expected scope URL"
+  );
+
+  return swReg;
+}
+
+async function waitForServiceWorkerRegistrationsRemoved(extension) {
+  info(`Wait ${extension.id} service worker registration to be deleted`);
+  const swm = Cc["@mozilla.org/serviceworkers/manager;1"].getService(
+    Ci.nsIServiceWorkerManager
+  );
+  let baseURI = Services.io.newURI(`moz-extension://${extension.uuid}/`);
+  let principal = Services.scriptSecurityManager.createContentPrincipal(
+    baseURI,
+    {}
+  );
+
+  await BrowserTestUtils.waitForCondition(() => {
+    let regs = swm.getAllRegistrations();
+
+    for (let i = 0; i < regs.length; i++) {
+      let reg = regs.queryElementAt(i, Ci.nsIServiceWorkerRegistrationInfo);
+      if (principal.equals(reg.principal)) {
+        return false;
+      }
+    }
+
+    info(`All ${extension.id} service worker registrations are gone`);
+    return true;
+  }, `All ${extension.id} service worker registrations should be deleted`);
+}