1 files changed, 98 insertions, 0 deletions
diff --git a/toolkit/components/contentrelevancy/tests/xpcshell/test_InputUtils.js b/toolkit/components/contentrelevancy/tests/xpcshell/test_InputUtils.js
new file mode 100644
index 0000000000..2bb2b8e62e
--- /dev/null
+++ b/toolkit/components/contentrelevancy/tests/xpcshell/test_InputUtils.js
@@ -0,0 +1,98 @@
+/* Any copyright is dedicated to the Public Domain.
+   http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+ChromeUtils.defineESModuleGetters(this, {
+  getFrecentRecentCombinedUrls:
+    "resource://gre/modules/contentrelevancy/private/InputUtils.sys.mjs",
+  getMostRecentUrls:
+    "resource://gre/modules/contentrelevancy/private/InputUtils.sys.mjs",
+  getTopFrecentUrls:
+    "resource://gre/modules/contentrelevancy/private/InputUtils.sys.mjs",
+  PlacesUtils: "resource://gre/modules/PlacesUtils.sys.mjs",
+  PlacesTestUtils: "resource://testing-common/PlacesTestUtils.sys.mjs",
+});
+
+const FRECENCY_SCORE_FOR_ONE_VISIT = 100;
+const TEST_VISITS = [
+  "http://test-1.com/",
+  "http://test-2.com/",
+  "http://test-3.com/",
+  "http://test-4.com/",
+];
+
+add_task(async function test_GetTopFrecentUrls() {
+  await PlacesUtils.history.clear();
+  let urls = new Set(await getTopFrecentUrls(3, FRECENCY_SCORE_FOR_ONE_VISIT));
+
+  Assert.strictEqual(urls.size, 0, "Should have no top frecent links.");
+
+  await PlacesTestUtils.addVisits(TEST_VISITS);
+  urls = new Set(await getTopFrecentUrls(3, FRECENCY_SCORE_FOR_ONE_VISIT));
+
+  Assert.strictEqual(urls.size, 3, "Should fetch the expected links");
+  urls.forEach(url => {
+    Assert.ok(TEST_VISITS.includes(url), "Should be a link of the test visits");
+  });
+});
+
+add_task(async function test_GetMostRecentUrls() {
+  await PlacesUtils.history.clear();
+  let urls = new Set(await getMostRecentUrls(3));
+
+  Assert.strictEqual(urls.size, 0, "Should have no recent links.");
+
+  // Add visits and page meta data.
+  await PlacesTestUtils.addVisits(TEST_VISITS);
+  for (let url of TEST_VISITS) {
+    await PlacesUtils.history.update({
+      description: "desc",
+      previewImageURL: "https://image/",
+      url,
+    });
+  }
+
+  urls = new Set(await getMostRecentUrls(3));
+
+  Assert.strictEqual(urls.size, 3, "Should fetch the expected links");
+  urls.forEach(url => {
+    Assert.ok(TEST_VISITS.includes(url), "Should be a link of the test visits");
+  });
+});
+
+add_task(async function test_GetFrecentRecentCombinedUrls() {
+  await PlacesUtils.history.clear();
+  let urls = new Set(await getFrecentRecentCombinedUrls(3));
+
+  Assert.strictEqual(urls.size, 0, "Should have no links.");
+
+  // Add visits and page meta data.
+  await PlacesTestUtils.addVisits(TEST_VISITS);
+  for (let url of TEST_VISITS) {
+    await PlacesUtils.history.update({
+      description: "desc",
+      previewImageURL: "https://image/",
+      url,
+    });
+  }
+
+  urls = new Set(await getFrecentRecentCombinedUrls(3));
+
+  Assert.strictEqual(urls.size, 3, "Should fetch the expected links");
+  urls.forEach(url => {
+    Assert.ok(TEST_VISITS.includes(url), "Should be a link of the test visits");
+  });
+
+  // Try getting twice as many URLs as the total in Places.
+  urls = new Set(await getFrecentRecentCombinedUrls(TEST_VISITS.length * 2));
+
+  Assert.strictEqual(
+    urls.size,
+    TEST_VISITS.length,
+    "Should not include duplicates"
+  );
+  urls.forEach(url => {
+    Assert.ok(TEST_VISITS.includes(url), "Should be a link of the test visits");
+  });
+});