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>
---
 .../tests/xpcshell/test_cleanup.js                 | 84 ++++++++++++++++++++++
 1 file changed, 84 insertions(+)
 create mode 100644 browser/components/enterprisepolicies/tests/xpcshell/test_cleanup.js

(limited to 'browser/components/enterprisepolicies/tests/xpcshell/test_cleanup.js')

diff --git a/browser/components/enterprisepolicies/tests/xpcshell/test_cleanup.js b/browser/components/enterprisepolicies/tests/xpcshell/test_cleanup.js
new file mode 100644
index 0000000000..4171987cbb
--- /dev/null
+++ b/browser/components/enterprisepolicies/tests/xpcshell/test_cleanup.js
@@ -0,0 +1,84 @@
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+async function checkMessages(expectedResult) {
+  let onBeforeAddons = false;
+  let onProfileAfterChange = false;
+  let onBeforeUIStartup = false;
+  let onAllWindowsRestored = false;
+
+  let errorListener = {
+    observe(subject) {
+      let message = subject.wrappedJSObject.arguments[0];
+      if (message.includes("_cleanup from onBeforeAddons")) {
+        onBeforeAddons = true;
+      } else if (message.includes("_cleanup from onProfileAfterChange")) {
+        onProfileAfterChange = true;
+      } else if (message.includes("_cleanup from onBeforeUIStartup")) {
+        onBeforeUIStartup = true;
+      } else if (message.includes("_cleanup from onAllWindowsRestored")) {
+        onAllWindowsRestored = true;
+      }
+    },
+  };
+
+  Services.console.registerListener(errorListener);
+
+  const ConsoleAPIStorage = Cc["@mozilla.org/consoleAPI-storage;1"].getService(
+    Ci.nsIConsoleAPIStorage
+  );
+  ConsoleAPIStorage.addLogEventListener(
+    errorListener.observe,
+    Cc["@mozilla.org/systemprincipal;1"].createInstance(Ci.nsIPrincipal)
+  );
+
+  await setupPolicyEngineWithJson({
+    policies: {},
+  });
+
+  equal(
+    onBeforeAddons,
+    expectedResult,
+    "onBeforeAddons should be " + expectedResult
+  );
+  equal(
+    onProfileAfterChange,
+    expectedResult,
+    "onProfileAfterChange should be" + expectedResult
+  );
+  equal(
+    onBeforeUIStartup,
+    expectedResult,
+    "onBeforeUIStartup should be" + expectedResult
+  );
+  equal(
+    onAllWindowsRestored,
+    expectedResult,
+    "onAllWindowsRestored should be" + expectedResult
+  );
+}
+
+/* If there is no existing policy, cleanup should not run. */
+add_task(async function test_cleanup_no_policy() {
+  await checkMessages(false);
+});
+
+add_task(async function setup_policy() {
+  await setupPolicyEngineWithJson({
+    policies: {
+      BlockAboutConfig: true,
+    },
+  });
+});
+
+/* Since there was a policy, cleanup should run. */
+add_task(async function test_cleanup_with_policy() {
+  await checkMessages(true);
+});
+
+/* Since cleanup was already done, cleanup should not run again. */
+add_task(async function test_cleanup_after_policy() {
+  await checkMessages(false);
+});
-- 
cgit v1.2.3