1 files changed, 56 insertions, 0 deletions
diff --git a/browser/components/pocket/test/unit/browser_pocket_pktTelemetry.js b/browser/components/pocket/test/unit/browser_pocket_pktTelemetry.js
new file mode 100644
index 0000000000..5346f6c158
--- /dev/null
+++ b/browser/components/pocket/test/unit/browser_pocket_pktTelemetry.js
@@ -0,0 +1,56 @@
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/ */
+"use strict";
+
+ChromeUtils.defineModuleGetter(
+  this,
+  "pktTelemetry",
+  "chrome://pocket/content/pktTelemetry.jsm"
+);
+
+function test_runner(test) {
+  let testTask = async () => {
+    // Before each
+    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_createPingPayload({ sandbox }) {
+  const impressionId = "{7fd5a1ac-6089-4212-91a7-fcdec1d2f533}";
+  const creationDate = "18578";
+  await SpecialPowers.pushPrefEnv({
+    set: [["browser.newtabpage.activity-stream.impressionId", impressionId]],
+  });
+  sandbox.stub(pktTelemetry, "_profileCreationDate").returns(creationDate);
+  const result = pktTelemetry.createPingPayload({ test: "test" });
+
+  Assert.deepEqual(result, {
+    test: "test",
+    pocket_logged_in_status: false,
+    profile_creation_date: creationDate,
+    impression_id: impressionId,
+  });
+});
+
+test_runner(async function test_generateStructuredIngestionEndpoint({
+  sandbox,
+}) {
+  sandbox
+    .stub(pktTelemetry, "_generateUUID")
+    .returns("{7fd5a1ac-6089-4212-91a7-fcdec1d2f533}");
+  const endpoint = pktTelemetry._generateStructuredIngestionEndpoint();
+  Assert.equal(
+    endpoint,
+    "https://incoming.telemetry.mozilla.org/submit/activity-stream/pocket-button/1/7fd5a1ac-6089-4212-91a7-fcdec1d2f533"
+  );
+});