gecko-dev/browser/components/migration/tests/unit/test_Chrome_bookmarks.js
Doug Thayer 0fd7f560d6 Bug 1426245 - Test changes r=mak
MozReview-Commit-ID: 4fhhzspxLJZ

Depends on D4606

Differential Revision: https://phabricator.services.mozilla.com/D5162

--HG--
extra : moz-landing-system : lando
2018-10-09 14:47:31 +00:00

99 lines
3.3 KiB
JavaScript

"use strict";
ChromeUtils.import("resource://gre/modules/AppConstants.jsm");
add_task(async function() {
let rootDir = do_get_file("chromefiles/", true);
let pathId;
let subDirs = ["Google", "Chrome"];
if (AppConstants.platform == "macosx") {
subDirs.unshift("Application Support");
pathId = "ULibDir";
} else if (AppConstants.platform == "win") {
subDirs.push("User Data");
pathId = "LocalAppData";
} else {
subDirs = [".config", "google-chrome"];
pathId = "Home";
}
registerFakePath(pathId, rootDir);
let target = rootDir.clone();
// Pretend this is the default profile
subDirs.push("Default");
while (subDirs.length) {
target.append(subDirs.shift());
}
// We don't import osfile.jsm until after registering the fake path, because
// importing osfile will sometimes greedily fetch certain path identifiers
// from the dir service, which means they get cached, which means we can't
// register a fake path for them anymore.
ChromeUtils.import("resource://gre/modules/osfile.jsm");
await OS.File.makeDir(target.path, {from: rootDir.parent.path, ignoreExisting: true});
target.append("Bookmarks");
await OS.File.remove(target.path, {ignoreAbsent: true});
let bookmarksData = {roots: {bookmark_bar: {children: []}, other: {children: []}}};
const MAX_BMS = 100;
let barKids = bookmarksData.roots.bookmark_bar.children;
let menuKids = bookmarksData.roots.other.children;
let currentMenuKids = menuKids;
let currentBarKids = barKids;
for (let i = 0; i < MAX_BMS; i++) {
currentBarKids.push({
url: "https://www.chrome-bookmark-bar-bookmark" + i + ".com",
name: "bookmark " + i,
type: "url",
});
currentMenuKids.push({
url: "https://www.chrome-menu-bookmark" + i + ".com",
name: "bookmark for menu " + i,
type: "url",
});
if (i % 20 == 19) {
let nextFolder = {
name: "toolbar folder " + Math.ceil(i / 20),
type: "folder",
children: [],
};
currentBarKids.push(nextFolder);
currentBarKids = nextFolder.children;
nextFolder = {
name: "menu folder " + Math.ceil(i / 20),
type: "folder",
children: [],
};
currentMenuKids.push(nextFolder);
currentMenuKids = nextFolder.children;
}
}
await OS.File.writeAtomic(target.path, JSON.stringify(bookmarksData), {encoding: "utf-8"});
let migrator = await MigrationUtils.getMigrator("chrome");
// Sanity check for the source.
Assert.ok(await migrator.isSourceAvailable());
let itemsSeen = {bookmarks: 0, folders: 0};
let listener = events => {
for (let event of events) {
if (!event.title.includes("Chrome")) {
itemsSeen[event.itemType == PlacesUtils.bookmarks.TYPE_FOLDER ? "folders" : "bookmarks"]++;
}
}
};
PlacesUtils.observers.addListener(["bookmark-added"], listener);
const PROFILE = {
id: "Default",
name: "Default",
};
await promiseMigration(migrator, MigrationUtils.resourceTypes.BOOKMARKS, PROFILE);
PlacesUtils.observers.removeListener(["bookmark-added"], listener);
Assert.equal(itemsSeen.bookmarks, 200, "Should have seen 200 bookmarks.");
Assert.equal(itemsSeen.folders, 10, "Should have seen 10 folders.");
Assert.equal(MigrationUtils._importQuantities.bookmarks, itemsSeen.bookmarks + itemsSeen.folders, "Telemetry reporting correct.");
});