BinarySearchTree with objects

Author: doganoo

src/Datastructure/Graph/Tree/BinaryTree/BinarySearchNode.php

@@ -42,7 +42,7 @@ class BinarySearchNode implements IBinaryNode {
      *
      * @param int $value
      */
-    public function __construct(int $value) {
+    public function __construct($value) {
         $this->value = $value;
     }
 
@@ -97,14 +97,14 @@ public function compareTo($object): int {
     /**
      * @return int
      */
-    public function getValue(): int {
+    public function getValue() {
         return $this->value;
     }
 
     /**
      * @param int $value
      */
-    public function setValue(int $value): void {
+    public function setValue($value): void {
         $this->value = $value;
     }
 }

tests/Graph/trees/BinarySearchTreeTest.php

@@ -26,6 +26,7 @@
 use doganoo\PHPAlgorithms\Algorithm\Traversal\InOrder;
 use doganoo\PHPAlgorithms\Algorithm\Traversal\PostOrder;
 use doganoo\PHPAlgorithms\Algorithm\Traversal\PreOrder;
+use doganoo\PHPAlgorithms\Common\Interfaces\Comparable;
 use doganoo\PHPAlgorithms\Datastructure\Graph\Tree\BinarySearchTree;
 
 /**
@@ -85,4 +86,53 @@ public function testPostOrder() {
         $traversal->traverse();
         $this->assertTrue($array === [1, 2, 6, 5]);
     }
-}
+
+    public function testWithObjects() {
+        $tree = new BinarySearchTree();
+        $upper = 10;
+        for ($i = 0; $i < $upper; $i++) {
+            $x = new TestNode($i);
+            $tree->insertValue($x);
+        }
+        $this->assertTrue($tree->height() === $upper);
+        $node = $tree->search(new TestNode(4));
+        $this->assertTrue($node !== null);
+        $node = $tree->search(new TestNode($upper + 5));
+        $this->assertTrue($node === null);
+
+    }
+}
+
+class TestNode implements Comparable {
+    private $id = 0;
+
+    public function __construct($id) {
+        $this->id = $id;
+    }
+
+    /**
+     * @param $object
+     * @return int
+     */
+    public function compareTo($object): int {
+        if (!$object instanceof $this) {
+            return -1;
+        }
+        if ($this->getId() < $object->getId()) {
+            return -1;
+        }
+        if ($this->getId() == $object->getId()) {
+            return 0;
+        }
+        if ($this->getId() > $object->getId()) {
+            return 1;
+        }
+        return -1;
+    }
+
+    public function getId(): int {
+        return $this->id;
+    }
+}
+
+;