1 files changed, 82 insertions, 0 deletions
diff --git a/browser/components/pocket/test/unit/browser_pocket_main.js b/browser/components/pocket/test/unit/browser_pocket_main.js
new file mode 100644
index 0000000000..c6955c09f7
--- /dev/null
+++ b/browser/components/pocket/test/unit/browser_pocket_main.js
@@ -0,0 +1,82 @@
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/ */
+"use strict";
+
+function test_runner(test) {
+  let testTask = async () => {
+    // Before each
+    pktUI.initPrefs();
+    const sandbox = sinon.createSandbox();
+    try {
+      await test({ sandbox });
+    } finally {
+      // After each
+      sandbox.restore();
+    }
+  };
+
+  // Copy the name of the test function to identify the test
+  Object.defineProperty(testTask, "name", { value: test.name });
+  add_task(testTask);
+}
+
+test_runner(async function test_main_getAndShowRecsForItem_on({ sandbox }) {
+  await SpecialPowers.pushPrefEnv({
+    set: [["extensions.pocket.onSaveRecs", true]],
+  });
+  pktUI.initPrefs();
+  const getRecsForItemStub = sandbox.stub(pktApi, "getRecsForItem");
+
+  pktUI.getAndShowRecsForItem(
+    {
+      resolved_id: "1234",
+    },
+    {
+      success() {},
+    }
+  );
+
+  Assert.ok(getRecsForItemStub.calledOnce);
+  Assert.equal(getRecsForItemStub.getCall(0).args[0], "1234");
+});
+
+test_runner(async function test_main_getAndShowRecsForItem_off({ sandbox }) {
+  await SpecialPowers.pushPrefEnv({
+    set: [["extensions.pocket.onSaveRecs", false]],
+  });
+  pktUI.initPrefs();
+  const getRecsForItemStub = sandbox.stub(pktApi, "getRecsForItem");
+
+  pktUI.getAndShowRecsForItem(
+    {
+      resolved_id: "1234",
+    },
+    {
+      success() {},
+    }
+  );
+
+  Assert.ok(getRecsForItemStub.notCalled);
+});
+
+test_runner(async function test_main_getAndShowRecsForItem_locale({ sandbox }) {
+  await SpecialPowers.pushPrefEnv({
+    set: [
+      ["extensions.pocket.onSaveRecs", true],
+      ["extensions.pocket.onSaveRecs.locales", "de"],
+    ],
+  });
+  pktUI.initPrefs();
+  const getRecsForItemStub = sandbox.stub(pktApi, "getRecsForItem");
+
+  pktUI.getAndShowRecsForItem(
+    {
+      resolved_id: "1234",
+    },
+    {
+      success() {},
+    }
+  );
+
+  Assert.ok(getRecsForItemStub.notCalled);
+});