1 files changed, 90 insertions, 0 deletions
diff --git a/toolkit/components/reader/test/browser_readerMode_pocket.js b/toolkit/components/reader/test/browser_readerMode_pocket.js
new file mode 100644
index 0000000000..9f0f051488
--- /dev/null
+++ b/toolkit/components/reader/test/browser_readerMode_pocket.js
@@ -0,0 +1,90 @@
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/ */
+"use strict";
+
+// This test verifies that the Save To Pocket button appears in reader mode,
+// and is toggled hidden and visible when pocket is disabled and enabled.
+
+const TEST_PATH = getRootDirectory(gTestPath).replace(
+  "chrome://mochitests/content",
+  "http://example.com"
+);
+
+async function getPocketButtonsCount(browser) {
+  return SpecialPowers.spawn(browser, [], () => {
+    return content.document.getElementsByClassName("pocket-button").length;
+  });
+}
+
+add_task(async function() {
+  // set the pocket preference before beginning.
+  await SpecialPowers.pushPrefEnv({
+    set: [["extensions.pocket.enabled", true]],
+  });
+
+  var readerButton = document.getElementById("reader-mode-button");
+
+  let tab1 = await BrowserTestUtils.openNewForegroundTab(
+    gBrowser,
+    TEST_PATH + "readerModeArticleShort.html"
+  );
+
+  let promiseTabLoad = promiseTabLoadEvent(tab1);
+  readerButton.click();
+  await promiseTabLoad;
+
+  let tab2 = await BrowserTestUtils.openNewForegroundTab(
+    gBrowser,
+    TEST_PATH + "readerModeArticleMedium.html"
+  );
+
+  promiseTabLoad = promiseTabLoadEvent(tab2);
+  readerButton.click();
+  await promiseTabLoad;
+
+  is(
+    await getPocketButtonsCount(tab1.linkedBrowser),
+    1,
+    "tab 1 has a pocket button"
+  );
+  is(
+    await getPocketButtonsCount(tab1.linkedBrowser),
+    1,
+    "tab 2 has a pocket button"
+  );
+
+  // Turn off the pocket preference. The Save To Pocket buttons should disappear.
+  await SpecialPowers.pushPrefEnv({
+    set: [["extensions.pocket.enabled", false]],
+  });
+
+  is(
+    await getPocketButtonsCount(tab1.linkedBrowser),
+    0,
+    "tab 1 has no pocket button"
+  );
+  is(
+    await getPocketButtonsCount(tab1.linkedBrowser),
+    0,
+    "tab 2 has no pocket button"
+  );
+
+  // Turn on the pocket preference. The Save To Pocket buttons should reappear again.
+  await SpecialPowers.pushPrefEnv({
+    set: [["extensions.pocket.enabled", true]],
+  });
+
+  is(
+    await getPocketButtonsCount(tab1.linkedBrowser),
+    1,
+    "tab 1 has a pocket button again"
+  );
+  is(
+    await getPocketButtonsCount(tab1.linkedBrowser),
+    1,
+    "tab 2 has a pocket button again"
+  );
+
+  BrowserTestUtils.removeTab(tab1);
+  BrowserTestUtils.removeTab(tab2);
+});