1 files changed, 52 insertions, 0 deletions
diff --git a/browser/components/enterprisepolicies/tests/xpcshell/test_defaultbrowsercheck.js b/browser/components/enterprisepolicies/tests/xpcshell/test_defaultbrowsercheck.js
new file mode 100644
index 0000000000..6daffa061a
--- /dev/null
+++ b/browser/components/enterprisepolicies/tests/xpcshell/test_defaultbrowsercheck.js
@@ -0,0 +1,52 @@
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+const { ShellService } = ChromeUtils.import(
+  "resource:///modules/ShellService.jsm"
+);
+
+add_task(async function test_default_browser_check() {
+  ShellService._checkedThisSession = false;
+  // On a normal profile, the default is true. However, this gets set to false on the
+  // testing profile. Let's start with true for a sanity check.
+
+  ShellService.shouldCheckDefaultBrowser = true;
+  equal(ShellService.shouldCheckDefaultBrowser, true, "Sanity check");
+
+  await setupPolicyEngineWithJson({
+    policies: {
+      DontCheckDefaultBrowser: true,
+    },
+  });
+
+  equal(
+    ShellService.shouldCheckDefaultBrowser,
+    false,
+    "Policy changed it to not check"
+  );
+
+  // Try to change it to true and check that it doesn't take effect
+  ShellService.shouldCheckDefaultBrowser = true;
+
+  equal(ShellService.shouldCheckDefaultBrowser, false, "Policy is enforced");
+});
+
+add_task(async function test_default_browser_check() {
+  await setupPolicyEngineWithJson({
+    policies: {
+      DontCheckDefaultBrowser: false,
+    },
+  });
+
+  equal(
+    ShellService.shouldCheckDefaultBrowser,
+    true,
+    "Policy changed it to check"
+  );
+
+  // Try to change it to false and check that it doesn't take effect
+  ShellService.shouldCheckDefaultBrowser = false;
+
+  equal(ShellService.shouldCheckDefaultBrowser, true, "Policy is enforced");
+});