[MD Bookmarks] Add Delete and Copy URL for Material Bookmarks.

chrome/test/data/webui/md_bookmarks/store_test.js

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 suite('<bookmarks-store>', function() {
   var store;
-  var TEST_TREE = {
-    id: '0',
-    children: [
+  var TEST_TREE;
+
+  setup(function() {
+    TEST_TREE = {
+      id: '0',
+      children: [
       {
         id: '1',
         children: [
           {id: '2', url: 'link2'},
           {id: '3', children: []}
         ]
       },
       {id: '4', url: 'link4'},
       {id: '5', url: 'link5'}
     ]
   };
 
-  setup(function() {
     store = document.createElement('bookmarks-store');
-    store.isTesting_ = true;
     replaceBody(store);
     store.setupStore_(TEST_TREE);
   });
 
   test('initNodes inserts nodes into idToNodeMap', function(){
     assertEquals(TEST_TREE, store.idToNodeMap_['0']);
     assertEquals(TEST_TREE.children[0], store.idToNodeMap_['1']);
     assertEquals(TEST_TREE.children[0].children[0], store.idToNodeMap_['2']);
     assertEquals(TEST_TREE.children[0].children[1], store.idToNodeMap_['3']);
     assertEquals(TEST_TREE.children[1], store.idToNodeMap_['4']);
@@ skipping 11 matching lines @@
     assertEquals(TEST_TREE.children[0].children[1], store.selectedNode);
     store.selectedId = '4';
     assertEquals(TEST_TREE.children[1], store.selectedNode);
     store.selectedId = '5';
     assertEquals(TEST_TREE.children[2], store.selectedNode);
   });
 
   test('correct paths generated for nodes', function() {
     var TEST_PATHS = {
       '0': 'rootNode',
-      '1': 'rootNode.children.0',
-      '2': 'rootNode.children.0.children.0',
-      '3': 'rootNode.children.0.children.1',
-      '4': 'rootNode.children.1',
-      '5': 'rootNode.children.2',
+      '1': 'rootNode.children.#0',
+      '2': 'rootNode.children.#0.children.#0',
+      '3': 'rootNode.children.#0.children.#1',
+      '4': 'rootNode.children.#1',
+      '5': 'rootNode.children.#2',
     };
 
     for (var id in store.idToNodeMap_)
       assertEquals(TEST_PATHS[id], store.idToNodeMap_[id].path);
   });
 
   test('store updates on selected event', function() {
     // First child of root is selected by default.
     assertEquals('1', store.selectedId);
     assertTrue(store.idToNodeMap_['1'].isSelected);
@@ skipping 32 matching lines @@
     store.fire('folder-open-changed', {id: '1', open: true});
 
     // Closing an ancestor folder of a selected folder selects the ancestor.
     store.fire('selected-folder-changed', '3');
     store.fire('folder-open-changed', {id: '1', open: false});
     assertFalse(store.idToNodeMap_['1'].isOpen);
     assertEquals('1', store.selectedId);
     assertTrue(store.idToNodeMap_['1'].isSelected);
     assertFalse(store.idToNodeMap_['3'].isSelected);
   });
+
+  test('deleting a node updates the tree', function() {
+    // Remove an empty folder/bookmark.
+    store.onBookmarkRemoved_('4', {parentId: '0', index: '1'});
+
+    // Check the tree is correct.
+    assertEquals('5', store.rootNode.children[1].id);
+
+    // idToNodeMap_ has been updated.
+    assertEquals(undefined, store.idToNodeMap_['4']);
+    assertEquals(store.rootNode.children[1], store.idToNodeMap_['5']);
+
+    // Paths have been updated.
+    var TEST_PATHS = {
+      '0': 'rootNode',
+      '1': 'rootNode.children.#0',
+      '2': 'rootNode.children.#0.children.#0',
+      '3': 'rootNode.children.#0.children.#1',
+      '5': 'rootNode.children.#1',
+    };
+
+    for (var id in store.idToNodeMap_)
+      assertEquals(TEST_PATHS[id], store.idToNodeMap_[id].path);
+
+    // Remove a folder with children.
+    store.onBookmarkRemoved_('1', {parentId: '0', index: '0'});
+
+    // Check the tree is correct.
+    assertEquals('5', store.rootNode.children[0].id);
+
+    // idToNodeMap_ has been updated.
+    assertEquals(undefined, store.idToNodeMap_['1']);
+    assertEquals(undefined, store.idToNodeMap_['2']);
+    assertEquals(undefined, store.idToNodeMap_['3']);
+    assertEquals(undefined, store.idToNodeMap_['4']);
+    assertEquals(store.rootNode.children[0], store.idToNodeMap_['5']);
+
+    // Paths have been updated.
+    TEST_PATHS = {
+      '0': 'rootNode',
+      '5': 'rootNode.children.#0',
+    };
+
+    for (var id in store.idToNodeMap_)
+      assertEquals(TEST_PATHS[id], store.idToNodeMap_[id].path);
+  });
+
+  test('selectedId updates after removing a selected folder', function() {
+    // Selected folder gets removed.
+    store.selectedId = '2';
+    store.onBookmarkRemoved_('2', {parentId:'1', index:'0'});
+    assertTrue(store.idToNodeMap_['1'].isSelected);
+    assertEquals('1', store.selectedId);
+
+    // A folder with selected folder in it gets removed.
+    store.selectedId = '3';
+    store.onBookmarkRemoved_('1', {parentId:'0', index:'0'});
+    assertTrue(store.idToNodeMap_['0'].isSelected);
+    assertEquals('0', store.selectedId);
+  });
 });