Allow to partially test below prototypes nodes

Author: pamil

Partial/PartialNode.php

@@ -5,11 +5,21 @@
 use Matthias\SymfonyConfigTest\Partial\Exception\ChildIsNotAnArrayNode;
 use Matthias\SymfonyConfigTest\Partial\Exception\UndefinedChildNode;
 use Symfony\Component\Config\Definition\ArrayNode;
+use Symfony\Component\Config\Definition\NodeInterface;
+use Symfony\Component\Config\Definition\PrototypedArrayNode;
 
 class PartialNode
 {
+    /**
+     * @var \ReflectionProperty
+     */
     private static $nodeChildrenProperty;
 
+    /**
+     * @var \ReflectionProperty
+     */
+    private static $nodePrototypeProperty;
+
     /**
      * Provide an ArrayNode instance (e.g. the root node created by a TreeBuilder) and a path that is relevant to you,
      * e.g. "dbal.connections": this will strip every node that is not contained in the given path (e.g. the "orm" node
@@ -60,8 +70,18 @@ public static function excludeEverythingNotInPath(ArrayNode $node, array $path =
         self::excludeEverythingNotInPath($nextNode, $path);
     }
 
+    /**
+     * @param ArrayNode $node
+     * @param string $childNodeName
+     *
+     * @return NodeInterface
+     */
     private static function childNode(ArrayNode $node, $childNodeName)
     {
+        if ($node instanceof PrototypedArrayNode && '*' === $childNodeName) {
+            return self::nodePrototypeProperty()->getValue($node);
+        }
+
         $children = self::nodeChildrenProperty()->getValue($node);
 
         if (!isset($children[$childNodeName])) {
@@ -74,6 +94,9 @@ private static function childNode(ArrayNode $node, $childNodeName)
         return $children[$childNodeName];
     }
 
+    /**
+     * @return \ReflectionProperty
+     */
     private static function nodeChildrenProperty()
     {
         if (!isset(self::$nodeChildrenProperty)) {
@@ -86,4 +109,20 @@ private static function nodeChildrenProperty()
 
         return self::$nodeChildrenProperty;
     }
+
+    /**
+     * @return \ReflectionProperty
+     */
+    private static function nodePrototypeProperty()
+    {
+        if (!isset(self::$nodePrototypeProperty)) {
+            self::$nodePrototypeProperty = new \ReflectionProperty(
+                'Symfony\Component\Config\Definition\PrototypedArrayNode',
+                'prototype'
+            );
+            self::$nodePrototypeProperty->setAccessible(true);
+        }
+
+        return self::$nodePrototypeProperty;
+    }
 }

Tests/Partial/PartialNodeTest.php

@@ -4,7 +4,9 @@
 
 use Matthias\SymfonyConfigTest\Partial\PartialNode;
 use Symfony\Component\Config\Definition\ArrayNode;
+use Symfony\Component\Config\Definition\Builder\ArrayNodeDefinition;
 use Symfony\Component\Config\Definition\Builder\TreeBuilder;
+use Symfony\Component\Config\Definition\PrototypedArrayNode;
 
 class PartialNodeTest extends \PHPUnit_Framework_TestCase
 {
@@ -100,6 +102,38 @@ public function it_strips_children_when_leaf_node_is_not_an_array()
         $this->nodeOnlyHasChild($node, 'node_3');
     }
 
+    /**
+     * @test
+     */
+    public function it_does_not_crash_on_prototypes()
+    {
+        $treeBuilder = new TreeBuilder();
+        /** @var ArrayNodeDefinition $root */
+        $root = $treeBuilder->root('root');
+        $root
+            ->prototype('array')
+                ->children()
+                    ->arrayNode('node_1')
+                        ->children()
+                            ->scalarNode('node_1_scalar_node')->end()
+                        ->end()
+                    ->end()
+                    ->arrayNode('node_2')
+                        ->children()
+                            ->scalarNode('node_2_scalar_node')
+        ;
+
+        /** @var PrototypedArrayNode $node */
+        $node = $treeBuilder->buildTree();
+
+        /** @var ArrayNode $prototypeNode */
+        $prototypeNode = $node->getPrototype();
+
+        PartialNode::excludeEverythingNotInPath($node, array('*', 'node_1'));
+
+        $this->nodeOnlyHasChild($prototypeNode, 'node_1');
+    }
+
     /**
      * @test
      */