1 files changed, 28 insertions, 0 deletions
diff --git a/toolkit/components/messaging-system/schemas/SpecialMessageActionSchemas/test/browser/browser_sma_disable_doh.js b/toolkit/components/messaging-system/schemas/SpecialMessageActionSchemas/test/browser/browser_sma_disable_doh.js
new file mode 100644
index 0000000000..aa61214360
--- /dev/null
+++ b/toolkit/components/messaging-system/schemas/SpecialMessageActionSchemas/test/browser/browser_sma_disable_doh.js
@@ -0,0 +1,28 @@
+/* Any copyright is dedicated to the Public Domain.
+   http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+const DOH_DOORHANGER_DECISION_PREF = "doh-rollout.doorhanger-decision";
+const NETWORK_TRR_MODE_PREF = "network.trr.mode";
+
+add_task(async function test_disable_doh() {
+  await SpecialPowers.pushPrefEnv({
+    set: [
+      [DOH_DOORHANGER_DECISION_PREF, "mochitest"],
+      [NETWORK_TRR_MODE_PREF, 0],
+    ],
+  });
+
+  await SMATestUtils.executeAndValidateAction({ type: "DISABLE_DOH" });
+
+  Assert.equal(
+    Services.prefs.getStringPref(DOH_DOORHANGER_DECISION_PREF, ""),
+    "UIDisabled",
+    "Pref should be set on disabled"
+  );
+  Assert.equal(
+    Services.prefs.getIntPref(NETWORK_TRR_MODE_PREF, 0),
+    5,
+    "Pref should be set on disabled"
+  );
+});