Adding upstream version 124.0.1.upstream/124.0.1

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 221 insertions, 0 deletions

diff --git a/toolkit/components/search/tests/xpcshell/test_webextensions_install.js b/toolkit/components/search/tests/xpcshell/test_webextensions_install.js
new file mode 100644
index 0000000000..6183d6f95a
--- /dev/null
+++ b/toolkit/components/search/tests/xpcshell/test_webextensions_install.js
@@ -0,0 +1,221 @@
+/* Any copyright is dedicated to the Public Domain.
+   http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+const { promiseShutdownManager, promiseStartupManager } = AddonTestUtils;
+
+let gBaseUrl;
+
+async function getEngineNames() {
+  let engines = await Services.search.getEngines();
+  return engines.map(engine => engine._name);
+}
+
+add_setup(async function () {
+  let server = useHttpServer();
+  server.registerContentType("sjs", "sjs");
+  gBaseUrl = `http://localhost:${server.identity.primaryPort}/`;
+
+  await SearchTestUtils.useTestEngines("test-extensions");
+  await promiseStartupManager();
+
+  Services.locale.availableLocales = [
+    ...Services.locale.availableLocales,
+    "af",
+  ];
+
+  registerCleanupFunction(async () => {
+    await promiseShutdownManager();
+    Services.prefs.clearUserPref("browser.search.region");
+  });
+});
+
+add_task(async function basic_install_test() {
+  await Services.search.init();
+  await promiseAfterSettings();
+
+  // On first boot, we get the configuration defaults
+  Assert.deepEqual(await getEngineNames(), ["Plain", "Special"]);
+
+  // User installs a new search engine
+  let extension = await SearchTestUtils.installSearchExtension(
+    {
+      encoding: "windows-1252",
+    },
+    { skipUnload: true }
+  );
+  Assert.deepEqual((await getEngineNames()).sort(), [
+    "Example",
+    "Plain",
+    "Special",
+  ]);
+
+  let engine = await Services.search.getEngineByName("Example");
+  Assert.equal(
+    engine.wrappedJSObject.queryCharset,
+    "windows-1252",
+    "Should have the correct charset"
+  );
+
+  // User uninstalls their engine
+  await extension.awaitStartup();
+  await extension.unload();
+  await promiseAfterSettings();
+  Assert.deepEqual(await getEngineNames(), ["Plain", "Special"]);
+});
+
+add_task(async function test_install_duplicate_engine() {
+  let name = "Plain";
+  consoleAllowList.push(`An engine called ${name} already exists`);
+  let extension = await SearchTestUtils.installSearchExtension(
+    {
+      name,
+      search_url: "https://example.com/plain",
+    },
+    { skipUnload: true }
+  );
+
+  let engine = await Services.search.getEngineByName("Plain");
+  let submission = engine.getSubmission("foo");
+  Assert.equal(
+    submission.uri.spec,
+    "https://duckduckgo.com/?q=foo&t=ffsb",
+    "Should have not changed the app provided engine."
+  );
+
+  // User uninstalls their engine
+  await extension.unload();
+});
+
+add_task(
+  // Not needed for new configuration.
+  { skip_if: () => SearchUtils.newSearchConfigEnabled },
+  async function basic_multilocale_test() {
+    await promiseSetHomeRegion("an");
+
+    Assert.deepEqual(await getEngineNames(), [
+      "Plain",
+      "Special",
+      "Multilocale AN",
+    ]);
+  }
+);
+
+add_task(
+  // Not needed for new configuration.
+  { skip_if: () => SearchUtils.newSearchConfigEnabled },
+  async function complex_multilocale_test() {
+    await promiseSetHomeRegion("af");
+
+    Assert.deepEqual(await getEngineNames(), [
+      "Plain",
+      "Special",
+      "Multilocale AF",
+      "Multilocale AN",
+    ]);
+  }
+);
+
+add_task(
+  // Not needed for new configuration.
+  { skip_if: () => SearchUtils.newSearchConfigEnabled },
+  async function test_manifest_selection() {
+    // Sets the home region without updating.
+    Region._setHomeRegion("an", false);
+    await promiseSetLocale("af");
+
+    let engine = await Services.search.getEngineByName("Multilocale AN");
+    Assert.ok(
+      engine.getIconURL().endsWith("favicon-an.ico"),
+      "Should have the correct favicon for an extension of one locale using a different locale."
+    );
+    Assert.equal(
+      engine.description,
+      "A enciclopedia Libre",
+      "Should have the correct engine name for an extension of one locale using a different locale."
+    );
+  }
+);
+
+add_task(async function test_load_favicon_invalid() {
+  let observed = TestUtils.consoleMessageObserved(msg => {
+    return msg.wrappedJSObject.arguments[0].includes(
+      "Content type does not match expected"
+    );
+  });
+
+  // User installs a new search engine
+  let extension = await SearchTestUtils.installSearchExtension(
+    {
+      favicon_url: `${gBaseUrl}/head_search.js`,
+    },
+    { skipUnload: true }
+  );
+
+  await observed;
+
+  let engine = await Services.search.getEngineByName("Example");
+  Assert.equal(null, engine.getIconURL(), "Should not have set an iconURI");
+
+  // User uninstalls their engine
+  await extension.awaitStartup();
+  await extension.unload();
+  await promiseAfterSettings();
+});
+
+add_task(async function test_load_favicon_invalid_redirect() {
+  let observed = TestUtils.consoleMessageObserved(msg => {
+    return msg.wrappedJSObject.arguments[0].includes(
+      "Content type does not match expected"
+    );
+  });
+
+  // User installs a new search engine
+  let extension = await SearchTestUtils.installSearchExtension(
+    {
+      favicon_url: `${gDataUrl}/iconsRedirect.sjs?type=invalid`,
+    },
+    { skipUnload: true }
+  );
+
+  await observed;
+
+  let engine = await Services.search.getEngineByName("Example");
+  Assert.equal(null, engine.getIconURL(), "Should not have set an iconURI");
+
+  // User uninstalls their engine
+  await extension.awaitStartup();
+  await extension.unload();
+  await promiseAfterSettings();
+});
+
+add_task(async function test_load_favicon_redirect() {
+  let promiseEngineChanged = SearchTestUtils.promiseSearchNotification(
+    SearchUtils.MODIFIED_TYPE.CHANGED,
+    SearchUtils.TOPIC_ENGINE_MODIFIED
+  );
+
+  // User installs a new search engine
+  let extension = await SearchTestUtils.installSearchExtension(
+    {
+      favicon_url: `${gDataUrl}/iconsRedirect.sjs`,
+    },
+    { skipUnload: true }
+  );
+
+  let engine = await Services.search.getEngineByName("Example");
+
+  await promiseEngineChanged;
+
+  Assert.ok(engine.getIconURL(), "Should have set an iconURI");
+  Assert.ok(
+    engine.getIconURL().startsWith("data:image/x-icon;base64,"),
+    "Should have saved the expected content type for the icon"
+  );
+
+  // User uninstalls their engine
+  await extension.awaitStartup();
+  await extension.unload();
+  await promiseAfterSettings();
+});