Source code
Revision control
Copy as Markdown
Other Tools
/* Any copyright is dedicated to the Public Domain.
*/
"use strict";
const { AddonTestUtils } = ChromeUtils.importESModule(
);
AddonTestUtils.initMochitest(this);
const MY_CONTEXT = 2;
let gDidSeeChannel = false;
let promiseAddonStarted = null;
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"
);
}
function confirm_install(panel) {
is(panel.getAttribute("name"), "XPI Test", "Should have seen the name");
promiseAddonStarted = AddonTestUtils.promiseWebExtensionStartup(
"amosigned-xpi@tests.mozilla.org"
);
return true;
}
function install_ended(install, addon) {
AddonTestUtils.checkInstallInfo(install, {
method: "amWebAPI",
source: "test-host",
sourceURL: /https:\/\/example.com\/.*\/mozaddonmanager.html/,
});
promiseAddonStarted.then(() => addon.uninstall());
}
// Verifies that the mozAddonManager install flow download the xpi file from
// the expected userContextId.
add_task(async function test_userContext_mozAddonManager_install() {
await SpecialPowers.pushPrefEnv({
set: [
["extensions.webapi.testing", true],
[PREF_INSTALL_REQUIREBUILTINCERTS, false],
],
});
const deferredInstallCompleted = Promise.withResolvers();
Harness.installConfirmCallback = confirm_install;
Harness.installEndedCallback = install_ended;
Harness.installsCompletedCallback = deferredInstallCompleted.resolve;
Harness.finalContentEvent = "InstallComplete";
Harness.setup();
var triggers = encodeURIComponent(
JSON.stringify({
url: SECURE_TESTROOT + "amosigned.xpi",
})
);
gBrowser.selectedTab = BrowserTestUtils.addTab(gBrowser, "", {
userContextId: MY_CONTEXT,
});
Services.obs.addObserver(check_channel, "http-on-before-connect");
BrowserTestUtils.startLoadingURIString(
gBrowser,
SECURE_TESTROOT + "mozaddonmanager.html?" + triggers
);
info("Wait for the install to be completed");
const count = await deferredInstallCompleted.promise;
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");
const results = await SpecialPowers.spawn(
gBrowser.selectedBrowser,
[],
() => {
return {
return: content.document.getElementById("return").textContent,
status: content.document.getElementById("status").textContent,
};
}
);
is(results.return, "true", "mozAddonManager should have claimed success");
is(results.status, "STATE_INSTALLED", "Callback should have seen a success");
gBrowser.removeCurrentTab();
Harness.finish();
await SpecialPowers.popPrefEnv();
});