1 files changed, 29 insertions, 0 deletions
diff --git a/browser/components/enterprisepolicies/tests/browser/browser_policy_disable_pocket.js b/browser/components/enterprisepolicies/tests/browser/browser_policy_disable_pocket.js
new file mode 100644
index 0000000000..414b855f8c
--- /dev/null
+++ b/browser/components/enterprisepolicies/tests/browser/browser_policy_disable_pocket.js
@@ -0,0 +1,29 @@
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+const PREF_POCKET = "extensions.pocket.enabled";
+
+async function checkPocket(shouldBeEnabled) {
+  return BrowserTestUtils.waitForCondition(() => {
+    return (
+      !!CustomizableUI.getWidget("save-to-pocket-button") == shouldBeEnabled
+    );
+  }, "Expecting Pocket to be " + shouldBeEnabled);
+}
+
+add_task(async function test_disable_firefox_screenshots() {
+  await BrowserTestUtils.withNewTab("data:text/html,Test", async function () {
+    // Sanity check to make sure Pocket is enabled on tests
+    await checkPocket(true);
+
+    await setupPolicyEngineWithJson({
+      policies: {
+        DisablePocket: true,
+      },
+    });
+
+    await checkPocket(false);
+  });
+});