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

/**
 * Tests the Unified Search Button with an empty search input.
 */

"use strict";

add_setup(async function setup() {
  await SpecialPowers.pushPrefEnv({
    set: [["browser.urlbar.scotchBonnet.enableOverride", true]],
  });
});

// Tests that search mode chiclet remains when input is empty and
// urlbar is unfocused.
add_task(async function test_search_mode_chiclet_unfocus_home_page() {
  let popup = await UrlbarTestUtils.openSearchModeSwitcher(window);
  let popupHidden = UrlbarTestUtils.searchModeSwitcherPopupClosed(window);
  popup.querySelector(".search-button-bookmarks").button.click();
  await popupHidden;

  await UrlbarTestUtils.assertSearchMode(window, {
    source: UrlbarUtils.RESULT_SOURCE.BOOKMARKS,
    entry: "searchbutton",
  });

  // Shift focus away from urlbar, similar to clicking on page content
  // area.
  EventUtils.synthesizeKey("KEY_Tab");

  await UrlbarTestUtils.assertSearchMode(window, {
    source: UrlbarUtils.RESULT_SOURCE.BOOKMARKS,
    entry: "searchbutton",
  });
});

// Tests that search mode chiclet remains on loaded sites as well.
add_task(async function test_search_mode_chiclet_unfocus_loaded_sites() {
  let newTab = await BrowserTestUtils.openNewForegroundTab(
    window.gBrowser,
    "https://example.com"
  );

  // Make unified search button appear
  await UrlbarTestUtils.promiseAutocompleteResultPopup({
    window,
    value: "",
  });

  let popup = await UrlbarTestUtils.openSearchModeSwitcher(window);
  let popupHidden = UrlbarTestUtils.searchModeSwitcherPopupClosed(window);
  popup.querySelector(".search-button-bookmarks").button.click();
  await popupHidden;

  await UrlbarTestUtils.assertSearchMode(window, {
    source: UrlbarUtils.RESULT_SOURCE.BOOKMARKS,
    entry: "searchbutton",
  });

  // Shift focus away from urlbar, similar to clicking on page content
  // area.
  EventUtils.synthesizeKey("KEY_Tab");

  await UrlbarTestUtils.assertSearchMode(window, {
    source: UrlbarUtils.RESULT_SOURCE.BOOKMARKS,
    entry: "searchbutton",
  });

  BrowserTestUtils.removeTab(newTab);
});