summaryrefslogtreecommitdiffstats
path: root/toolkit/components/extensions/test/mochitest/test_ext_browsingData_indexedDB.html
blob: f8ea41ddabe01134447c09a888982e53a65c83fe (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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
<!-- Any copyright is dedicated to the Public Domain.
   - http://creativecommons.org/publicdomain/zero/1.0/ -->
<!DOCTYPE HTML>
<html>
<head>
  <title>Test browsingData.remove indexedDB</title>
  <script src="/tests/SimpleTest/SimpleTest.js"></script>
  <script src="/tests/SimpleTest/ExtensionTestUtils.js"></script>
  <script src="head.js"></script>
  <link rel="stylesheet" type="text/css" href="/tests/SimpleTest/test.css" />
</head>
<body>

<script type="text/javascript">
"use strict";

add_task(async function testIndexedDB() {
  await SpecialPowers.pushPrefEnv({
    set: [["privacy.userContext.enabled", true]],
  });

  async function background() {
    const PAGE =
      "/tests/toolkit/components/extensions/test/mochitest/file_indexedDB.html";

    let tabs = [];

    browser.test.onMessage.addListener(async msg => {
      if (msg == "cleanup") {
        await Promise.all(tabs.map(tabId => browser.tabs.remove(tabId)));
        browser.test.sendMessage("done");
        return;
      }

      await browser.browsingData.remove(msg, { indexedDB: true });
      browser.test.sendMessage("indexedDBRemoved");
    });

    // Create two tabs.
    let tab = await browser.tabs.create({ url: `https://example.org${PAGE}` });
    tabs.push(tab.id);

    tab = await browser.tabs.create({ url: `https://example.com${PAGE}` });
    tabs.push(tab.id);

    // Create tab with cookieStoreId "firefox-container-1"
    tab = await browser.tabs.create({ url: `https://example.net${PAGE}`, cookieStoreId: 'firefox-container-1' });
    tabs.push(tab.id);
  }

  function contentScript() {
    // eslint-disable-next-line mozilla/balanced-listeners
    window.addEventListener(
      "message",
      msg => {
        browser.test.sendMessage("indexedDBCreated");
      },
      true
    );
  }

  let extension = ExtensionTestUtils.loadExtension({
    background,
    manifest: {
      permissions: ["browsingData", "tabs", "cookies"],
      content_scripts: [
        {
          matches: [
            "https://example.org/*/file_indexedDB.html",
            "https://example.com/*/file_indexedDB.html",
            "https://example.net/*/file_indexedDB.html",
          ],
          js: ["script.js"],
          run_at: "document_start",
        },
      ],
    },
    files: {
      "script.js": contentScript,
    },
  });

  await extension.startup();

  await extension.awaitMessage("indexedDBCreated");
  await extension.awaitMessage("indexedDBCreated");
  await extension.awaitMessage("indexedDBCreated");

  function getUsage() {
    return new Promise(resolve => {
      let qms = SpecialPowers.Services.qms;
      let cb = SpecialPowers.wrapCallback(request => resolve(request.result));
      qms.getUsage(cb);
    });
  }

  async function getOrigins() {
    let origins = [];
    let result = await getUsage();
    for (let i = 0; i < result.length; ++i) {
      if (result[i].usage === 0) {
        continue;
      }
      if (
        result[i].origin.startsWith("https://example.org") ||
        result[i].origin.startsWith("https://example.com") ||
        result[i].origin.startsWith("https://example.net")
      ) {
        origins.push(result[i].origin);
      }
    }
    return origins.sort();
  }

  let origins = await getOrigins();
  is(origins.length, 3, "IndexedDB databases have been populated.");

  // Deleting private browsing mode data is silently ignored.
  extension.sendMessage({ cookieStoreId: "firefox-private" });
  await extension.awaitMessage("indexedDBRemoved");

  origins = await getOrigins();
  is(origins.length, 3, "All indexedDB remains after clearing firefox-private");

  // Delete by hostname
  extension.sendMessage({ hostnames: ["example.com"] });
  await extension.awaitMessage("indexedDBRemoved");

  origins = await getOrigins();
  is(origins.length, 2, "IndexedDB data only for only two domains left");
  ok(origins[0].startsWith("https://example.net"), "example.net not deleted");
  ok(origins[1].startsWith("https://example.org"), "example.org not deleted");

  // TODO: Bug 1643740
  if (AppConstants.platform != "android") {
    // Delete by cookieStoreId
    extension.sendMessage({ cookieStoreId: "firefox-container-1" });
    await extension.awaitMessage("indexedDBRemoved");

    origins = await getOrigins();
    is(origins.length, 1, "IndexedDB data only for only one domain");
    ok(origins[0].startsWith("https://example.org"), "example.org not deleted");
  }

  // Delete all
  extension.sendMessage({});
  await extension.awaitMessage("indexedDBRemoved");

  origins = await getOrigins();
  is(origins.length, 0, "All IndexedDB data has been removed.");

  await extension.sendMessage("cleanup");
  await extension.awaitMessage("done");

  await extension.unload();
});
</script>
</body>
</html>