1 files changed, 105 insertions, 0 deletions
diff --git a/browser/components/migration/tests/unit/test_Safari_bookmarks.js b/browser/components/migration/tests/unit/test_Safari_bookmarks.js
new file mode 100644
index 0000000000..79cbcf9203
--- /dev/null
+++ b/browser/components/migration/tests/unit/test_Safari_bookmarks.js
@@ -0,0 +1,105 @@
+"use strict";
+
+const { CustomizableUI } = ChromeUtils.import(
+  "resource:///modules/CustomizableUI.jsm"
+);
+
+add_task(async function() {
+  registerFakePath("ULibDir", do_get_file("Library/"));
+
+  let migrator = await MigrationUtils.getMigrator("safari");
+  // Sanity check for the source.
+  Assert.ok(await migrator.isSourceAvailable());
+
+  // Wait for the imported bookmarks. We don't check that "From Safari"
+  // folders are created on the toolbar since the profile
+  // we're importing to has less than 3 bookmarks in the destination
+  // so a "From Safari" folder isn't created.
+  let expectedParents = [PlacesUtils.toolbarFolderId];
+  let bookmarkRoots = 0;
+  let bookmarkRootMap = {
+    [PlacesUtils.bookmarks.toolbarGuid]:
+      MigrationUtils.SOURCE_BOOKMARK_ROOTS_BOOKMARKS_TOOLBAR,
+    [PlacesUtils.bookmarks.menuGuid]:
+      MigrationUtils.SOURCE_BOOKMARK_ROOTS_BOOKMARKS_MENU,
+    [PlacesUtils.bookmarks.unfiledGuid]:
+      MigrationUtils.SOURCE_BOOKMARK_ROOTS_UNFILED,
+  };
+  let itemCount = 0;
+
+  let gotFolder = false;
+  let listener = events => {
+    for (let event of events) {
+      itemCount++;
+      if (
+        event.itemType == PlacesUtils.bookmarks.TYPE_BOOKMARK &&
+        bookmarkRootMap[event.parentGuid]
+      ) {
+        bookmarkRoots |= bookmarkRootMap[event.parentGuid];
+      }
+
+      if (
+        event.itemType == PlacesUtils.bookmarks.TYPE_FOLDER &&
+        event.title == "Stuff"
+      ) {
+        gotFolder = true;
+      }
+      if (expectedParents.length) {
+        let index = expectedParents.indexOf(event.parentId);
+        Assert.ok(index != -1, "Found expected parent");
+        expectedParents.splice(index, 1);
+      }
+    }
+  };
+  PlacesUtils.observers.addListener(["bookmark-added"], listener);
+  let observerNotified = false;
+  Services.obs.addObserver((aSubject, aTopic, aData) => {
+    let [toolbar, visibility] = JSON.parse(aData);
+    Assert.equal(
+      toolbar,
+      CustomizableUI.AREA_BOOKMARKS,
+      "Notification should be received for bookmarks toolbar"
+    );
+    Assert.equal(
+      visibility,
+      "true",
+      "Notification should say to reveal the bookmarks toolbar"
+    );
+    observerNotified = true;
+  }, "browser-set-toolbar-visibility");
+
+  await promiseMigration(migrator, MigrationUtils.resourceTypes.BOOKMARKS);
+  PlacesUtils.observers.removeListener(["bookmark-added"], listener);
+
+  // Check the bookmarks have been imported to all the expected parents.
+  Assert.ok(!expectedParents.length, "No more expected parents");
+  Assert.ok(gotFolder, "Should have seen the folder get imported");
+  Assert.equal(itemCount, 13, "Should import all 13 items.");
+  // Check that the telemetry matches:
+  Assert.equal(
+    MigrationUtils._importQuantities.bookmarks,
+    itemCount,
+    "Telemetry reporting correct."
+  );
+  let telemetryRootsMatchesExpectations = await TestUtils.waitForCondition(
+    () => {
+      let snapshot = Services.telemetry.getSnapshotForKeyedHistograms(
+        "main",
+        false
+      ).parent.FX_MIGRATION_BOOKMARKS_ROOTS;
+      if (!snapshot || !snapshot.safari) {
+        return false;
+      }
+      let sum = arr => Object.values(arr).reduce((a, b) => a + b, 0);
+      let sumOfValues = sum(snapshot.safari.values);
+      info(`Expected ${bookmarkRoots}, got ${sumOfValues}`);
+      return sumOfValues == bookmarkRoots;
+    },
+    "Wait until telemetry is updated"
+  );
+  ok(
+    telemetryRootsMatchesExpectations,
+    "The value in the roots histogram should match expectations"
+  );
+  Assert.ok(observerNotified, "The observer should be notified upon migration");
+});