Source code
Revision control
Copy as Markdown
Other Tools
Test Info: Warnings
- This test gets skipped with pattern: os == 'android'
- Manifest: toolkit/mozapps/extensions/test/xpcshell/xpcshell.toml
/* Any copyright is dedicated to the Public Domain.
*/
// The test extension uses an insecure update url.
Services.prefs.setBoolPref(PREF_EM_CHECK_UPDATE_SECURITY, false);
createAppInfo("xpcshell@tests.mozilla.org", "XPCShell", "1", "1.9.2");
// Install one extension
// Start download of update check (but delay HTTP response)
// Cancel update check
// - ensure we get cancel notification
// complete HTTP response
// - ensure no callbacks after cancel
// - ensure update is gone
// Create an addon update listener containing a promise
// that resolves when the update is cancelled
function makeCancelListener() {
let resolve, reject;
let promise = new Promise((_resolve, _reject) => {
resolve = _resolve;
reject = _reject;
});
return {
onUpdateAvailable() {
reject("Should not have seen onUpdateAvailable notification");
},
onUpdateFinished(aAddon, aError) {
info("onUpdateCheckFinished: " + aAddon.id + " " + aError);
resolve(aError);
},
promise,
};
}
let testserver = createHttpServer({ hosts: ["example.com"] });
// Set up the HTTP server so that we can control when it responds
let _httpResolve;
function resetUpdateListener() {
return new Promise(resolve => {
_httpResolve = resolve;
});
}
testserver.registerPathHandler("/data/test_update.json", (req, resp) => {
resp.processAsync();
_httpResolve([req, resp]);
});
const UPDATE_RESPONSE = {
addons: {
"addon1@tests.mozilla.org": {
updates: [
{
version: "2.0",
applications: {
gecko: {
strict_min_version: "1",
strict_max_version: "1",
},
},
},
],
},
},
};
add_task(async function cancel_during_check() {
await promiseStartupManager();
await promiseInstallWebExtension({
manifest: {
name: "Test Addon 1",
version: "1.0",
browser_specific_settings: {
},
},
},
});
let a1 = await promiseAddonByID("addon1@tests.mozilla.org");
Assert.notEqual(a1, null);
let requestPromise = resetUpdateListener();
let listener = makeCancelListener();
a1.findUpdates(listener, AddonManager.UPDATE_WHEN_USER_REQUESTED);
// Wait for the http request to arrive
let [, /* request */ response] = await requestPromise;
// cancelUpdate returns true if there is an update check in progress
Assert.ok(a1.cancelUpdate());
let updateResult = await listener.promise;
Assert.equal(AddonManager.UPDATE_STATUS_CANCELLED, updateResult);
// Now complete the HTTP request
response.write(JSON.stringify(UPDATE_RESPONSE));
response.finish();
// trying to cancel again should return false, i.e. nothing to cancel
Assert.ok(!a1.cancelUpdate());
});
// Test that update check is cancelled if the XPI provider shuts down while
// the update check is in progress
add_task(async function shutdown_during_check() {
// Reset our HTTP listener
let requestPromise = resetUpdateListener();
let a1 = await promiseAddonByID("addon1@tests.mozilla.org");
Assert.notEqual(a1, null);
let listener = makeCancelListener();
a1.findUpdates(listener, AddonManager.UPDATE_WHEN_USER_REQUESTED);
// Wait for the http request to arrive
let [, /* request */ response] = await requestPromise;
await promiseShutdownManager();
let updateResult = await listener.promise;
Assert.equal(AddonManager.UPDATE_STATUS_CANCELLED, updateResult);
// Now complete the HTTP request
response.write(JSON.stringify(UPDATE_RESPONSE));
response.finish();
});