From 36d22d82aa202bb199967e9512281e9a53db42c9 Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@progress-linux.org>
Date: Sun, 7 Apr 2024 21:33:14 +0200
Subject: Adding upstream version 115.7.0esr.

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

(limited to 'toolkit/components/normandy/test/browser/browser_ShieldPreferences.js')

diff --git a/toolkit/components/normandy/test/browser/browser_ShieldPreferences.js b/toolkit/components/normandy/test/browser/browser_ShieldPreferences.js
new file mode 100644
index 0000000000..e455a5f25b
--- /dev/null
+++ b/toolkit/components/normandy/test/browser/browser_ShieldPreferences.js
@@ -0,0 +1,91 @@
+"use strict";
+
+const { PreferenceExperiments } = ChromeUtils.importESModule(
+  "resource://normandy/lib/PreferenceExperiments.sys.mjs"
+);
+const { ShieldPreferences } = ChromeUtils.importESModule(
+  "resource://normandy/lib/ShieldPreferences.sys.mjs"
+);
+
+const OPT_OUT_STUDIES_ENABLED_PREF = "app.shield.optoutstudies.enabled";
+
+const { NormandyTestUtils } = ChromeUtils.importESModule(
+  "resource://testing-common/NormandyTestUtils.sys.mjs"
+);
+const { addonStudyFactory, preferenceStudyFactory } =
+  NormandyTestUtils.factories;
+
+ShieldPreferences.init();
+
+decorate_task(
+  withMockPreferences(),
+  AddonStudies.withStudies([
+    addonStudyFactory({ active: true }),
+    addonStudyFactory({ active: true }),
+  ]),
+  async function testDisableStudiesWhenOptOutDisabled({
+    mockPreferences,
+    addonStudies: [study1, study2],
+  }) {
+    mockPreferences.set(OPT_OUT_STUDIES_ENABLED_PREF, true);
+    const observers = [
+      studyEndObserved(study1.recipeId),
+      studyEndObserved(study2.recipeId),
+    ];
+    Services.prefs.setBoolPref(OPT_OUT_STUDIES_ENABLED_PREF, false);
+    await Promise.all(observers);
+
+    const newStudy1 = await AddonStudies.get(study1.recipeId);
+    const newStudy2 = await AddonStudies.get(study2.recipeId);
+    ok(
+      !newStudy1.active && !newStudy2.active,
+      "Setting the opt-out pref to false stops all active opt-out studies."
+    );
+  }
+);
+
+decorate_task(
+  withMockPreferences(),
+  PreferenceExperiments.withMockExperiments([
+    preferenceStudyFactory({ active: true }),
+    preferenceStudyFactory({ active: true }),
+  ]),
+  withStub(PreferenceExperiments, "stop"),
+  async function testDisableExperimentsWhenOptOutDisabled({
+    mockPreferences,
+    prefExperiments: [study1, study2],
+    stopStub,
+  }) {
+    mockPreferences.set(OPT_OUT_STUDIES_ENABLED_PREF, true);
+    let stopArgs = [];
+    let stoppedBoth = new Promise(resolve => {
+      let calls = 0;
+      stopStub.callsFake(function () {
+        stopArgs.push(Array.from(arguments));
+        calls++;
+        if (calls == 2) {
+          resolve();
+        }
+      });
+    });
+    Services.prefs.setBoolPref(OPT_OUT_STUDIES_ENABLED_PREF, false);
+    await stoppedBoth;
+
+    Assert.deepEqual(stopArgs, [
+      [
+        study1.slug,
+        {
+          reason: "general-opt-out",
+          caller: "observePrefChange::general-opt-out",
+        },
+      ],
+      [
+        study2.slug,
+        {
+          reason: "general-opt-out",
+          caller: "observePrefChange::general-opt-out",
+        },
+      ],
+    ]);
+  }
+);
-- 
cgit v1.2.3