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
|
/* Any copyright is dedicated to the Public Domain.
* http://creativecommons.org/publicdomain/zero/1.0/
*/
"use strict";
const { AddonTestUtils } = ChromeUtils.importESModule(
"resource://testing-common/AddonTestUtils.sys.mjs"
);
AddonTestUtils.initMochitest(this);
const MY_CONTEXT = 2;
let gDidSeeChannel = false;
function check_channel(subject) {
if (!(subject instanceof Ci.nsIHttpChannel)) {
return;
}
let channel = subject.QueryInterface(Ci.nsIHttpChannel);
let uri = channel.URI;
if (!uri || !uri.spec.endsWith("amosigned.xpi")) {
return;
}
gDidSeeChannel = true;
ok(true, "Got request for " + uri.spec);
let loadInfo = channel.loadInfo;
is(
loadInfo.originAttributes.userContextId,
MY_CONTEXT,
"Got expected usercontextid"
);
}
// ----------------------------------------------------------------------------
// Tests we send the right cookies when installing through an InstallTrigger call
function test() {
// This test depends on InstallTrigger.install availability.
setInstallTriggerPrefs();
Harness.installConfirmCallback = confirm_install;
Harness.installEndedCallback = install_ended;
Harness.installsCompletedCallback = finish_test;
Harness.finalContentEvent = "InstallComplete";
Harness.setup();
let principal = Services.scriptSecurityManager.createContentPrincipal(
Services.io.newURI("http://example.com/"),
{ userContextId: MY_CONTEXT }
);
PermissionTestUtils.add(principal, "install", Services.perms.ALLOW_ACTION);
var triggers = encodeURIComponent(
JSON.stringify({
"Unsigned XPI": {
URL: TESTROOT + "amosigned.xpi",
IconURL: TESTROOT + "icon.png",
toString() {
return this.URL;
},
},
})
);
gBrowser.selectedTab = BrowserTestUtils.addTab(gBrowser, "", {
userContextId: MY_CONTEXT,
});
Services.obs.addObserver(check_channel, "http-on-before-connect");
BrowserTestUtils.loadURIString(
gBrowser,
TESTROOT + "installtrigger.html?" + triggers
);
}
function confirm_install(panel) {
is(panel.getAttribute("name"), "XPI Test", "Should have seen the name");
return true;
}
function install_ended(install, addon) {
AddonTestUtils.checkInstallInfo(install, {
method: "installTrigger",
source: "test-host",
sourceURL: /http:\/\/example.com\/.*\/installtrigger.html/,
});
return addon.uninstall();
}
const finish_test = async function (count) {
ok(
gDidSeeChannel,
"Should have seen the request for the XPI and verified it was sent the right way."
);
is(count, 1, "1 Add-on should have been successfully installed");
Services.obs.removeObserver(check_channel, "http-on-before-connect");
PermissionTestUtils.remove("http://example.com", "install");
const results = await SpecialPowers.spawn(
gBrowser.selectedBrowser,
[],
() => {
return {
return: content.document.getElementById("return").textContent,
status: content.document.getElementById("status").textContent,
};
}
);
is(results.return, "true", "installTrigger should have claimed success");
is(results.status, "0", "Callback should have seen a success");
gBrowser.removeCurrentTab();
Harness.finish();
};
|