Source code
Revision control
Copy as Markdown
Other Tools
/* 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
const Cm = Components.manager;
const CONTRACT_ID = "@mozilla.org/content-permission/prompt;1";
var registrar = Cm.QueryInterface(Ci.nsIComponentRegistrar);
var oldClassID, oldFactory;
var newClassID = Services.uuid.generateUUID();
var newFactory = {
createInstance(aIID) {
return new MockPermissionPromptInstance().QueryInterface(aIID);
},
QueryInterface: ChromeUtils.generateQI(["nsIFactory"]),
};
export var MockPermissionPrompt = {
init() {
this.reset();
if (!registrar.isCIDRegistered(newClassID)) {
try {
oldClassID = registrar.contractIDToCID(CONTRACT_ID);
oldFactory = Cm.getClassObject(Cc[CONTRACT_ID], Ci.nsIFactory);
} catch (ex) {
oldClassID = "";
oldFactory = null;
dump(
"TEST-INFO | can't get permission prompt registered component, " +
"assuming there is none"
);
}
if (oldFactory) {
registrar.unregisterFactory(oldClassID, oldFactory);
}
registrar.registerFactory(newClassID, "", CONTRACT_ID, newFactory);
}
},
reset() {},
cleanup() {
this.reset();
if (oldFactory) {
registrar.unregisterFactory(newClassID, newFactory);
registrar.registerFactory(oldClassID, "", CONTRACT_ID, oldFactory);
}
},
};
function MockPermissionPromptInstance() {}
MockPermissionPromptInstance.prototype = {
QueryInterface: ChromeUtils.generateQI(["nsIContentPermissionPrompt"]),
promptResult: Ci.nsIPermissionManager.UNKNOWN_ACTION,
prompt(request) {
let perms = request.types.QueryInterface(Ci.nsIArray);
for (let idx = 0; idx < perms.length; idx++) {
let perm = perms.queryElementAt(idx, Ci.nsIContentPermissionType);
if (
Services.perms.testExactPermissionFromPrincipal(
request.principal,
perm.type
) != Ci.nsIPermissionManager.ALLOW_ACTION
) {
request.cancel();
return;
}
}
request.allow();
},
};