1 files changed, 189 insertions, 0 deletions
diff --git a/toolkit/components/places/tests/bookmarks/test_bookmarks_eraseEverything.js b/toolkit/components/places/tests/bookmarks/test_bookmarks_eraseEverything.js
new file mode 100644
index 0000000000..2ef2ec2cc2
--- /dev/null
+++ b/toolkit/components/places/tests/bookmarks/test_bookmarks_eraseEverything.js
@@ -0,0 +1,189 @@
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/ */
+
+add_task(async function test_eraseEverything() {
+  await PlacesTestUtils.addVisits({
+    uri: NetUtil.newURI("http://example.com/"),
+  });
+  await PlacesTestUtils.addVisits({
+    uri: NetUtil.newURI("http://mozilla.org/"),
+  });
+  let frecencyForExample = frecencyForUrl("http://example.com/");
+  let frecencyForMozilla = frecencyForUrl("http://example.com/");
+  Assert.ok(frecencyForExample > 0);
+  Assert.ok(frecencyForMozilla > 0);
+  let unfiledFolder = await PlacesUtils.bookmarks.insert({
+    parentGuid: PlacesUtils.bookmarks.unfiledGuid,
+    type: PlacesUtils.bookmarks.TYPE_FOLDER,
+  });
+  checkBookmarkObject(unfiledFolder);
+  let unfiledBookmark = await PlacesUtils.bookmarks.insert({
+    parentGuid: PlacesUtils.bookmarks.unfiledGuid,
+    type: PlacesUtils.bookmarks.TYPE_BOOKMARK,
+    url: "http://example.com/",
+  });
+  checkBookmarkObject(unfiledBookmark);
+  let unfiledBookmarkInFolder = await PlacesUtils.bookmarks.insert({
+    parentGuid: unfiledFolder.guid,
+    type: PlacesUtils.bookmarks.TYPE_BOOKMARK,
+    url: "http://mozilla.org/",
+  });
+  checkBookmarkObject(unfiledBookmarkInFolder);
+
+  let menuFolder = await PlacesUtils.bookmarks.insert({
+    parentGuid: PlacesUtils.bookmarks.menuGuid,
+    type: PlacesUtils.bookmarks.TYPE_FOLDER,
+  });
+  checkBookmarkObject(menuFolder);
+  let menuBookmark = await PlacesUtils.bookmarks.insert({
+    parentGuid: PlacesUtils.bookmarks.menuGuid,
+    type: PlacesUtils.bookmarks.TYPE_BOOKMARK,
+    url: "http://example.com/",
+  });
+  checkBookmarkObject(menuBookmark);
+  let menuBookmarkInFolder = await PlacesUtils.bookmarks.insert({
+    parentGuid: menuFolder.guid,
+    type: PlacesUtils.bookmarks.TYPE_BOOKMARK,
+    url: "http://mozilla.org/",
+  });
+  checkBookmarkObject(menuBookmarkInFolder);
+
+  let toolbarFolder = await PlacesUtils.bookmarks.insert({
+    parentGuid: PlacesUtils.bookmarks.toolbarGuid,
+    type: PlacesUtils.bookmarks.TYPE_FOLDER,
+  });
+  checkBookmarkObject(toolbarFolder);
+  let toolbarBookmark = await PlacesUtils.bookmarks.insert({
+    parentGuid: PlacesUtils.bookmarks.toolbarGuid,
+    type: PlacesUtils.bookmarks.TYPE_BOOKMARK,
+    url: "http://example.com/",
+  });
+  checkBookmarkObject(toolbarBookmark);
+  let toolbarBookmarkInFolder = await PlacesUtils.bookmarks.insert({
+    parentGuid: toolbarFolder.guid,
+    type: PlacesUtils.bookmarks.TYPE_BOOKMARK,
+    url: "http://mozilla.org/",
+  });
+  checkBookmarkObject(toolbarBookmarkInFolder);
+
+  await PlacesTestUtils.promiseAsyncUpdates();
+  Assert.ok(frecencyForUrl("http://example.com/") > frecencyForExample);
+  Assert.ok(frecencyForUrl("http://example.com/") > frecencyForMozilla);
+
+  const promise = PlacesTestUtils.waitForNotification(
+    "pages-rank-changed",
+    () => true,
+    "places"
+  );
+
+  await PlacesUtils.bookmarks.eraseEverything();
+
+  // Ensure we get an pages-rank-changed event.
+  await promise;
+
+  Assert.equal(frecencyForUrl("http://example.com/"), frecencyForExample);
+  Assert.equal(frecencyForUrl("http://example.com/"), frecencyForMozilla);
+});
+
+add_task(async function test_eraseEverything_roots() {
+  await PlacesUtils.bookmarks.eraseEverything();
+
+  // Ensure the roots have not been removed.
+  Assert.ok(
+    await PlacesUtils.bookmarks.fetch(PlacesUtils.bookmarks.unfiledGuid)
+  );
+  Assert.ok(
+    await PlacesUtils.bookmarks.fetch(PlacesUtils.bookmarks.toolbarGuid)
+  );
+  Assert.ok(await PlacesUtils.bookmarks.fetch(PlacesUtils.bookmarks.menuGuid));
+  Assert.ok(await PlacesUtils.bookmarks.fetch(PlacesUtils.bookmarks.tagsGuid));
+  Assert.ok(await PlacesUtils.bookmarks.fetch(PlacesUtils.bookmarks.rootGuid));
+});
+
+add_task(async function test_eraseEverything_reparented() {
+  // Create a folder with 1 bookmark in it...
+  let folder1 = await PlacesUtils.bookmarks.insert({
+    parentGuid: PlacesUtils.bookmarks.toolbarGuid,
+    type: PlacesUtils.bookmarks.TYPE_FOLDER,
+  });
+  let bookmark1 = await PlacesUtils.bookmarks.insert({
+    parentGuid: folder1.guid,
+    url: "http://example.com/",
+  });
+  // ...and a second folder.
+  let folder2 = await PlacesUtils.bookmarks.insert({
+    parentGuid: PlacesUtils.bookmarks.toolbarGuid,
+    type: PlacesUtils.bookmarks.TYPE_FOLDER,
+  });
+
+  // Reparent the bookmark to the 2nd folder.
+  bookmark1.parentGuid = folder2.guid;
+  await PlacesUtils.bookmarks.update(bookmark1);
+
+  // Erase everything.
+  await PlacesUtils.bookmarks.eraseEverything();
+
+  // All the above items should no longer be in the GUIDHelper cache.
+  for (let guid of [folder1.guid, bookmark1.guid, folder2.guid]) {
+    await Assert.rejects(
+      PlacesUtils.promiseItemId(guid),
+      /no item found for the given GUID/
+    );
+  }
+});
+
+add_task(async function test_notifications() {
+  let bms = await PlacesUtils.bookmarks.insertTree({
+    guid: PlacesUtils.bookmarks.unfiledGuid,
+    children: [
+      {
+        title: "test",
+        url: "http://example.com",
+      },
+      {
+        title: "folder",
+        type: PlacesUtils.bookmarks.TYPE_FOLDER,
+        children: [
+          {
+            title: "test2",
+            url: "http://example.com/2",
+          },
+        ],
+      },
+    ],
+  });
+
+  let skipDescendantsObserver = expectPlacesObserverNotifications(
+    ["bookmark-removed"],
+    false,
+    true
+  );
+  let receiveAllObserver = expectPlacesObserverNotifications(
+    ["bookmark-removed"],
+    false
+  );
+
+  await PlacesUtils.bookmarks.eraseEverything();
+
+  let expectedNotifications = [
+    {
+      type: "bookmark-removed",
+      guid: bms[1].guid,
+    },
+    {
+      type: "bookmark-removed",
+      guid: bms[0].guid,
+    },
+  ];
+
+  // If we're skipping descendents, we'll only be notified of the folder.
+  skipDescendantsObserver.check(expectedNotifications);
+
+  // Note: Items of folders get notified first.
+  expectedNotifications.unshift({
+    type: "bookmark-removed",
+    guid: bms[2].guid,
+  });
+
+  receiveAllObserver.check(expectedNotifications);
+});