diff options
author | Matt A. Tobin <email@mattatobin.com> | 2018-02-10 02:51:36 -0500 |
---|---|---|
committer | Matt A. Tobin <email@mattatobin.com> | 2018-02-10 02:51:36 -0500 |
commit | 37d5300335d81cecbecc99812747a657588c63eb (patch) | |
tree | 765efa3b6a56bb715d9813a8697473e120436278 /toolkit/mozapps/webextensions/test/AddonManagerTesting.jsm | |
parent | b2bdac20c02b12f2057b9ef70b0a946113a00e00 (diff) | |
parent | 4fb11cd5966461bccc3ed1599b808237be6b0de9 (diff) | |
download | UXP-37d5300335d81cecbecc99812747a657588c63eb.tar UXP-37d5300335d81cecbecc99812747a657588c63eb.tar.gz UXP-37d5300335d81cecbecc99812747a657588c63eb.tar.lz UXP-37d5300335d81cecbecc99812747a657588c63eb.tar.xz UXP-37d5300335d81cecbecc99812747a657588c63eb.zip |
Merge branch 'ext-work'
Diffstat (limited to 'toolkit/mozapps/webextensions/test/AddonManagerTesting.jsm')
-rw-r--r-- | toolkit/mozapps/webextensions/test/AddonManagerTesting.jsm | 115 |
1 files changed, 115 insertions, 0 deletions
diff --git a/toolkit/mozapps/webextensions/test/AddonManagerTesting.jsm b/toolkit/mozapps/webextensions/test/AddonManagerTesting.jsm new file mode 100644 index 000000000..eb9a92195 --- /dev/null +++ b/toolkit/mozapps/webextensions/test/AddonManagerTesting.jsm @@ -0,0 +1,115 @@ +/* This Source Code Form is subject to the terms of the Mozilla Public + * License, v. 2.0. If a copy of the MPL was not distributed with this + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ + +// This file is a test-only JSM containing utility methods for +// interacting with the add-ons manager. + +"use strict"; + +this.EXPORTED_SYMBOLS = [ + "AddonManagerTesting", +]; + +const {utils: Cu} = Components; + +Cu.import("resource://gre/modules/Promise.jsm"); +Cu.import("resource://gre/modules/XPCOMUtils.jsm"); + +XPCOMUtils.defineLazyModuleGetter(this, "AddonManager", + "resource://gre/modules/AddonManager.jsm"); + +this.AddonManagerTesting = { + /** + * Get the add-on that is specified by its ID. + * + * @return {Promise<Object>} A promise that resolves returning the found addon or null + * if it is not found. + */ + getAddonById: function (id) { + return new Promise(resolve => AddonManager.getAddonByID(id, addon => resolve(addon))); + }, + + /** + * Uninstall an add-on that is specified by its ID. + * + * The returned promise resolves on successful uninstall and rejects + * if the add-on is not unknown. + * + * @return Promise<restartRequired> + */ + uninstallAddonByID: function (id) { + let deferred = Promise.defer(); + + AddonManager.getAddonByID(id, (addon) => { + if (!addon) { + deferred.reject(new Error("Add-on is not known: " + id)); + return; + } + + let listener = { + onUninstalling: function (addon, needsRestart) { + if (addon.id != id) { + return; + } + + if (needsRestart) { + AddonManager.removeAddonListener(listener); + deferred.resolve(true); + } + }, + + onUninstalled: function (addon) { + if (addon.id != id) { + return; + } + + AddonManager.removeAddonListener(listener); + deferred.resolve(false); + }, + + onOperationCancelled: function (addon) { + if (addon.id != id) { + return; + } + + AddonManager.removeAddonListener(listener); + deferred.reject(new Error("Uninstall cancelled.")); + }, + }; + + AddonManager.addAddonListener(listener); + addon.uninstall(); + }); + + return deferred.promise; + }, + + /** + * Install an XPI add-on from a URL. + * + * @return Promise<addon> + */ + installXPIFromURL: function (url, hash, name, iconURL, version) { + let deferred = Promise.defer(); + + AddonManager.getInstallForURL(url, (install) => { + let fail = () => { deferred.reject(new Error("Add-on install failed.")) }; + + let listener = { + onDownloadCancelled: fail, + onDownloadFailed: fail, + onInstallCancelled: fail, + onInstallFailed: fail, + onInstallEnded: function (install, addon) { + deferred.resolve(addon); + }, + }; + + install.addListener(listener); + install.install(); + }, "application/x-xpinstall", hash, name, iconURL, version); + + return deferred.promise; + }, +}; |