gecko/toolkit/mozapps/extensions/test/xpcshell/test_bug384052.js

123 lines
3.9 KiB
JavaScript

const Cc = Components.classes;
const Ci = Components.interfaces;
const CLASS_ID = Components.ID("{12345678-1234-1234-1234-123456789abc}");
const CONTRACT_ID = "@mozilla.org/test-parameter-source;1";
var gTestURL = "http://127.0.0.1:4444/?itemID=%ITEM_ID%&custom1=%CUSTOM1%&custom2=%CUSTOM2%";
var gExpectedURL = null;
var gSeenExpectedURL = false;
var gComponentRegistrar = Components.manager.QueryInterface(Ci.nsIComponentRegistrar);
var gObserverService = Cc["@mozilla.org/observer-service;1"].getService(Ci.nsIObserverService);
var gCategoryManager = Cc["@mozilla.org/categorymanager;1"].getService(Ci.nsICategoryManager);
// Factory for our parameter handler
var paramHandlerFactory = {
QueryInterface: function(iid) {
if (iid.equals(Ci.nsIFactory) || iid.equals(Ci.nsISupports))
return this;
throw Components.results.NS_ERROR_NO_INTERFACE;
},
createInstance: function(outer, iid) {
var bag = Cc["@mozilla.org/hash-property-bag;1"]
.createInstance(Ci.nsIWritablePropertyBag);
bag.setProperty("CUSTOM1", "custom_parameter_1");
bag.setProperty("CUSTOM2", "custom_parameter_2");
return bag.QueryInterface(iid);
}
};
// Observer that should recognize when extension manager requests our URL
var observer = {
QueryInterface: function(iid) {
if (iid.equals(Ci.nsIObserver) || iid.equals(Ci.nsISupports))
return this;
throw Components.results.NS_ERROR_NO_INTERFACE;
},
observe: function(subject, topic, data) {
if (topic == "http-on-modify-request") {
var channel = subject.QueryInterface(Ci.nsIChannel);
if (channel.originalURI.spec == gExpectedURL)
gSeenExpectedURL = true;
}
}
}
function initTest()
{
do_test_pending();
// Setup extension manager
createAppInfo("xpcshell@tests.mozilla.org", "XPCShell", "1", "1.9");
// Register our parameter handlers
gComponentRegistrar.registerFactory(CLASS_ID, "Test component", CONTRACT_ID, paramHandlerFactory);
gCategoryManager.addCategoryEntry("extension-update-params", "CUSTOM1", CONTRACT_ID, false, false);
gCategoryManager.addCategoryEntry("extension-update-params", "CUSTOM2", CONTRACT_ID, false, false);
// Register observer to get notified when URLs are requested
gObserverService.addObserver(observer, "http-on-modify-request", false);
// Install a test extension into the profile
let dir = gProfD.clone();
dir.append("extensions");
writeInstallRDFForExtension({
id: "test@mozilla.org",
version: "1.0",
name: "Test extension",
updateURL: gTestURL,
targetApplications: [{
id: "xpcshell@tests.mozilla.org",
minVersion: "1",
maxVersion: "1"
}],
}, dir);
startupManager();
}
function shutdownTest()
{
shutdownManager();
gComponentRegistrar.unregisterFactory(CLASS_ID, paramHandlerFactory);
gCategoryManager.deleteCategoryEntry("extension-update-params", "CUSTOM1", false);
gCategoryManager.deleteCategoryEntry("extension-update-params", "CUSTOM2", false);
gObserverService.removeObserver(observer, "http-on-modify-request");
do_test_finished();
}
function run_test()
{
initTest();
gExpectedURL = gTestURL.replace(/%ITEM_ID%/, "test@mozilla.org")
.replace(/%CUSTOM1%/, "custom_parameter_1")
.replace(/%CUSTOM2%/, "custom_parameter_2");
AddonManager.getAddonByID("test@mozilla.org", function(item) {
// Initiate update
item.findUpdates({
onCompatibilityUpdateAvailable: function(addon) {
do_throw("Should not have seen a compatibility update");
},
onUpdateAvailable: function(addon, install) {
do_throw("Should not have seen an available update");
},
onUpdateFinished: function(addon, error) {
do_check_eq(error, AddonManager.UPDATE_STATUS_DOWNLOAD_ERROR);
do_check_true(gSeenExpectedURL);
shutdownTest();
}
}, AddonManager.UPDATE_WHEN_USER_REQUESTED);
});
}