Add new withChildren() method.

loophp · Dec 10, 2018 · 9aedb95 · 9aedb95
1 parent 4295b3d
commit 9aedb95
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 0 deletions.
diff --git a/spec/drupol/phptree/Node/NodeSpec.php b/spec/drupol/phptree/Node/NodeSpec.php
@@ -183,4 +183,23 @@ public function it_can_get_its_sibblings()
             ->getSibblings()
             ->shouldReturn([$node2, $node3]);
     }
+
+    public function it_can_use_withChildren()
+    {
+        $this
+            ->withChildren()
+            ->shouldNotReturn($this);
+
+        $child = new Node();
+
+        $this
+            ->withChildren($child)
+            ->children()
+            ->shouldReturn([$child]);
+
+        $this
+            ->withChildren()
+            ->children()
+            ->shouldReturn([]);
+    }
 }
diff --git a/src/Node/Node.php b/src/Node/Node.php
@@ -159,4 +159,19 @@ function ($carry, $node) {
             0
         );
     }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function withChildren(NodeInterface ...$nodes): NodeInterface
+    {
+        $clone = clone $this;
+        $clone->storage['children'] = [];
+
+        foreach ($nodes as $node) {
+            $clone->add($node);
+        }
+
+        return $clone;
+    }
 }