1 files changed, 134 insertions, 0 deletions
diff --git a/browser/components/extensions/test/browser/browser_ext_runtime_setUninstallURL.js b/browser/components/extensions/test/browser/browser_ext_runtime_setUninstallURL.js
new file mode 100644
index 0000000000..2530c28a6d
--- /dev/null
+++ b/browser/components/extensions/test/browser/browser_ext_runtime_setUninstallURL.js
@@ -0,0 +1,134 @@
+"use strict";
+
+// testing/mochitest/tests/SimpleTest/ExtensionTestUtils.js loads
+// ExtensionTestCommon, and is slated as part of the SimpleTest
+// environment in tools/lint/eslint/eslint-plugin-mozilla/lib/environments/simpletest.js
+// However, nothing but the ExtensionTestUtils global gets put
+// into the scope, and so although eslint thinks this global is
+// available, it really isn't.
+// eslint-disable-next-line mozilla/no-redeclare-with-import-autofix
+let { ExtensionTestCommon } = ChromeUtils.importESModule(
+  "resource://testing-common/ExtensionTestCommon.sys.mjs"
+);
+
+async function makeAndInstallXPI(id, backgroundScript, loadedURL) {
+  let xpi = ExtensionTestCommon.generateXPI({
+    manifest: { browser_specific_settings: { gecko: { id } } },
+    background: backgroundScript,
+  });
+  SimpleTest.registerCleanupFunction(function cleanupXPI() {
+    Services.obs.notifyObservers(xpi, "flush-cache-entry");
+    xpi.remove(false);
+  });
+
+  let loadPromise = BrowserTestUtils.waitForNewTab(gBrowser, loadedURL);
+
+  info(`installing ${xpi.path}`);
+  let addon = await AddonManager.installTemporaryAddon(xpi);
+  info("installed");
+
+  // A WebExtension is started asynchronously, we have our test extension
+  // open a new tab to signal that the background script has executed.
+  let loadTab = await loadPromise;
+  BrowserTestUtils.removeTab(loadTab);
+
+  return addon;
+}
+
+add_task(async function test_setuninstallurl_badargs() {
+  async function background() {
+    await browser.test.assertRejects(
+      browser.runtime.setUninstallURL("this is not a url"),
+      /Invalid URL/,
+      "setUninstallURL with an invalid URL should fail"
+    );
+
+    await browser.test.assertRejects(
+      browser.runtime.setUninstallURL("file:///etc/passwd"),
+      /must have the scheme http or https/,
+      "setUninstallURL with an illegal URL should fail"
+    );
+
+    browser.test.notifyPass("setUninstallURL bad params");
+  }
+
+  let extension = ExtensionTestUtils.loadExtension({
+    background,
+  });
+  await extension.startup();
+  await extension.awaitFinish();
+  await extension.unload();
+});
+
+// Test the documented behavior of setUninstallURL() that passing an
+// empty string is equivalent to not setting an uninstall URL
+// (i.e., no new tab is opened upon uninstall)
+add_task(async function test_setuninstall_empty_url() {
+  async function backgroundScript() {
+    await browser.runtime.setUninstallURL("");
+    browser.tabs.create({ url: "http://example.com/addon_loaded" });
+  }
+
+  let addon = await makeAndInstallXPI(
+    "test_uinstallurl2@tests.mozilla.org",
+    backgroundScript,
+    "http://example.com/addon_loaded"
+  );
+
+  addon.uninstall(true);
+  info("uninstalled");
+
+  // no need to explicitly check for the absence of a new tab,
+  // BrowserTestUtils will eventually complain if one is opened.
+});
+
+// Test the documented behavior of setUninstallURL() that passing an
+// empty string is equivalent to not setting an uninstall URL
+// (i.e., no new tab is opened upon uninstall)
+// here we pass a null value to string and test
+add_task(async function test_setuninstall_null_url() {
+  async function backgroundScript() {
+    await browser.runtime.setUninstallURL(null);
+    browser.tabs.create({ url: "http://example.com/addon_loaded" });
+  }
+
+  let addon = await makeAndInstallXPI(
+    "test_uinstallurl2@tests.mozilla.org",
+    backgroundScript,
+    "http://example.com/addon_loaded"
+  );
+
+  addon.uninstall(true);
+  info("uninstalled");
+
+  // no need to explicitly check for the absence of a new tab,
+  // BrowserTestUtils will eventually complain if one is opened.
+});
+
+add_task(async function test_setuninstallurl() {
+  async function backgroundScript() {
+    await browser.runtime.setUninstallURL(
+      "http://example.com/addon_uninstalled"
+    );
+    browser.tabs.create({ url: "http://example.com/addon_loaded" });
+  }
+
+  let addon = await makeAndInstallXPI(
+    "test_uinstallurl@tests.mozilla.org",
+    backgroundScript,
+    "http://example.com/addon_loaded"
+  );
+
+  // look for a new tab with the uninstall url.
+  let uninstallPromise = BrowserTestUtils.waitForNewTab(
+    gBrowser,
+    "http://example.com/addon_uninstalled"
+  );
+
+  addon.uninstall(true);
+  info("uninstalled");
+
+  let uninstalledTab = await uninstallPromise;
+  isnot(uninstalledTab, null, "opened tab with uninstall url");
+  BrowserTestUtils.removeTab(uninstalledTab);
+});