From 6bf0a5cb5034a7e684dcc3500e841785237ce2dd Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@progress-linux.org>
Date: Sun, 7 Apr 2024 19:32:43 +0200
Subject: Adding upstream version 1:115.7.0.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
---
 .../tests/browser/browser_policy_disable_shield.js | 68 ++++++++++++++++++++++
 1 file changed, 68 insertions(+)
 create mode 100644 browser/components/enterprisepolicies/tests/browser/browser_policy_disable_shield.js

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

diff --git a/browser/components/enterprisepolicies/tests/browser/browser_policy_disable_shield.js b/browser/components/enterprisepolicies/tests/browser/browser_policy_disable_shield.js
new file mode 100644
index 0000000000..ef36a0d36c
--- /dev/null
+++ b/browser/components/enterprisepolicies/tests/browser/browser_policy_disable_shield.js
@@ -0,0 +1,68 @@
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+add_task(async function test_policy_disable_shield() {
+  const { RecipeRunner } = ChromeUtils.importESModule(
+    "resource://normandy/lib/RecipeRunner.sys.mjs"
+  );
+  const { BaseAction } = ChromeUtils.importESModule(
+    "resource://normandy/actions/BaseAction.sys.mjs"
+  );
+  const { BaseStudyAction } = ChromeUtils.importESModule(
+    "resource://normandy/actions/BaseStudyAction.sys.mjs"
+  );
+
+  const baseAction = new BaseAction();
+  const baseStudyAction = new BaseStudyAction();
+
+  await SpecialPowers.pushPrefEnv({
+    set: [
+      ["app.normandy.api_url", "https://localhost/selfsupport-dummy/"],
+      ["app.shield.optoutstudies.enabled", true],
+    ],
+  });
+
+  ok(RecipeRunner, "RecipeRunner exists");
+
+  RecipeRunner.checkPrefs();
+  ok(RecipeRunner.enabled, "RecipeRunner is enabled");
+
+  baseAction._preExecution();
+  is(
+    baseAction.state,
+    BaseAction.STATE_PREPARING,
+    "Base action is not disabled"
+  );
+
+  baseStudyAction._preExecution();
+  is(
+    baseStudyAction.state,
+    BaseAction.STATE_PREPARING,
+    "Base study action is not disabled"
+  );
+
+  await setupPolicyEngineWithJson({
+    policies: {
+      DisableFirefoxStudies: true,
+    },
+  });
+
+  RecipeRunner.checkPrefs();
+  ok(RecipeRunner.enabled, "RecipeRunner is still enabled");
+
+  baseAction._preExecution();
+  is(
+    baseAction.state,
+    BaseAction.STATE_PREPARING,
+    "Base action is not disabled"
+  );
+
+  baseStudyAction._preExecution();
+  is(
+    baseStudyAction.state,
+    BaseAction.STATE_DISABLED,
+    "Base study action is disabled"
+  );
+});
-- 
cgit v1.2.3