Source code

Revision control

Copy as Markdown

Other Tools

var { NewTabUtils } = ChromeUtils.importESModule(
"resource://gre/modules/NewTabUtils.sys.mjs"
);
Cu.importGlobalProperties(["btoa"]);
ChromeUtils.defineESModuleGetters(this, {
PlacesUtils: "resource://gre/modules/PlacesUtils.sys.mjs",
Sqlite: "resource://gre/modules/Sqlite.sys.mjs",
});
const PREF_NEWTAB_ENHANCED = "browser.newtabpage.enhanced";
const SEARCH_SHORTCUTS_EXPERIMENT_PREF =
"browser.newtabpage.activity-stream.improvesearch.topSiteSearchShortcuts";
// use time at the start of the tests, chnaging it inside timeDaysAgo()
// may cause tiny time differences, which break expected sql ordering
const TIME_NOW = new Date().getTime();
Services.prefs.setBoolPref(PREF_NEWTAB_ENHANCED, true);
Services.prefs.setBoolPref(SEARCH_SHORTCUTS_EXPERIMENT_PREF, false);
do_get_profile();
// utility function to compute past timestamp in microseconds
function timeDaysAgo(numDays) {
return (TIME_NOW - numDays * 24 * 60 * 60 * 1000) * 1000;
}
// tests that timestamp falls within 10 days of now
function isVisitDateOK(timestampMS) {
let range = 10 * 24 * 60 * 60 * 1000;
return Math.abs(Date.now() - timestampMS) < range;
}
// a set up function to prep the activity stream provider
function setUpActivityStreamTest() {
return (async function () {
await PlacesUtils.history.clear();
await PlacesUtils.bookmarks.eraseEverything();
let faviconExpiredPromise = new Promise(resolve => {
Services.obs.addObserver(resolve, "places-favicons-expired");
});
PlacesUtils.favicons.expireAllFavicons();
await faviconExpiredPromise;
})();
}
function do_check_links(actualLinks, expectedLinks) {
Assert.ok(Array.isArray(actualLinks));
Assert.equal(actualLinks.length, expectedLinks.length);
for (let i = 0; i < expectedLinks.length; i++) {
let expected = expectedLinks[i];
let actual = actualLinks[i];
Assert.equal(actual.url, expected.url);
Assert.equal(actual.title, expected.title);
Assert.equal(actual.frecency, expected.frecency);
Assert.equal(actual.lastVisitDate, expected.lastVisitDate);
}
}
function makeLinks(frecRangeStart, frecRangeEnd, step) {
let links = [];
// Remember, links are ordered by frecency descending.
for (let i = frecRangeEnd; i > frecRangeStart; i -= step) {
links.push(makeLink(i));
}
return links;
}
function makeLink(frecency) {
return {
url: "http://example" + frecency + ".com/",
title: "My frecency is " + frecency,
frecency,
lastVisitDate: 0,
};
}