77 lines
2.8 KiB
JavaScript
77 lines
2.8 KiB
JavaScript
/* Any copyright is dedicated to the Public Domain.
|
|
http://creativecommons.org/publicdomain/zero/1.0/ */
|
|
|
|
"use strict";
|
|
|
|
add_task(async function test_helperapp() {
|
|
// Set up the test infrastructure:
|
|
const kProt = "foopydoopydoo";
|
|
const extProtocolSvc = Cc[
|
|
"@mozilla.org/uriloader/external-protocol-service;1"
|
|
].getService(Ci.nsIExternalProtocolService);
|
|
const handlerSvc = Cc["@mozilla.org/uriloader/handler-service;1"].getService(
|
|
Ci.nsIHandlerService
|
|
);
|
|
let handlerInfo = extProtocolSvc.getProtocolHandlerInfo(kProt);
|
|
if (handlerSvc.exists(handlerInfo)) {
|
|
handlerSvc.fillHandlerInfo(handlerInfo, "");
|
|
}
|
|
// Say we want to use a specific app:
|
|
handlerInfo.preferredAction = Ci.nsIHandlerInfo.useHelperApp;
|
|
handlerInfo.alwaysAskBeforeHandling = false;
|
|
|
|
// Say it's us:
|
|
let selfFile = Services.dirsvc.get("XREExeF", Ci.nsIFile);
|
|
// Make sure it's the .app
|
|
if (AppConstants.platform == "macosx") {
|
|
while (
|
|
!selfFile.leafName.endsWith(".app") &&
|
|
!selfFile.leafName.endsWith(".app/")
|
|
) {
|
|
selfFile = selfFile.parent;
|
|
}
|
|
}
|
|
let selfHandlerApp = Cc[
|
|
"@mozilla.org/uriloader/local-handler-app;1"
|
|
].createInstance(Ci.nsILocalHandlerApp);
|
|
selfHandlerApp.executable = selfFile;
|
|
handlerInfo.possibleApplicationHandlers.appendElement(selfHandlerApp);
|
|
handlerInfo.preferredApplicationHandler = selfHandlerApp;
|
|
handlerSvc.store(handlerInfo);
|
|
|
|
await BrowserTestUtils.withNewTab("about:blank", async browser => {
|
|
// Now, do some safety stubbing. If we do end up recursing we spawn
|
|
// infinite tabs. We definitely don't want that. Avoid it by stubbing
|
|
// our external URL handling bits:
|
|
let oldAddTab = gBrowser.addTab;
|
|
registerCleanupFunction(() => (gBrowser.addTab = oldAddTab));
|
|
let wrongThingHappenedPromise = new Promise(resolve => {
|
|
gBrowser.addTab = function (aURI) {
|
|
ok(false, "Tried to open unexpected URL in a tab: " + aURI);
|
|
resolve(null);
|
|
// Pass a dummy object to avoid upsetting BrowserContentHandler -
|
|
// if it thinks opening the tab failed, it tries to open a window instead,
|
|
// which we can't prevent as easily, and at which point we still end up
|
|
// with runaway tabs.
|
|
return {};
|
|
};
|
|
});
|
|
|
|
let askedUserPromise = waitForProtocolAppChooserDialog(browser, true);
|
|
|
|
gBrowser.fixupAndLoadURIString(kProt + ":test", {
|
|
triggeringPrincipal: Services.scriptSecurityManager.getSystemPrincipal(),
|
|
loadFlags: Ci.nsIWebNavigation.LOAD_FLAGS_FROM_EXTERNAL,
|
|
});
|
|
let dialog = await Promise.race([
|
|
wrongThingHappenedPromise,
|
|
askedUserPromise,
|
|
]);
|
|
ok(dialog, "Should have gotten a dialog");
|
|
|
|
let closePromise = waitForProtocolAppChooserDialog(browser, false);
|
|
dialog.close();
|
|
await closePromise;
|
|
askedUserPromise = null;
|
|
});
|
|
});
|