Merge pull request #192 from maMykola/fix-closure-with-use-statment

Fix ContextFactory::createForNamespace with closure and use statement

Author: jaapio

src/Types/ContextFactory.php

@@ -181,6 +181,7 @@ public function createForNamespace(string $namespace, string $fileContents): Con
                     $currentNamespace = $this->parseNamespace($tokens);
                     break;
                 case T_CLASS:
+                case T_TRAIT:
                     // Fast-forward the iterator through the class so that any
                     // T_USE tokens found within are skipped - these are not
                     // valid namespace use statements so should be ignored.

tests/unit/Types/ContextFactoryTest.php

@@ -139,6 +139,35 @@ public function bar()
             $this->assertSame([], $context->getNamespaceAliases());
         }
 
+        /**
+         * @covers ::createForNamespace
+         * @uses phpDocumentor\Reflection\Types\Context
+         */
+        public function testTraitContainsClosureWithUseStatement() : void
+        {
+            $php = '<?php declare(strict_types=1);
+                namespace Foo;
+
+                trait FooTrait {
+                    protected function check(array $data, string $key) : void
+                    {
+                        array_walk($data, function(&$item) use ($key) {
+                            // update item based on the key
+                        });
+                    }
+                }
+
+                class FooClass {
+                    use FooTrait;
+                }
+            ';
+
+            $fixture = new ContextFactory();
+            $context = $fixture->createForNamespace('Foo', $php);
+
+            $this->assertSame([], $context->getNamespaceAliases());
+        }
+
         /**
          * @covers ::createFromReflector
          */