1 files changed, 115 insertions, 0 deletions
diff --git a/browser/components/newtab/test/unit/lib/UTEventReporting.test.js b/browser/components/newtab/test/unit/lib/UTEventReporting.test.js
new file mode 100644
index 0000000000..6255568438
--- /dev/null
+++ b/browser/components/newtab/test/unit/lib/UTEventReporting.test.js
@@ -0,0 +1,115 @@
+import { UTSessionPing, UTUserEventPing } from "test/schemas/pings";
+import { GlobalOverrider } from "test/unit/utils";
+import { UTEventReporting } from "lib/UTEventReporting.sys.mjs";
+
+const FAKE_EVENT_PING_PC = {
+  event: "CLICK",
+  source: "TOP_SITES",
+  addon_version: "123",
+  user_prefs: 63,
+  session_id: "abc",
+  page: "about:newtab",
+  action_position: 5,
+  locale: "en-US",
+};
+const FAKE_SESSION_PING_PC = {
+  session_duration: 1234,
+  addon_version: "123",
+  user_prefs: 63,
+  session_id: "abc",
+  page: "about:newtab",
+  locale: "en-US",
+};
+const FAKE_EVENT_PING_UT = [
+  "activity_stream",
+  "event",
+  "CLICK",
+  "TOP_SITES",
+  {
+    addon_version: "123",
+    user_prefs: "63",
+    session_id: "abc",
+    page: "about:newtab",
+    action_position: "5",
+  },
+];
+const FAKE_SESSION_PING_UT = [
+  "activity_stream",
+  "end",
+  "session",
+  "1234",
+  {
+    addon_version: "123",
+    user_prefs: "63",
+    session_id: "abc",
+    page: "about:newtab",
+  },
+];
+
+describe("UTEventReporting", () => {
+  let globals;
+  let sandbox;
+  let utEvents;
+
+  beforeEach(() => {
+    globals = new GlobalOverrider();
+    sandbox = globals.sandbox;
+    sandbox.stub(global.Services.telemetry, "setEventRecordingEnabled");
+    sandbox.stub(global.Services.telemetry, "recordEvent");
+
+    utEvents = new UTEventReporting();
+  });
+
+  afterEach(() => {
+    globals.restore();
+  });
+
+  describe("#sendUserEvent()", () => {
+    it("should queue up the correct data to send to Events Telemetry", async () => {
+      utEvents.sendUserEvent(FAKE_EVENT_PING_PC);
+      assert.calledWithExactly(
+        global.Services.telemetry.recordEvent,
+        ...FAKE_EVENT_PING_UT
+      );
+
+      let ping = global.Services.telemetry.recordEvent.firstCall.args;
+      assert.validate(ping, UTUserEventPing);
+    });
+  });
+
+  describe("#sendSessionEndEvent()", () => {
+    it("should queue up the correct data to send to Events Telemetry", async () => {
+      utEvents.sendSessionEndEvent(FAKE_SESSION_PING_PC);
+      assert.calledWithExactly(
+        global.Services.telemetry.recordEvent,
+        ...FAKE_SESSION_PING_UT
+      );
+
+      let ping = global.Services.telemetry.recordEvent.firstCall.args;
+      assert.validate(ping, UTSessionPing);
+    });
+  });
+
+  describe("#uninit()", () => {
+    it("should call setEventRecordingEnabled with a false value", () => {
+      assert.equal(
+        global.Services.telemetry.setEventRecordingEnabled.firstCall.args[0],
+        "activity_stream"
+      );
+      assert.equal(
+        global.Services.telemetry.setEventRecordingEnabled.firstCall.args[1],
+        true
+      );
+
+      utEvents.uninit();
+      assert.equal(
+        global.Services.telemetry.setEventRecordingEnabled.secondCall.args[0],
+        "activity_stream"
+      );
+      assert.equal(
+        global.Services.telemetry.setEventRecordingEnabled.secondCall.args[1],
+        false
+      );
+    });
+  });
+});