Fixed tests

Author: lazychaser

tests/NodeTest.php

@@ -236,14 +236,14 @@ public function testAncestorsReturnsAncestorsWithoutNodeItself()
 
     public function testGetsAncestorsByStatic()
     {
-        $path = Category::ancestorsOf(3)->lists('name')->all();
+        $path = all(Category::ancestorsOf(3)->lists('name'));
 
         $this->assertEquals(array('store', 'notebooks'), $path);
     }
 
     public function testGetsAncestorsDirect()
     {
-        $path = Category::find(8)->getAncestors()->lists('id')->all();
+        $path = all(Category::find(8)->getAncestors()->lists('id'));
 
         $this->assertEquals(array(1, 5, 7), $path);
     }
@@ -264,7 +264,7 @@ public function testDescendants()
 
     public function testWithDepthWorks()
     {
-        $nodes = Category::withDepth()->limit(4)->lists('depth')->all();
+        $nodes = all(Category::withDepth()->limit(4)->lists('depth'));
 
         $this->assertEquals(array(0, 1, 2, 2), $nodes);
     }
@@ -592,15 +592,15 @@ public function testWhereDescendantsOf()
     public function testAncestorsByNode()
     {
         $category = $this->findCategory('apple');
-        $ancestors = Category::whereAncestorOf($category)->lists('id')->all();
+        $ancestors = all(Category::whereAncestorOf($category)->lists('id'));
 
         $this->assertEquals([ 1, 2 ], $ancestors);
     }
 
     public function testDescendantsByNode()
     {
         $category = $this->findCategory('notebooks');
-        $res = Category::whereDescendantOf($category)->lists('id')->all();
+        $res = all(Category::whereDescendantOf($category)->lists('id'));
 
         $this->assertEquals([ 3, 4 ], $res);
     }
@@ -616,4 +616,9 @@ public function testMultipleDeletionsDoNotBrakeTree()
 
         $this->assertTreeNotBroken();
     }
+}
+
+function all($items)
+{
+    return is_array($items) ? $items : $items->all();
 }