From 26a029d407be480d791972afb5975cf62c9360a6 Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@progress-linux.org>
Date: Fri, 19 Apr 2024 02:47:55 +0200
Subject: Adding upstream version 124.0.1.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
---
 dom/localstorage/test/unit/test_migration.js | 127 +++++++++++++++++++++++++++
 1 file changed, 127 insertions(+)
 create mode 100644 dom/localstorage/test/unit/test_migration.js

(limited to 'dom/localstorage/test/unit/test_migration.js')

diff --git a/dom/localstorage/test/unit/test_migration.js b/dom/localstorage/test/unit/test_migration.js
new file mode 100644
index 0000000000..1249eb076f
--- /dev/null
+++ b/dom/localstorage/test/unit/test_migration.js
@@ -0,0 +1,127 @@
+/**
+ * Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/
+ */
+
+add_task(async function testSteps() {
+  const principalInfos = [
+    { url: "http://localhost", attrs: {} },
+    { url: "http://www.mozilla.org", attrs: {} },
+    { url: "http://example.com", attrs: {} },
+    { url: "http://example.org", attrs: { userContextId: 5 } },
+
+    { url: "http://origin.test", attrs: {} },
+
+    { url: "http://prefix.test", attrs: {} },
+    { url: "http://prefix.test", attrs: { userContextId: 10 } },
+
+    { url: "http://pattern.test", attrs: { userContextId: 15 } },
+    { url: "http://pattern.test:8080", attrs: { userContextId: 15 } },
+    { url: "https://pattern.test", attrs: { userContextId: 15 } },
+  ];
+
+  const data = {
+    key: "foo",
+    value: "bar",
+  };
+
+  function verifyData(clearedOrigins) {
+    info("Getting storages");
+
+    let storages = [];
+    for (let i = 0; i < principalInfos.length; i++) {
+      let principalInfo = principalInfos[i];
+      let principal = getPrincipal(principalInfo.url, principalInfo.attrs);
+      let storage = getLocalStorage(principal);
+      storages.push(storage);
+    }
+
+    info("Verifying data");
+
+    for (let i = 0; i < storages.length; i++) {
+      let value = storages[i].getItem(data.key + i);
+      if (clearedOrigins.includes(i)) {
+        is(value, null, "Correct value");
+      } else {
+        is(value, data.value + i, "Correct value");
+      }
+    }
+  }
+
+  info("Setting pref");
+
+  Services.prefs.setBoolPref(
+    "dom.storage.enable_unsupported_legacy_implementation",
+    false
+  );
+
+  info("Stage 1 - Testing archived data migration");
+
+  info("Clearing");
+
+  let request = clear();
+  await requestFinished(request);
+
+  info("Installing package");
+
+  // The profile contains storage.sqlite and webappsstore.sqlite. The file
+  // create_db.js in the package was run locally, specifically it was
+  // temporarily added to xpcshell.ini and then executed:
+  // mach xpcshell-test --interactive dom/localstorage/test/unit/create_db.js
+  installPackage("migration_profile");
+
+  verifyData([]);
+
+  info("Stage 2 - Testing archived data clearing");
+
+  for (let type of ["origin", "prefix", "pattern"]) {
+    info("Clearing");
+
+    request = clear();
+    await requestFinished(request);
+
+    info("Installing package");
+
+    // See the comment for the first installPackage() call.
+    installPackage("migration_profile");
+
+    let clearedOrigins = [];
+
+    switch (type) {
+      case "origin": {
+        let principal = getPrincipal("http://origin.test", {});
+        request = clearOrigin(principal, "default");
+        await requestFinished(request);
+
+        clearedOrigins.push(4);
+
+        break;
+      }
+
+      case "prefix": {
+        let principal = getPrincipal("http://prefix.test", {});
+        request = clearOriginsByPrefix(principal, "default");
+        await requestFinished(request);
+
+        clearedOrigins.push(5, 6);
+
+        break;
+      }
+
+      case "pattern": {
+        request = clearOriginsByPattern(JSON.stringify({ userContextId: 15 }));
+        await requestFinished(request);
+
+        clearedOrigins.push(7, 8, 9);
+
+        break;
+      }
+
+      default: {
+        throw new Error("Unknown type: " + type);
+      }
+    }
+
+    verifyData(clearedOrigins);
+  }
+});
-- 
cgit v1.2.3