1 files changed, 81 insertions, 0 deletions
diff --git a/toolkit/components/contentrelevancy/tests/xpcshell/test_Telemetry.js b/toolkit/components/contentrelevancy/tests/xpcshell/test_Telemetry.js
new file mode 100644
index 0000000000..9deef6bad5
--- /dev/null
+++ b/toolkit/components/contentrelevancy/tests/xpcshell/test_Telemetry.js
@@ -0,0 +1,81 @@
+/* Any copyright is dedicated to the Public Domain.
+   http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+const { ContentRelevancyManager } = ChromeUtils.importESModule(
+  "resource://gre/modules/ContentRelevancyManager.sys.mjs"
+);
+
+const PREF_CONTENT_RELEVANCY_ENABLED = "toolkit.contentRelevancy.enabled";
+
+add_setup(async function setup() {
+  // FOG needs a profile directory to put its data in.
+  do_get_profile();
+
+  // FOG needs to be initialized in order for data to flow.
+  Services.fog.initializeFOG();
+
+  Services.prefs.setBoolPref(PREF_CONTENT_RELEVANCY_ENABLED, true);
+  ContentRelevancyManager.init();
+
+  registerCleanupFunction(() => {
+    Services.prefs.clearUserPref(PREF_CONTENT_RELEVANCY_ENABLED);
+  });
+});
+
+/**
+ * Test classification metrics - succeed.
+ */
+add_task(async function test_classify_succeed() {
+  Services.fog.testResetFOG();
+
+  Assert.equal(null, Glean.relevancyClassify.succeed.testGetValue());
+  Assert.equal(null, Glean.relevancyClassify.duration.testGetValue());
+
+  await ContentRelevancyManager._test_doClassification();
+
+  Assert.deepEqual(
+    {
+      input_size: 0,
+      input_classified_size: 0,
+      input_inconclusive_size: 0,
+      output_interest_size: 0,
+      interest_top_1_hits: 0,
+      interest_top_2_hits: 0,
+      interest_top_3_hits: 0,
+    },
+    Glean.relevancyClassify.succeed.testGetValue()[0].extra,
+    "Should record the succeed event"
+  );
+  Assert.ok(
+    Glean.relevancyClassify.duration.testGetValue().sum > 0,
+    "Should record the duration"
+  );
+});
+
+/**
+ * Test classification metrics - fail - insufficient-input.
+ */
+add_task(async function test_classify_fail_case1() {
+  Services.fog.testResetFOG();
+
+  Assert.equal(null, Glean.relevancyClassify.fail.testGetValue());
+  Assert.equal(null, Glean.relevancyClassify.duration.testGetValue());
+
+  // Require at least 1 input to trigger the failure.
+  await ContentRelevancyManager._test_doClassification({ minUrlsForTest: 1 });
+
+  Assert.deepEqual(
+    {
+      reason: "insufficient-input",
+    },
+    Glean.relevancyClassify.fail.testGetValue()[0].extra,
+    "Should record the fail event"
+  );
+  Assert.equal(
+    null,
+    Glean.relevancyClassify.duration.testGetValue(),
+    "Should not record the duration"
+  );
+});