summaryrefslogtreecommitdiffstats
path: root/dom/localstorage/test/unit/test_migration.js
blob: 1249eb076fa137dac074b78ff41d05ac631fc03e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
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);
  }
});