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
|
<!DOCTYPE HTML>
<html>
<head>
<title>Test for simple WebExtension</title>
<script type="text/javascript" src="/tests/SimpleTest/SimpleTest.js"></script>
<script type="text/javascript" src="/tests/SimpleTest/SpawnTask.js"></script>
<script type="text/javascript" src="/tests/SimpleTest/ExtensionTestUtils.js"></script>
<script type="text/javascript" src="/tests/SimpleTest/EventUtils.js"></script>
<script type="text/javascript" src="head.js"></script>
<script type="text/javascript" src="head_unlimitedStorage.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 test_background_storagePersist() {
await SpecialPowers.pushPrefEnv({
"set": [
["dom.storageManager.enabled", true],
// Enable the storageManager permission prompt.
["browser.storageManager.enabled", true],
["dom.storageManager.prompt.testing", false],
["dom.storageManager.prompt.testing.allow", false],
],
});
const EXTENSION_ID = "test-storagePersist@mozilla";
const extension = ExtensionTestUtils.loadExtension({
useAddonManager: "permanent",
manifest: {
permissions: ["unlimitedStorage"],
applications: {
gecko: {
id: EXTENSION_ID,
},
},
},
background: async function() {
const PROMISE_RACE_TIMEOUT = 8000;
browser.test.sendMessage("extension-uuid", window.location.host);
const requestStoragePersist = async () => {
const persistAllowed = await navigator.storage.persist();
if (!persistAllowed) {
throw new Error("navigator.storage.persist() has been denied");
}
};
await Promise.race([
requestStoragePersist(),
new Promise((resolve, reject) => {
setTimeout(() => {
reject(new Error("Timeout opening persistent db from background page"));
}, PROMISE_RACE_TIMEOUT);
}),
]).then(
() => {
browser.test.notifyPass("indexeddb-storagePersistent-unlimitedStorage-done");
},
(error) => {
browser.test.fail(`error while testing persistent IndexedDB storage: ${error}`);
browser.test.notifyFail("indexeddb-storagePersistent-unlimitedStorage-done");
}
);
},
});
await extension.startup();
const uuid = await extension.awaitMessage("extension-uuid");
await extension.awaitFinish("indexeddb-storagePersistent-unlimitedStorage-done");
await extension.unload();
checkSitePermissions(uuid, Services.perms.UNKNOWN_ACTION, "has been cleared");
});
add_task(async function test_unlimitedStorage_removed_on_update() {
const EXTENSION_ID = "test-unlimitedStorage-removed-on-update@mozilla";
let extension = ExtensionTestUtils.loadExtension({
manifest: {
permissions: ["unlimitedStorage"],
applications: {
gecko: {
id: EXTENSION_ID,
},
},
},
background: async function() {
browser.test.sendMessage("extension-uuid", window.location.host);
},
});
await extension.startup();
const uuid = await extension.awaitMessage("extension-uuid");
checkSitePermissions(uuid, Services.perms.ALLOW_ACTION, "has been allowed");
await extension.unload();
// Simulate an update which do not require the unlimitedStorage permission.
let updatedExtension = ExtensionTestUtils.loadExtension({
manifest: {
applications: {
gecko: {
id: EXTENSION_ID,
},
},
},
background: async function() {
browser.test.sendMessage("updated-extension-uuid", window.location.host);
},
});
updatedExtension.startup();
const updatedExtensionUUID = await updatedExtension.awaitMessage("updated-extension-uuid");
is(uuid, updatedExtensionUUID, "The updated extension has the same uuid");
checkSitePermissions(uuid, Services.perms.UNKNOWN_ACTION, "has been cleared");
await updatedExtension.unload();
});
</script>
</body>
</html>
|
