1 files changed, 99 insertions, 0 deletions
diff --git a/browser/components/search/test/browser_hiddenOneOffs_cleanup.js b/browser/components/search/test/browser_hiddenOneOffs_cleanup.js
new file mode 100644
index 000000000..9a584feb6
--- /dev/null
+++ b/browser/components/search/test/browser_hiddenOneOffs_cleanup.js
@@ -0,0 +1,99 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+const testPref = "Foo,FooDupe";
+
+function promiseNewEngine(basename) {
+  return new Promise((resolve, reject) => {
+    info("Waiting for engine to be added: " + basename);
+    Services.search.init({
+      onInitComplete: function() {
+        let url = getRootDirectory(gTestPath) + basename;
+        Services.search.addEngine(url, null, "", false, {
+          onSuccess: function (engine) {
+            info("Search engine added: " + basename);
+            resolve(engine);
+          },
+          onError: function (errCode) {
+            ok(false, "addEngine failed with error code " + errCode);
+            reject();
+          }
+        });
+      }
+    });
+  });
+}
+
+add_task(function* test_remove() {
+  yield promiseNewEngine("testEngine_dupe.xml");
+  yield promiseNewEngine("testEngine.xml");
+  Services.prefs.setCharPref("browser.search.hiddenOneOffs", testPref);
+
+  info("Removing testEngine_dupe.xml");
+  Services.search.removeEngine(Services.search.getEngineByName("FooDupe"));
+
+  let hiddenOneOffs =
+    Services.prefs.getCharPref("browser.search.hiddenOneOffs").split(",");
+
+  is(hiddenOneOffs.length, 1,
+     "hiddenOneOffs has the correct engine count post removal.");
+  is(hiddenOneOffs.some(x => x == "FooDupe"), false,
+     "Removed Engine is not in hiddenOneOffs after removal");
+  is(hiddenOneOffs.some(x => x == "Foo"), true,
+     "Current hidden engine is not affected by removal.");
+
+  info("Removing testEngine.xml");
+  Services.search.removeEngine(Services.search.getEngineByName("Foo"));
+
+  is(Services.prefs.getCharPref("browser.search.hiddenOneOffs"), "",
+     "hiddenOneOffs is empty after removing all hidden engines.");
+});
+
+add_task(function* test_add() {
+  yield promiseNewEngine("testEngine.xml");
+  info("setting prefs to " + testPref);
+  Services.prefs.setCharPref("browser.search.hiddenOneOffs", testPref);
+  yield promiseNewEngine("testEngine_dupe.xml");
+
+  let hiddenOneOffs =
+    Services.prefs.getCharPref("browser.search.hiddenOneOffs").split(",");
+
+  is(hiddenOneOffs.length, 1,
+     "hiddenOneOffs has the correct number of hidden engines present post add.");
+  is(hiddenOneOffs.some(x => x == "FooDupe"), false,
+     "Added engine is not present in hidden list.");
+  is(hiddenOneOffs.some(x => x == "Foo"), true,
+     "Adding an engine does not remove engines from hidden list.");
+});
+
+add_task(function* test_diacritics() {
+  const diacritic_engine = "Foo \u2661";
+  let Preferences =
+    Cu.import("resource://gre/modules/Preferences.jsm", {}).Preferences;
+
+  Preferences.set("browser.search.hiddenOneOffs", diacritic_engine);
+  yield promiseNewEngine("testEngine_diacritics.xml");
+
+  let hiddenOneOffs =
+    Preferences.get("browser.search.hiddenOneOffs").split(",");
+  is(hiddenOneOffs.some(x => x == diacritic_engine), false,
+     "Observer cleans up added hidden engines that include a diacritic.");
+
+  Preferences.set("browser.search.hiddenOneOffs", diacritic_engine);
+
+  info("Removing testEngine_diacritics.xml");
+  Services.search.removeEngine(Services.search.getEngineByName(diacritic_engine));
+
+  hiddenOneOffs =
+    Preferences.get("browser.search.hiddenOneOffs").split(",");
+  is(hiddenOneOffs.some(x => x == diacritic_engine), false,
+     "Observer cleans up removed hidden engines that include a diacritic.");
+});
+
+registerCleanupFunction(() => {
+  info("Removing testEngine.xml");
+  Services.search.removeEngine(Services.search.getEngineByName("Foo"));
+  info("Removing testEngine_dupe.xml");
+  Services.search.removeEngine(Services.search.getEngineByName("FooDupe"));
+  Services.prefs.clearUserPref("browser.search.hiddenOneOffs");
+});