1 files changed, 88 insertions, 0 deletions
diff --git a/browser/components/preferences/tests/browser_searchShowSuggestionsFirst.js b/browser/components/preferences/tests/browser_searchShowSuggestionsFirst.js
new file mode 100644
index 0000000000..0826000abe
--- /dev/null
+++ b/browser/components/preferences/tests/browser_searchShowSuggestionsFirst.js
@@ -0,0 +1,88 @@
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/ */
+
+const PREF_NAME = "browser.urlbar.matchBuckets";
+const HISTORY_FIRST_PREF_VALUE = "general:5,suggestion:Infinity";
+const CHECKBOX_ID = "showSearchSuggestionsFirstCheckbox";
+
+// Open preferences with search suggestions shown first (the default).
+add_task(async function openWithSearchSuggestionsShownFirst() {
+  await SpecialPowers.pushPrefEnv({
+    set: [["browser.urlbar.suggest.searches", true]],
+  });
+
+  // The pref should be cleared initially so that search suggestions are shown
+  // first (the default).
+  Assert.equal(
+    Services.prefs.getCharPref(PREF_NAME, ""),
+    "",
+    "Pref should be cleared initially"
+  );
+
+  // Open preferences.  The checkbox should be checked.
+  await openPreferencesViaOpenPreferencesAPI("search", { leaveOpen: true });
+  let doc = gBrowser.selectedBrowser.contentDocument;
+  let checkbox = doc.getElementById(CHECKBOX_ID);
+  Assert.equal(checkbox.checked, true, "Checkbox should be checked");
+
+  // Uncheck the checkbox.
+  checkbox.checked = false;
+  checkbox.doCommand();
+
+  // The pref should now be set so that history is shown first.
+  Assert.equal(
+    Services.prefs.getCharPref(PREF_NAME, ""),
+    HISTORY_FIRST_PREF_VALUE,
+    "Pref should now be set to show history first"
+  );
+
+  // Clear the pref.
+  Services.prefs.clearUserPref(PREF_NAME);
+
+  // The checkbox should have become checked again.
+  Assert.equal(
+    checkbox.checked,
+    true,
+    "Checkbox should become checked after clearing pref"
+  );
+
+  // Clean up.
+  gBrowser.removeCurrentTab();
+});
+
+// Open preferences with history shown first.
+add_task(async function openWithHistoryShownFirst() {
+  // Set the pref to show history first.
+  Services.prefs.setCharPref(PREF_NAME, HISTORY_FIRST_PREF_VALUE);
+
+  // Open preferences.  The checkbox should be unchecked.
+  await openPreferencesViaOpenPreferencesAPI("search", { leaveOpen: true });
+  let doc = gBrowser.selectedBrowser.contentDocument;
+  let checkbox = doc.getElementById(CHECKBOX_ID);
+  Assert.equal(checkbox.checked, false, "Checkbox should be unchecked");
+
+  // Check the checkbox.
+  checkbox.checked = true;
+  checkbox.doCommand();
+
+  // The pref should now be cleared so that search suggestions are shown first.
+  Assert.equal(
+    Services.prefs.getCharPref(PREF_NAME, ""),
+    "",
+    "Pref should now be cleared to show search suggestions first"
+  );
+
+  // Set the pref again.
+  Services.prefs.setCharPref(PREF_NAME, HISTORY_FIRST_PREF_VALUE);
+
+  // The checkbox should have become unchecked again.
+  Assert.equal(
+    checkbox.checked,
+    false,
+    "Checkbox should become unchecked after setting pref"
+  );
+
+  // Clean up.
+  gBrowser.removeCurrentTab();
+  Services.prefs.clearUserPref(PREF_NAME);
+});