1 files changed, 163 insertions, 0 deletions
diff --git a/uriloader/exthandler/tests/unit/test_defaults_handlerService.js b/uriloader/exthandler/tests/unit/test_defaults_handlerService.js
new file mode 100644
index 0000000000..f9f9feda23
--- /dev/null
+++ b/uriloader/exthandler/tests/unit/test_defaults_handlerService.js
@@ -0,0 +1,163 @@
+/* Any copyright is dedicated to the Public Domain.
+http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+XPCOMUtils.defineLazyServiceGetter(
+  this,
+  "gExternalProtocolService",
+  "@mozilla.org/uriloader/external-protocol-service;1",
+  "nsIExternalProtocolService"
+);
+
+const kDefaultHandlerList = Services.prefs
+  .getChildList("gecko.handlerService.schemes")
+  .filter(p => {
+    try {
+      let val = Services.prefs.getComplexValue(p, Ci.nsIPrefLocalizedString)
+        .data;
+      return !!val;
+    } catch (ex) {
+      return false;
+    }
+  });
+
+add_task(async function test_check_defaults_get_added() {
+  let protocols = new Set(
+    kDefaultHandlerList.map(p => p.match(/schemes\.(\w+)/)[1])
+  );
+  for (let protocol of protocols) {
+    const kPrefStr = `schemes.${protocol}.`;
+    let matchingPrefs = kDefaultHandlerList.filter(p => p.includes(kPrefStr));
+    let protocolHandlerCount = matchingPrefs.length / 2;
+    Assert.ok(
+      protocolHandlerCount,
+      `Prefs for ${protocol} have at least 1 protocol handler`
+    );
+    Assert.ok(
+      gHandlerService.wrappedJSObject._store.data.schemes[protocol].stubEntry,
+      `Expect stub for ${protocol}`
+    );
+    let info = gExternalProtocolService.getProtocolHandlerInfo(protocol, {});
+    Assert.ok(
+      info,
+      `Should be able to get protocol handler info for ${protocol}`
+    );
+    let handlers = Array.from(
+      info.possibleApplicationHandlers.enumerate(Ci.nsIHandlerApp)
+    );
+    handlers = handlers.filter(h => h instanceof Ci.nsIWebHandlerApp);
+    Assert.equal(
+      handlers.length,
+      protocolHandlerCount,
+      `Default web handlers for ${protocol} should match`
+    );
+    let { alwaysAskBeforeHandling, preferredAction } = info;
+    // Actually store something, pretending there was a change:
+    let infoToWrite = gExternalProtocolService.getProtocolHandlerInfo(
+      protocol,
+      {}
+    );
+    gHandlerService.store(infoToWrite);
+    ok(
+      !gHandlerService.wrappedJSObject._store.data.schemes[protocol].stubEntry,
+      "Expect stub entry info to go away"
+    );
+
+    let newInfo = gExternalProtocolService.getProtocolHandlerInfo(protocol, {});
+    Assert.equal(
+      alwaysAskBeforeHandling,
+      newInfo.alwaysAskBeforeHandling,
+      protocol + " - always ask shouldn't change"
+    );
+    Assert.equal(
+      preferredAction,
+      newInfo.preferredAction,
+      protocol + " - preferred action shouldn't change"
+    );
+    await deleteHandlerStore();
+  }
+});
+
+add_task(async function test_check_default_modification() {
+  let mailtoHandlerCount =
+    kDefaultHandlerList.filter(p => p.includes("mailto")).length / 2;
+  Assert.ok(mailtoHandlerCount, "Prefs have at least 1 mailto handler");
+  Assert.ok(
+    true,
+    JSON.stringify(gHandlerService.wrappedJSObject._store.data.schemes.mailto)
+  );
+  Assert.ok(
+    gHandlerService.wrappedJSObject._store.data.schemes.mailto.stubEntry,
+    "Expect stub for mailto"
+  );
+  let mailInfo = gExternalProtocolService.getProtocolHandlerInfo("mailto", {});
+  mailInfo.alwaysAskBeforeHandling = false;
+  mailInfo.preferredAction = Ci.nsIHandlerInfo.useSystemDefault;
+  gHandlerService.store(mailInfo);
+  Assert.ok(
+    !gHandlerService.wrappedJSObject._store.data.schemes.mailto.stubEntry,
+    "Stub entry should be removed immediately."
+  );
+  let newMail = gExternalProtocolService.getProtocolHandlerInfo("mailto", {});
+  Assert.equal(newMail.preferredAction, Ci.nsIHandlerInfo.useSystemDefault);
+  Assert.equal(newMail.alwaysAskBeforeHandling, false);
+  await deleteHandlerStore();
+});
+
+/**
+ * Check that we don't add bogus handlers.
+ */
+add_task(async function test_check_restrictions() {
+  const kTestData = {
+    testdeleteme: [
+      ["Delete me", ""],
+      ["Delete me insecure", "http://example.com/%s"],
+      ["Delete me no substitution", "https://example.com/"],
+      ["Keep me", "https://example.com/%s"],
+    ],
+    testreallydeleteme: [
+      // used to check we remove the entire entry.
+      ["Delete me", "http://example.com/%s"],
+    ],
+  };
+  for (let [scheme, handlers] of Object.entries(kTestData)) {
+    let count = 1;
+    for (let [name, uriTemplate] of handlers) {
+      let pref = `gecko.handlerService.schemes.${scheme}.${count}.`;
+      let obj = Cc["@mozilla.org/pref-localizedstring;1"].createInstance(
+        Ci.nsIPrefLocalizedString
+      );
+      obj.data = name;
+      Services.prefs.setComplexValue(
+        pref + "name",
+        Ci.nsIPrefLocalizedString,
+        obj
+      );
+      obj.data = uriTemplate;
+      Services.prefs.setComplexValue(
+        pref + "uriTemplate",
+        Ci.nsIPrefLocalizedString,
+        obj
+      );
+      count++;
+    }
+  }
+
+  gHandlerService.wrappedJSObject._injectDefaultProtocolHandlers();
+  let schemeData = gHandlerService.wrappedJSObject._store.data.schemes;
+
+  Assert.ok(schemeData.testdeleteme, "Expect an entry for testdeleteme");
+  Assert.ok(
+    schemeData.testdeleteme.stubEntry,
+    "Expect a stub entry for testdeleteme"
+  );
+
+  Assert.deepEqual(
+    schemeData.testdeleteme.handlers,
+    [null, { name: "Keep me", uriTemplate: "https://example.com/%s" }],
+    "Expect only one handler is kept."
+  );
+
+  Assert.ok(!schemeData.testreallydeleteme, "No entry for reallydeleteme");
+});