[MD Bookmarks] Test Util

chrome/browser/resources/md_bookmarks/store.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.
 
-Polymer({
+var BookmarksStore = Polymer({
   is: 'bookmarks-store',
 
   properties: {
     /** @type {BookmarkTreeNode} */
     rootNode: {
       type: Object,
       notify: true,
     },
 
     selectedId: {
@@ skipping 43 matching lines @@
   },
 
   /**
    * @param {BookmarkTreeNode} rootNode
    * @private
    */
   setupStore_: function(rootNode) {
     this.rootNode = rootNode;
     this.idToNodeMap_ = {};
     this.rootNode.path = 'rootNode';
-    this.generatePaths_(rootNode, 0);
-    this.initNodes_(this.rootNode);
+    BookmarksStore.generatePaths(rootNode, 0);
+    BookmarksStore.initNodes(this.rootNode, this.idToNodeMap_);
     this.fire('selected-folder-changed', this.rootNode.children[0].id);
   },
 
   /**
    * Selects the folder specified by the event and deselects the previously
    * selected folder.
    * @param {CustomEvent} e
    * @private
    */
   onSelectedFolderChanged_: function(e) {
@@ skipping 44 matching lines @@
    *           parentId: string,
    *           node: BookmarkTreeNode}} removeInfo
    */
   onBookmarkRemoved_: function(id, removeInfo) {
     if (this.isAncestorOfSelected_(this.idToNodeMap_[id]))
       this.fire('selected-folder-changed', removeInfo.parentId);
 
     var parentNode = this.idToNodeMap_[removeInfo.parentId];
     this.splice(parentNode.path + '.children', removeInfo.index, 1);
     this.removeDescendantsFromMap_(id);
-    this.generatePaths_(parentNode, removeInfo.index);
+    BookmarksStore.generatePaths(parentNode, removeInfo.index);
   },
 
   /**
-   * Initializes the nodes in the bookmarks tree as follows:
-   * - Populates |idToNodeMap_| with a mapping of all node ids to their
-   *   corresponding BookmarkTreeNode.
-   * - Sets all the nodes to not selected and open by default.
-   * @param {BookmarkTreeNode} bookmarkNode
-   * @private
-   */
-  initNodes_: function(bookmarkNode) {
-    this.idToNodeMap_[bookmarkNode.id] = bookmarkNode;
-    if (bookmarkNode.url)
-      return;
-
-    bookmarkNode.isSelected = false;
-    bookmarkNode.isOpen = true;
-    for (var i = 0; i < bookmarkNode.children.length; i++) {
-      this.initNodes_(bookmarkNode.children[i]);
-    }
-  },
-
-  /**
-   * Stores the path from the store to a node inside the node.
-   * @param {BookmarkTreeNode} bookmarkNode
-   * @param {number} startIndex
-   * @private
-   */
-  generatePaths_: function(bookmarkNode, startIndex) {
-    if (!bookmarkNode.children)
-      return;
-
-    for (var i = startIndex; i < bookmarkNode.children.length; i++) {
-      bookmarkNode.children[i].path = bookmarkNode.path + '.children.#' + i;
-      this.generatePaths_(bookmarkNode.children[i], 0);
-    }
-  },
-
-  /**
    * Remove all descendants of a given node from the map.
    * @param {string} id
    * @private
    */
   removeDescendantsFromMap_: function(id) {
     var node = this.idToNodeMap_[id];
     if (!node)
       return;
 
     if (node.children) {
       for (var i = 0; i < node.children.length; i++)
         this.removeDescendantsFromMap_(node.children[i].id);
     }
     delete this.idToNodeMap_[id];
+  },
+});
+
+/**
+ * Stores the path from the store to a node inside the node.
+ * @param {BookmarkTreeNode} bookmarkNode
+ * @param {number} startIndex
+ */
+BookmarksStore.generatePaths = function(bookmarkNode, startIndex) {
+  if (!bookmarkNode.children)
+    return;
+
+  for (var i = startIndex; i < bookmarkNode.children.length; i++) {
+    bookmarkNode.children[i].path = bookmarkNode.path + '.children.#' + i;
+    BookmarksStore.generatePaths(bookmarkNode.children[i], 0);
   }
-});
+};
+
+/**
+ * Initializes the nodes in the bookmarks tree as follows:
+ * - Populates |idToNodeMap_| with a mapping of all node ids to their
+ *   corresponding BookmarkTreeNode.
+ * - Sets all the nodes to not selected and open by default.
+ * @param {BookmarkTreeNode} bookmarkNode
+ * @param {Object=} idToNodeMap
+ */
+BookmarksStore.initNodes = function(bookmarkNode, idToNodeMap) {
+  if (idToNodeMap)
+    idToNodeMap[bookmarkNode.id] = bookmarkNode;
+
+  if (bookmarkNode.url)
+    return;
+
+  bookmarkNode.isSelected = false;
+  bookmarkNode.isOpen = true;
+  for (var i = 0; i < bookmarkNode.children.length; i++)
+    BookmarksStore.initNodes(bookmarkNode.children[i], idToNodeMap);
+};