diff options
Diffstat (limited to 'toolkit/modules/tests/xpcshell/head.js')
-rw-r--r-- | toolkit/modules/tests/xpcshell/head.js | 78 |
1 files changed, 78 insertions, 0 deletions
diff --git a/toolkit/modules/tests/xpcshell/head.js b/toolkit/modules/tests/xpcshell/head.js new file mode 100644 index 0000000000..6d133495f6 --- /dev/null +++ b/toolkit/modules/tests/xpcshell/head.js @@ -0,0 +1,78 @@ +var { NewTabUtils } = ChromeUtils.importESModule( + "resource://gre/modules/NewTabUtils.sys.mjs" +); +Cu.importGlobalProperties(["btoa"]); + +ChromeUtils.defineESModuleGetters(this, { + PlacesTestUtils: "resource://testing-common/PlacesTestUtils.sys.mjs", + 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, + }; +} |