"use strict";

let { AddonManager } = ChromeUtils.import(
  "resource://gre/modules/AddonManager.jsm"
);
let { ExtensionTestCommon } = ChromeUtils.import(
  "resource://testing-common/ExtensionTestCommon.jsm"
);

async function makeAndInstallXPI(id, backgroundScript, loadedURL) {
  let xpi = ExtensionTestCommon.generateXPI({
    manifest: { applications: { 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);
});
