Mypal68/devtools/client/aboutdebugging/test/browser_addons_install.js

/* Any copyright is dedicated to the Public Domain.
   http://creativecommons.org/publicdomain/zero/1.0/ */
"use strict";

loader.lazyImporter(
  this,
  "AddonTestUtils",
  "resource://testing-common/AddonTestUtils.jsm"
);

AddonTestUtils.initMochitest(this);

function mockFilePicker(window, file) {
  // Mock the file picker to select a test addon
  const MockFilePicker = SpecialPowers.MockFilePicker;
  MockFilePicker.init(window);
  MockFilePicker.setFiles([file]);
}

/**
 * Write out an extension with a manifest.json to dir.
 *
 * @param {object} manfiest
 *        The manifest file as an object.
 * @param {nsIFile} dir
 *        An nsIFile for representing the output folder.
 * @return {Promise} Promise that resolves to the output folder when done.
 */
function promiseWriteWebManifestForExtension(manifest, dir) {
  const files = {
    "manifest.json": JSON.stringify(manifest),
  };
  return AddonTestUtils.promiseWriteFilesToExtension(
    dir.path,
    manifest.applications.gecko.id,
    files,
    true
  );
}

add_task(async function testWebextensionInstallSuccess() {
  const { tab, document } = await openAboutDebugging("addons");
  await waitForInitialAddonList(document);

  // Install this add-on, and verify that it appears in the about:debugging UI
  await installAddon({
    document,
    path: "addons/unpacked/manifest.json",
    name: "test-devtools",
  });

  // Install the add-on, and verify that it disappears in the about:debugging UI
  await uninstallAddon({
    document,
    id: "test-devtools@mozilla.org",
    name: "test-devtools",
  });

  await closeAboutDebugging(tab);
});

add_task(async function testWebextensionInstallError() {
  const { tab, document, window } = await openAboutDebugging("addons");
  await waitForInitialAddonList(document);

  // Trigger the file picker by clicking on the button
  mockFilePicker(window, getSupportsFile("addons/bad/manifest.json").file);
  document.getElementById("load-addon-from-file").click();

  info("wait for the install error to appear");
  const top = document.querySelector(".addons-top");
  await waitUntilElement(".addons-install-error", top);

  await closeAboutDebugging(tab);
});

add_task(async function testWebextensionInstallErrorRetry() {
  const { tab, document, window } = await openAboutDebugging("addons");
  await waitForInitialAddonList(document);

  const tempdir = AddonTestUtils.tempDir.clone();
  const addonId = "invalid-addon-install-retry@mozilla.org";
  const addonName = "invalid-addon-install-retry";
  const manifest = {
    name: addonName,
    description: "test invalid-addon-install-retry",
    // eslint-disable-next-line camelcase
    manifest_version: 2,
    version: "1.0",
    applications: { gecko: { id: addonId } },
    // These should all be wrapped in arrays.
    // eslint-disable-next-line camelcase
    content_scripts: { matches: "http://*/", js: "foo.js" },
  };

  await promiseWriteWebManifestForExtension(manifest, tempdir);

  // Mock the file picker to select a test addon.
  const manifestFile = tempdir.clone();
  manifestFile.append(addonId, "manifest.json");
  mockFilePicker(window, manifestFile);

  // Trigger the file picker by clicking on the button.
  document.getElementById("load-addon-from-file").click();

  info("wait for the install error to appear");
  const top = document.querySelector(".addons-top");
  await waitUntilElement(".addons-install-error", top);

  const retryButton = document.querySelector("button.addons-install-retry");
  is(retryButton.textContent, "Retry", "Retry button has a good label");

  // Fix the manifest so the add-on will install.
  // eslint-disable-next-line camelcase
  manifest.content_scripts = [
    {
      matches: ["http://*/"],
      js: ["foo.js"],
    },
  ];
  await promiseWriteWebManifestForExtension(manifest, tempdir);

  const addonEl = document.querySelector(`[data-addon-id="${addonId}"]`);
  // Verify this add-on isn't installed yet.
  ok(!addonEl, "Addon is not installed yet");

  // Retry the install.
  retryButton.click();

  info("Wait for the add-on to be shown");
  await waitUntilElement(`[data-addon-id="${addonId}"]`, document);
  info("Addon is installed");

  // Install the add-on, and verify that it disappears in the about:debugging UI
  await uninstallAddon({ document, id: addonId, name: addonName });

  await closeAboutDebugging(tab);
});