v3.0.1

CHANGELOG.md

@@ -6,6 +6,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [3.0.1] - 2020-03-29
+### Fixed
+- return the updated node in `.insert` 
+
 ## [3.0.0] - 2020-03-28
 ### Changed
 - New release

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@datastructures-js/binary-search-tree",
-  "version": "3.0.0",
+  "version": "3.0.1",
   "description": "binary search tree implementation in javascript",
   "main": "index.js",
   "scripts": {

src/binarySearchTree.js

@@ -23,22 +23,22 @@ class BinarySearchTree {
    * @return {BinarySearchTreeNode}
    */
   insert(key, value, node = this.rootNode) {
-    const newNode = new BinarySearchTreeNode(key, value);
-
     if (node === null) {
-      this.rootNode = newNode;
+      this.rootNode = new BinarySearchTreeNode(key, value);
       this.nodesCount += 1;
-      return newNode;
+      return this.rootNode;
     }
 
     if (key < node.getKey() && node.getLeft() === null) {
+      const newNode = new BinarySearchTreeNode(key, value);
       node.setLeft(newNode);
       newNode.setParent(node);
       this.nodesCount += 1;
       return newNode;
     }
 
     if (key > node.getKey() && node.getRight() === null) {
+      const newNode = new BinarySearchTreeNode(key, value);
       node.setRight(newNode);
       newNode.setParent(node);
       this.nodesCount += 1;
@@ -47,7 +47,7 @@ class BinarySearchTree {
 
     if (key === node.getKey()) {
       node.setValue(value);
-      return newNode;
+      return node;
     }
 
     if (key < node.getKey()) {

test/binarySearchTree.test.js

@@ -7,14 +7,18 @@ describe('binarySearchTree tests', () => {
 
   describe('.insert(key, value)', () => {
     it('should insert nodes to the tree', () => {
-      bst.insert(50, 'n1');
-      bst.insert(80, 'n2');
-      bst.insert(30, 'n3');
-      bst.insert(90, 'n4');
-      bst.insert(60, 'n5');
-      bst.insert(40, 'n6');
-      bst.insert(20, 'n20');
-      bst.insert(20, 'n7'); // updates value of existing node
+      expect(bst.insert(50, 'n1')).to.be.instanceof(BinarySearchTreeNode);
+      expect(bst.insert(80, 'n2')).to.be.instanceof(BinarySearchTreeNode);
+      expect(bst.insert(30, 'n3')).to.be.instanceof(BinarySearchTreeNode);
+      expect(bst.insert(90, 'n4')).to.be.instanceof(BinarySearchTreeNode);
+      expect(bst.insert(60, 'n5')).to.be.instanceof(BinarySearchTreeNode);
+      expect(bst.insert(40, 'n6')).to.be.instanceof(BinarySearchTreeNode);
+      expect(bst.insert(20, 'n20')).to.be.instanceof(BinarySearchTreeNode);
+
+      // updates value of existing node
+      const updated = bst.insert(20, 'n7');
+      expect(updated).to.be.instanceof(BinarySearchTreeNode);
+      expect(updated.getParent().getKey()).to.equal(30);
     });
   });