MD Bookmarks: Remove deleted nodes from state tree

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

 // Copyright 2017 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('selection state', function() {
   var state;
   var action;
 
   function select(items, anchor, add) {
     return {
@@ skipping 69 matching lines @@
   });
 
   test('deselect items', function() {
     action = select(['1', '2', '3'], '3', false);
     state = bookmarks.SelectionState.updateSelection(state, action);
 
     action = bookmarks.actions.deselectItems();
     state = bookmarks.SelectionState.updateSelection(state, action);
     assertDeepEquals({}, state.items);
   });
+
+  test('deselects items when they are deleted', function() {
+    var nodeList = testTree(createFolder('0', [
+      createFolder(
+          '1',
+          [
+            createItem('2'),
+            createItem('3'),
+            createItem('4'),
+          ]),
+      createItem('5'),
+    ]));
+
+    action = select(['2', '4', '5'], '4', false);
+    state = bookmarks.SelectionState.updateSelection(state, action);
+
+    action = bookmarks.actions.removeBookmark('1', '0', 0, nodeList);
+    state = bookmarks.SelectionState.updateSelection(state, action);
+
+    assertDeepEquals(['5'], normalizeSet(state.items));
+    assertEquals(null, state.anchor);
+  });
 });
 
 suite('closed folder state', function() {
   var nodes;
   var state;
   var action;
 
   setup(function() {
     nodes = testTree(
         createFolder(
@@ skipping 48 matching lines @@
   });
 });
 
 suite('selected folder', function() {
   var nodes;
   var state;
   var action;
 
   setup(function() {
     nodes = testTree(createFolder('1', [
-      createFolder('2', []),
+      createFolder(
+          '2',
+          [
+            createFolder('3', []),
+            createFolder('4', []),
+          ]),
     ]));
 
     state = '1';
   });
 
   test('updates from selectFolder action', function() {
     action = bookmarks.actions.selectFolder('2');
     state = bookmarks.SelectedFolderState.updateSelectedFolder(
         state, action, nodes);
     assertEquals('2', state);
   });
 
   test('updates when parent of selected folder is closed', function() {
     action = bookmarks.actions.selectFolder('2');
     state = bookmarks.SelectedFolderState.updateSelectedFolder(
         state, action, nodes);
 
     action = bookmarks.actions.changeFolderOpen('1', false);
     state = bookmarks.SelectedFolderState.updateSelectedFolder(
         state, action, nodes);
     assertEquals('1', state);
   });
+
+  test('selects ancestor when selected folder is deleted', function() {
+    action = bookmarks.actions.selectFolder('3');
+    state = bookmarks.SelectedFolderState.updateSelectedFolder(
+        state, action, nodes);
+
+    // Delete the selected folder:
+    action = bookmarks.actions.removeBookmark('3', '2', 0, nodes);
+    state = bookmarks.SelectedFolderState.updateSelectedFolder(
+        state, action, nodes);
+
+    assertEquals('2', state);
+
+    action = bookmarks.actions.selectFolder('4');
+    state = bookmarks.SelectedFolderState.updateSelectedFolder(
+        state, action, nodes);
+
+    // Delete an ancestor of the selected folder:
+    action = bookmarks.actions.removeBookmark('2', '1', 0, nodes);
+    state = bookmarks.SelectedFolderState.updateSelectedFolder(
+        state, action, nodes);
+
+    assertEquals('1', state);
+  });
 });
 
 suite('node state', function() {
   var state;
   var action;
 
   setup(function() {
     state = testTree(
         createFolder(
             '1',
@@ skipping 26 matching lines @@
 
     // Cannot edit URL of a folder:
     action = bookmarks.actions.editBookmark('4', {url: 'folder.com'});
     state = bookmarks.NodeState.updateNodes(state, action);
 
     assertEquals('folder', state['4'].title);
     assertEquals(undefined, state['4'].url);
   });
 
   test('updates when a node is deleted', function() {
-    action = bookmarks.actions.removeBookmark('3', '1', 1);
+    action = bookmarks.actions.removeBookmark('3', '1', 1, state);
     state = bookmarks.NodeState.updateNodes(state, action);
 
     assertDeepEquals(['2', '4'], state['1'].children);
 
-    // TODO(tsergeant): Deleted nodes should be removed from the nodes map
-    // entirely.
+    assertDeepEquals(['2', '4'], state['1'].children);
+    assertEquals(undefined, state['3']);
+  });
+
+  test('removes all children of deleted nodes', function() {
+    action = bookmarks.actions.removeBookmark('1', '0', 0, state);
+    state = bookmarks.NodeState.updateNodes(state, action);
+
+    assertDeepEquals(['0', '5'], Object.keys(state).sort());
   });
 
   test('updates when a node is moved', function() {
     // Move within the same folder backwards.
     action = bookmarks.actions.moveBookmark('3', '1', 0, '1', 1);
     state = bookmarks.NodeState.updateNodes(state, action);
 
     assertDeepEquals(['3', '2', '4'], state['1'].children);
 
     // Move within the same folder forwards.
@@ skipping 63 matching lines @@
 
     // Case 2: Clear search by selecting a new folder.
     action = bookmarks.actions.selectFolder('2');
     var selectedState = bookmarks.reduceAction(searchedState, action);
 
     assertEquals('2', selectedState.selectedFolder);
     assertDeepEquals(['3'], bookmarks.util.getDisplayedList(selectedState));
     assertEquals('', selectedState.search.term);
     assertDeepEquals([], selectedState.search.results);
   });
+
+  test('removes deleted nodes', function() {
+    var action;
+
+    action = bookmarks.actions.setSearchTerm('test');
+    state = bookmarks.reduceAction(state, action);
+
+    action = bookmarks.actions.setSearchResults(['1', '3', '2']);
+    state = bookmarks.reduceAction(state, action);
+
+    action = bookmarks.actions.removeBookmark('2', '1', 0, state.nodes);
+    state = bookmarks.reduceAction(state, action);
+
+    // 2 and 3 should be removed, since 2 was deleted and 3 was a descendant of
+    // 2.
+    assertDeepEquals(['1'], state.search.results);
+  });
 });