From 26a029d407be480d791972afb5975cf62c9360a6 Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@progress-linux.org>
Date: Fri, 19 Apr 2024 02:47:55 +0200
Subject: Adding upstream version 124.0.1.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
---
 .../test/xpcshell/test_ext_theme_experiments.js    | 109 +++++++++++++++++++++
 1 file changed, 109 insertions(+)
 create mode 100644 toolkit/components/extensions/test/xpcshell/test_ext_theme_experiments.js

(limited to 'toolkit/components/extensions/test/xpcshell/test_ext_theme_experiments.js')

diff --git a/toolkit/components/extensions/test/xpcshell/test_ext_theme_experiments.js b/toolkit/components/extensions/test/xpcshell/test_ext_theme_experiments.js
new file mode 100644
index 0000000000..4c3bf7b4d9
--- /dev/null
+++ b/toolkit/components/extensions/test/xpcshell/test_ext_theme_experiments.js
@@ -0,0 +1,109 @@
+/* Any copyright is dedicated to the Public Domain.
+   http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+const { TestUtils } = ChromeUtils.importESModule(
+  "resource://testing-common/TestUtils.sys.mjs"
+);
+
+AddonTestUtils.init(this);
+AddonTestUtils.overrideCertDB();
+AddonTestUtils.usePrivilegedSignatures = id => id.startsWith("privileged");
+AddonTestUtils.createAppInfo(
+  "xpcshell@tests.mozilla.org",
+  "XPCShell",
+  "1",
+  "42"
+);
+
+add_task(async function setup() {
+  await ExtensionTestUtils.startAddonManager();
+});
+
+// This test checks whether the theme experiments work for privileged static themes
+// and are ignored for unprivileged static themes.
+async function test_experiment_static_theme({ privileged }) {
+  let extensionManifest = {
+    theme: {
+      colors: {},
+      images: {},
+      properties: {},
+    },
+    theme_experiment: {
+      colors: {},
+      images: {},
+      properties: {},
+    },
+  };
+
+  const addonId = `${
+    privileged ? "privileged" : "unprivileged"
+  }-static-theme@test-extension`;
+  const themeFiles = {
+    "manifest.json": {
+      name: "test theme",
+      version: "1.0",
+      manifest_version: 2,
+      browser_specific_settings: {
+        gecko: { id: addonId },
+      },
+      ...extensionManifest,
+    },
+  };
+
+  const promiseThemeUpdated = TestUtils.topicObserved(
+    "lightweight-theme-styling-update"
+  );
+
+  let themeAddon;
+  const { messages } = await AddonTestUtils.promiseConsoleOutput(async () => {
+    let { addon } = await AddonTestUtils.promiseInstallXPI(themeFiles);
+    // Enable the newly installed static theme.
+    await addon.enable();
+    themeAddon = addon;
+  });
+
+  const themeExperimentNotAllowed = {
+    message: /This extension is not allowed to run theme experiments/,
+  };
+  AddonTestUtils.checkMessages(messages, {
+    forbidden: privileged ? [themeExperimentNotAllowed] : [],
+    expected: privileged ? [] : [themeExperimentNotAllowed],
+  });
+
+  if (privileged) {
+    // ext-theme.js Theme class constructor doesn't call Theme.prototype.load
+    // if the static theme includes theme_experiment but isn't allowed to.
+    info("Wait for theme updated observer service topic to be notified");
+    const [topicSubject] = await promiseThemeUpdated;
+    let themeData = topicSubject.wrappedJSObject;
+    ok(
+      themeData.experiment,
+      "Expect theme experiment property to be defined in theme update data"
+    );
+  }
+
+  const policy = WebExtensionPolicy.getByID(themeAddon.id);
+  equal(
+    policy.extension.isPrivileged,
+    privileged,
+    `The static theme should be ${privileged ? "privileged" : "unprivileged"}`
+  );
+
+  await themeAddon.uninstall();
+}
+
+add_task(function test_privileged_theme() {
+  return test_experiment_static_theme({ privileged: true });
+});
+
+add_task(
+  {
+    // Some builds (e.g. thunderbird) have experiments enabled by default.
+    pref_set: [["extensions.experiments.enabled", false]],
+  },
+  function test_unprivileged_theme() {
+    return test_experiment_static_theme({ privileged: false });
+  }
+);
-- 
cgit v1.2.3