Add RBTree function

mgechev · mgechev · commit 62e78c512d54 · 2014-05-15T05:16:28.000-07:00
diff --git a/src/data-structures/red-black-tree.js b/src/data-structures/red-black-tree.js
@@ -22,4 +22,11 @@
 
   global.Node = Node;
 
+
+  function RBTree() {
+    this._root = null;
+  }
+
+  global.RBTree = RBTree;
+
 }(typeof window === 'undefined' ? module.exports : window));
diff --git a/test/data-structures/red-black-tree.spec.js b/test/data-structures/red-black-tree.spec.js
@@ -1,7 +1,8 @@
 'use strict';
 
 var mod = require('../../src/data-structures/red-black-tree.js'),
-    Node = mod.Node;
+    Node = mod.Node,
+    RBTree = mod.RBTree;
 
 describe('Node', function () {
   it('should be a constructor function', function () {
@@ -12,5 +13,17 @@ describe('Node', function () {
     expect(node.getKey()).toBe('key');
     expect(node.getValue()).toBe('value');
     expect(node.isRed()).toBeTruthy();
+    node = new Node('key', 'value', undefined);
+    //if we set isRed to falcy it should be turned to red
+    expect(node.isRed()).toBe(false);
+  });
+});
+
+describe('RBTree', function () {
+  it('should be a constructor function', function () {
+    expect(typeof RBTree).toBe('function');
+  });
+  it('should initialize root to null by default', function () {
+    expect(new RBTree()._root).toBeNull();
   });
 });
