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
|
/* -- Mode: indent-tabs-mode: nil; js-indent-level: 2 -- */
/* vim: set sts=2 sw=2 et tw=80: */
"use strict";
const { PermissionTestUtils } = ChromeUtils.importESModule(
"resource://testing-common/PermissionTestUtils.sys.mjs"
);
function openPermissionPopup() {
let promise = BrowserTestUtils.waitForEvent(
window,
"popupshown",
true,
event => event.target == gPermissionPanel._permissionPopup
);
gPermissionPanel._identityPermissionBox.click();
info("Wait permission popup to be shown");
return promise;
}
function closePermissionPopup() {
let promise = BrowserTestUtils.waitForEvent(
gPermissionPanel._permissionPopup,
"popuphidden"
);
gPermissionPanel._permissionPopup.hidePopup();
info("Wait permission popup to be hidden");
return promise;
}
async function testPermissionPopup({ expectPermissionHidden }) {
await openPermissionPopup();
if (expectPermissionHidden) {
let permissionsList = document.getElementById(
"permission-popup-permission-list"
);
is(
permissionsList.querySelectorAll(
".permission-popup-permission-label-persistent-storage"
).length,
0,
"Persistent storage Permission should be hidden"
);
}
await closePermissionPopup();
// We need to test this after the popup has been closed.
// The permission icon will be shown as long as the popup is open, event if
// no permissions are set.
let permissionsGrantedIcon = document.getElementById(
"permissions-granted-icon"
);
if (expectPermissionHidden) {
ok(
BrowserTestUtils.is_hidden(permissionsGrantedIcon),
"Permission Granted Icon is hidden"
);
} else {
ok(
BrowserTestUtils.is_visible(permissionsGrantedIcon),
"Permission Granted Icon is visible"
);
}
}
add_task(async function testPersistentStoragePermissionHidden() {
let extension = ExtensionTestUtils.loadExtension({
background() {
browser.test.sendMessage("url", browser.runtime.getURL("testpage.html"));
},
manifest: {
name: "Test Extension",
permissions: ["unlimitedStorage"],
},
files: {
"testpage.html": "<h1>Extension Test Page</h1>",
},
});
await extension.startup();
let url = await extension.awaitMessage("url");
await BrowserTestUtils.withNewTab("about:blank", async browser => {
// Wait the tab to be fully loade, then run the test on the permission prompt.
let loaded = BrowserTestUtils.browserLoaded(browser, false, url);
BrowserTestUtils.loadURIString(browser, url);
await loaded;
await testPermissionPopup({ expectPermissionHidden: true });
});
await extension.unload();
});
add_task(async function testPersistentStoragePermissionVisible() {
let extension = ExtensionTestUtils.loadExtension({
background() {
browser.test.sendMessage("url", browser.runtime.getURL("testpage.html"));
},
manifest: {
name: "Test Extension",
},
files: {
"testpage.html": "<h1>Extension Test Page</h1>",
},
});
await extension.startup();
let url = await extension.awaitMessage("url");
let policy = WebExtensionPolicy.getByID(extension.id);
let principal = policy.extension.principal;
PermissionTestUtils.add(
principal,
"persistent-storage",
Services.perms.ALLOW_ACTION
);
await BrowserTestUtils.withNewTab("about:blank", async browser => {
// Wait the tab to be fully loade, then run the test on the permission prompt.
let loaded = BrowserTestUtils.browserLoaded(browser, false, url);
BrowserTestUtils.loadURIString(browser, url);
await loaded;
await testPermissionPopup({ expectPermissionHidden: false });
});
await extension.unload();
});
|
