/* -*- Mode: indent-tabs-mode: nil; js-indent-level: 2 -*- */
/* vim: set sts=2 sw=2 et tw=80: */
"use strict";

function confirmDefaults() {
  if (gURLBar.searchButton) {
    is(
      getComputedStyle(document.getElementById("identity-box")).display,
      "none",
      "Identity box should be hidden"
    );
  } else {
    is(
      getComputedStyle(document.getElementById("identity-icon")).listStyleImage,
      'url("chrome://global/skin/icons/search-glass.svg")',
      "Identity icon should be the search icon"
    );
  }

  let label = document.getElementById("identity-icon-label");
  ok(
    BrowserTestUtils.is_hidden(label),
    "No label should be used before the extension is started"
  );
}

async function waitForIndentityBoxMutation({ expectExtensionIcon }) {
  const el = document.getElementById("identity-box");
  await BrowserTestUtils.waitForMutationCondition(
    el,
    {
      attributeFilter: ["class"],
    },
    () => el.classList.contains("extensionPage") == expectExtensionIcon
  );
}

function confirmExtensionPage() {
  let identityIconEl = document.getElementById("identity-icon");

  is(
    getComputedStyle(identityIconEl).listStyleImage,
    'url("chrome://mozapps/skin/extensions/extension.svg")',
    "Identity icon should be the default extension icon"
  );

  is(
    identityIconEl.tooltipText,
    "Loaded by extension: Test Extension",
    "The correct tooltip should be used"
  );

  let label = document.getElementById("identity-icon-label");
  is(
    label.value,
    "Extension (Test Extension)",
    "The correct label should be used"
  );
  ok(BrowserTestUtils.is_visible(label), "No label should be visible");
}

add_task(async function testIdentityIndication() {
  let extension = ExtensionTestUtils.loadExtension({
    background() {
      browser.test.sendMessage("url", browser.runtime.getURL("icon.png"));
    },
    manifest: {
      name: "Test Extension",
    },
    files: {
      "icon.png": "",
    },
  });

  await extension.startup();

  confirmDefaults();

  let url = await extension.awaitMessage("url");

  const promiseIdentityBoxExtension = waitForIndentityBoxMutation({
    expectExtensionIcon: true,
  });
  await BrowserTestUtils.withNewTab({ gBrowser, url }, async function () {
    await promiseIdentityBoxExtension;
    confirmExtensionPage();
  });

  const promiseIdentityBoxDefault = waitForIndentityBoxMutation({
    expectExtensionIcon: false,
  });
  await extension.unload();
  await promiseIdentityBoxDefault;

  confirmDefaults();
});

add_task(async function testIdentityIndicationNewTab() {
  let extension = ExtensionTestUtils.loadExtension({
    background() {
      browser.test.sendMessage("url", browser.runtime.getURL("newtab.html"));
    },
    manifest: {
      name: "Test Extension",
      browser_specific_settings: {
        gecko: {
          id: "@newtab",
        },
      },
      chrome_url_overrides: {
        newtab: "newtab.html",
      },
    },
    files: {
      "newtab.html": "<h1>New tab!</h1>",
    },
    useAddonManager: "temporary",
  });

  await extension.startup();

  confirmDefaults();

  let url = await extension.awaitMessage("url");
  const promiseIdentityBoxExtension = waitForIndentityBoxMutation({
    expectExtensionIcon: true,
  });
  await BrowserTestUtils.withNewTab({ gBrowser, url }, async function () {
    await promiseIdentityBoxExtension;
    confirmExtensionPage();
    is(gURLBar.value, "", "The URL bar is blank");
  });

  const promiseIdentityBoxDefault = waitForIndentityBoxMutation({
    expectExtensionIcon: false,
  });
  await extension.unload();
  await promiseIdentityBoxDefault;

  confirmDefaults();
});