1 files changed, 50 insertions, 0 deletions
diff --git a/toolkit/components/search/tests/xpcshell/test_searchSuggest_extraParams.js b/toolkit/components/search/tests/xpcshell/test_searchSuggest_extraParams.js
new file mode 100644
index 0000000000..8bac7d39cb
--- /dev/null
+++ b/toolkit/components/search/tests/xpcshell/test_searchSuggest_extraParams.js
@@ -0,0 +1,50 @@
+/* Any copyright is dedicated to the Public Domain.
+   http://creativecommons.org/publicdomain/zero/1.0/ */
+
+const TEST_CONFIG = [
+  {
+    webExtension: { id: "get@search.mozilla.org" },
+    appliesTo: [{ included: { everywhere: true } }],
+    suggestExtraParams: [
+      {
+        name: "custom_param",
+        pref: "test_pref_param",
+        condition: "pref",
+      },
+    ],
+  },
+];
+
+add_task(async function setup() {
+  await SearchTestUtils.useTestEngines("method-extensions", null, TEST_CONFIG);
+  await AddonTestUtils.promiseStartupManager();
+  await Services.search.init();
+});
+
+add_task(async function test_custom_suggest_param() {
+  let engine = Services.search.getEngineByName("Get Engine");
+  Assert.notEqual(engine, null, "Should have found an engine");
+
+  let submissionSuggest = engine.getSubmission(
+    "bar",
+    SearchUtils.URL_TYPE.SUGGEST_JSON
+  );
+  Assert.equal(
+    submissionSuggest.uri.spec,
+    "https://example.com/?webExtension=1&suggest=bar",
+    "Suggest URLs should match"
+  );
+
+  let defaultBranch = Services.prefs.getDefaultBranch("browser.search.");
+  defaultBranch.setCharPref("param.test_pref_param", "good");
+
+  let nextSubmissionSuggest = engine.getSubmission(
+    "bar",
+    SearchUtils.URL_TYPE.SUGGEST_JSON
+  );
+  Assert.equal(
+    nextSubmissionSuggest.uri.spec,
+    "https://example.com/?custom_param=good&webExtension=1&suggest=bar",
+    "Suggest URLs should include custom param"
+  );
+});