1 files changed, 131 insertions, 0 deletions
diff --git a/dom/tests/mochitest/localstorage/localStorageCommon.js b/dom/tests/mochitest/localstorage/localStorageCommon.js
new file mode 100644
index 0000000000..364e708870
--- /dev/null
+++ b/dom/tests/mochitest/localstorage/localStorageCommon.js
@@ -0,0 +1,131 @@
+function localStorageFlush(cb) {
+  if (SpecialPowers.Services.domStorageManager.nextGenLocalStorageEnabled) {
+    SimpleTest.executeSoon(function () {
+      cb();
+    });
+    return;
+  }
+
+  var ob = {
+    observe(sub, top, dat) {
+      os().removeObserver(ob, "domstorage-test-flushed");
+      cb();
+    },
+  };
+  os().addObserver(ob, "domstorage-test-flushed");
+  notify("domstorage-test-flush-force");
+}
+
+function localStorageReload(callback) {
+  if (SpecialPowers.Services.domStorageManager.nextGenLocalStorageEnabled) {
+    localStorage.close();
+    let qms = SpecialPowers.Services.qms;
+    let principal = SpecialPowers.wrap(document).nodePrincipal;
+    let request = qms.resetStoragesForPrincipal(principal, "default", "ls");
+    request.callback = SpecialPowers.wrapCallback(function () {
+      localStorage.open();
+      callback();
+    });
+    return;
+  }
+
+  notify("domstorage-test-reload");
+  SimpleTest.executeSoon(function () {
+    callback();
+  });
+}
+
+function localStorageFlushAndReload(callback) {
+  if (SpecialPowers.Services.domStorageManager.nextGenLocalStorageEnabled) {
+    localStorage.close();
+    let qms = SpecialPowers.Services.qms;
+    let principal = SpecialPowers.wrap(document).nodePrincipal;
+    let request = qms.resetStoragesForPrincipal(principal, "default", "ls");
+    request.callback = SpecialPowers.wrapCallback(function () {
+      localStorage.open();
+      callback();
+    });
+    return;
+  }
+
+  localStorageFlush(function () {
+    localStorageReload(callback);
+  });
+}
+
+function localStorageClearAll(callback) {
+  if (SpecialPowers.Services.domStorageManager.nextGenLocalStorageEnabled) {
+    let qms = SpecialPowers.Services.qms;
+    let ssm = SpecialPowers.Services.scriptSecurityManager;
+
+    qms.getUsage(
+      SpecialPowers.wrapCallback(function (request) {
+        if (request.resultCode != SpecialPowers.Cr.NS_OK) {
+          callback();
+          return;
+        }
+
+        let clearRequestCount = 0;
+        for (let item of request.result) {
+          let principal = ssm.createContentPrincipalFromOrigin(item.origin);
+          let clearRequest = qms.clearStoragesForPrincipal(
+            principal,
+            "default",
+            "ls"
+          );
+          clearRequestCount++;
+          clearRequest.callback = SpecialPowers.wrapCallback(function () {
+            if (--clearRequestCount == 0) {
+              callback();
+            }
+          });
+        }
+      })
+    );
+    return;
+  }
+
+  os().notifyObservers(null, "cookie-changed", "cleared");
+  SimpleTest.executeSoon(function () {
+    callback();
+  });
+}
+
+function localStorageClearDomain(domain, callback) {
+  if (SpecialPowers.Services.domStorageManager.nextGenLocalStorageEnabled) {
+    let qms = SpecialPowers.Services.qms;
+    let principal = SpecialPowers.wrap(document).effectiveStoragePrincipal;
+    let request = qms.clearStoragesForPrincipal(principal, "default", "ls");
+    let cb = SpecialPowers.wrapCallback(callback);
+    request.callback = cb;
+    return;
+  }
+
+  os().notifyObservers(null, "extension:purge-localStorage", domain);
+  SimpleTest.executeSoon(function () {
+    callback();
+  });
+}
+
+function os() {
+  return SpecialPowers.Services.obs;
+}
+
+function notify(top) {
+  os().notifyObservers(null, top);
+}
+
+/**
+ * Enable testing observer notifications in DOMStorageObserver.cpp.
+ */
+function localStorageEnableTestingMode(cb) {
+  SpecialPowers.pushPrefEnv(
+    {
+      set: [
+        ["dom.storage.testing", true],
+        ["dom.quotaManager.testing", true],
+      ],
+    },
+    cb
+  );
+}